diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/SysEdFileInfoController.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/SysEdFileInfoController.java index 4c4fa1e..fd9fda8 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/SysEdFileInfoController.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/controller/SysEdFileInfoController.java @@ -210,8 +210,8 @@ public class SysEdFileInfoController { */ @GetMapping("/importPrj") @UserOperation(value = "导入本地工程", modelName = UserOperationModuleEnum.SYS_PRJ_DATABASE) - public ElectromagneticResult importPrj(@RequestParam("file") MultipartFile file) { - return edFileInfoService.importPrj(file); + public ElectromagneticResult importPrj(@RequestParam("file") MultipartFile file, @RequestParam(name = "templateCode", defaultValue = "bhdcfzrjzt", required = false) String templateCode) { + return edFileInfoService.importPrj(file, templateCode); } } diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/EdFileInfoService.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/EdFileInfoService.java index 517d026..9671972 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/EdFileInfoService.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/EdFileInfoService.java @@ -247,5 +247,5 @@ public interface EdFileInfoService { /** * 导入工程 */ - ElectromagneticResult importPrj(MultipartFile file); + ElectromagneticResult importPrj(MultipartFile file, String templateCode); } diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java index c2d4426..717967f 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/EdFileInfoServiceImpl.java @@ -1517,13 +1517,13 @@ public class EdFileInfoServiceImpl extends ServiceImpl importPrj(MultipartFile file) { + public ElectromagneticResult importPrj(MultipartFile file, String templateCode) { try { // 创建工程 String prjId = updateImportPrj2Db(file); // 层级沿用,先这样写,后续根据需求扩展 EdFileInfo prj = this.baseMapper.selectOne(Wrappers.lambdaQuery(EdFileInfo.class) - .eq(EdFileInfo::getTemplateCode, "bhdcfzrjzt") + .eq(EdFileInfo::getTemplateCode, templateCode) .eq(EdFileInfo::getEffectFlag, EffectFlagEnum.EFFECT.code)); commonService.follow(prj.getId(), prjId, DataOwnEnum.SYS_PRJ.code); // 工程发布