增加替换文件上传

yangys

2025-08-11 b5b668d2728665ca05d15be8862ceb0a2796fc67

 blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/DNCSendBackService.java

@@ -2,6 +2,7 @@
package org.springblade.mdm.program.service;

import com.alibaba.fastjson.JSONObject;
import io.netty.util.internal.StringUtil;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
@@ -17,6 +18,7 @@
import org.springblade.core.tool.utils.Func;
import org.springblade.mdm.flow.entity.FlowProgramFile;
import org.springblade.mdm.flow.service.CureFlowService;
import org.springblade.mdm.flow.service.FlowCommonService;
import org.springblade.mdm.flow.service.FlowProgramFileService;
import org.springblade.mdm.program.entity.NcNode;
import org.springblade.mdm.program.entity.NcProgramExchange;
@@ -50,6 +52,7 @@
   private final NcNodeService ncNodeService;
   private final OssTemplate ossTemplate;
   private final BladeRedis bladeRedis;
   private final FlowCommonService flowCommonService;

   private String getFileKey(){
      return "dncexpfile-"+ AuthUtil.getUserId();
@@ -61,6 +64,13 @@
    */
   public List<DncSendBackData> dncSendBackUpload(MultipartFile file) {
      List<DncSendBackData> list;
      if(file == null || file.isEmpty()){
         throw new ServiceException("文件为空");
      }

      if(!StringUtils.endsWith(file.getOriginalFilename(),".zip")){
         throw new ServiceException("文件必须为zip包 ");
      }
      try {
         BladeFile bfile = ossTemplate.putFile(file);//上传，供后续入库使用
         //设置一个缓存，2小时过期
@@ -69,7 +79,7 @@
         InputStream zipFileInputStream = file.getInputStream();//test

         //byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
         list = parseDncZipFromByteArray(zipFileInputStream);
         list = parseProgramListFromZip(zipFileInputStream);

      } catch (IOException e) {
         log.error("上传dnc回传文件失败",e);
@@ -85,11 +95,10 @@
    * @return 回传程序列表
    * @throws IOException 文件操作异常
    */
   List<DncSendBackData> parseDncZipFromByteArray(InputStream inputStream) throws IOException {
   List<DncSendBackData> parseProgramListFromZip(InputStream inputStream) throws IOException {
      List<DncSendBackData> list = new ArrayList<>();
      Path tempZipFile = createTempFile(inputStream);
      //Map<String,String> fileMd5Map = new HashMap<>();
      Map<String,DncSendBackData> fileDataMap = new HashMap<>();

      try (ZipFile zipFile = new ZipFile(tempZipFile.toFile())) {

         ZipEntry entry;
@@ -102,9 +111,11 @@
               //目录，才是程序包，
               //查询数据库，对应上才能确认时有效的程序包
               String packageName = StringUtils.removeEnd(entryName,"/");
               List<NcNode> pkgList = ncNodeService.lambdaQuery().eq(NcNode::getNodeType,NcNode.TYPE_PROGRAM_PACKAGE).eq(NcNode::getName, packageName).eq(NcNode::getIsLastEdition,1).list();
               if(!pkgList.isEmpty()) {
                  NcNode programPackageNode = pkgList.get(0);
               NcNode programPackageNode = ncNodeService.getLastEditionTryingProgramPackage(packageName);
               NcNode curedProgramPkg = ncNodeService.getLastEditionCuredProgramPackage(packageName);

               if(programPackageNode!=null && curedProgramPkg == null) {//无固化程序，才放入列表
                  //NcNode programPackageNode = pkgList.get(0);
                  progData.setId(programPackageNode.getId());
                  progData.setProgramName(packageName);

@@ -112,6 +123,7 @@
                  progData.setProgramNo(programPackageNode.getProgramNo());
                  list.add(progData);
               }

            }
         }

@@ -130,11 +142,13 @@
      List<Long> idList = Func.toLongList(ids);
      //
      NcProgramExchange exchange;
      String pkgFileName = bladeRedis.get(getFileKey());
      String filekey = getFileKey();
      String pkgFileName = bladeRedis.get(filekey);
      log.info("filekey={},文件名={}",filekey,pkgFileName);

      Map<Long,List<FlowProgramFile>> programPackageFileMap = new HashMap<>();
      List<NcNode> newProgramPckageList = updateNodeDataByDNCBackData(pkgFileName,idList,programPackageFileMap);

      Map<Long,List<NcNode>> programPackageSubMap = new HashMap<>();
      List<NcNode> newProgramPckageList = updateNodeDataByDNCBackData(pkgFileName,idList,programPackageSubMap);
      log.info("需要启动固化流程的程序包名数量:{}",newProgramPckageList.size());
      for(NcNode pkgNode :newProgramPckageList){
         exchange = new NcProgramExchange();
         exchange.setName(pkgNode.getName());
@@ -143,18 +157,21 @@

         this.save(exchange);
      }
      bladeRedis.del(filekey);
      this.ossTemplate.removeFile(pkgFileName);
      log.info("删除oss文件:{}",pkgFileName);

      cureFlowService.startCure(newProgramPckageList,programPackageFileMap);
      cureFlowService.startCure(newProgramPckageList,programPackageSubMap);
   }

   /**
    * 更新节点，主要是创建 程序包名 的新版本。
    * @param pkgFileName zip文件名
    * @param programPackageIdList 程序包名 节点的id列表
    * @param programPackageFileMap 新的 程序包节点id -> =文件列表 map，用于回传数据
    * @param programPackageSubMap 新的 程序包节点id -> =文件列表 map，用于回传数据
    * @throws IOException 访问文件异常
    */
   List<NcNode> updateNodeDataByDNCBackData(String pkgFileName, List<Long> programPackageIdList,Map<Long,List<FlowProgramFile>> programPackageFileMap) throws IOException {
   List<NcNode> updateNodeDataByDNCBackData(String pkgFileName, List<Long> programPackageIdList,Map<Long,List<NcNode>> programPackageSubMap) throws IOException {
      InputStream inputStream = this.ossTemplate.statFileStream(pkgFileName);
      Path tempZipFile = createTempFile(inputStream);
      List<NcNode> newProgramPackageNodeList = new ArrayList<>();
@@ -167,19 +184,35 @@
            entry = entris.nextElement();
            entryNameList.add(entry.getName());
         }

         log.info("allentrynames:{}",entryNameList);
         //根据内部文件，读取和分析程序包和程序文件数据
         List<String> dirList = entryNameList.stream().filter(s -> s.endsWith("/")).toList();
         for(String dir : dirList){
            String programPackageName = StringUtils.removeEnd(dir,"/");
            NcNode oriProgramPkg = this.ncNodeService.getLastEditionProgramPackage(programPackageName);

            NcNode oriProgramPkg = this.ncNodeService.getLastEditionTryingProgramPackage(programPackageName);
            NcNode curedProgramPkg = ncNodeService.getLastEditionCuredProgramPackage(programPackageName);
            if(curedProgramPkg != null){
               throw new ServiceException(programPackageName+"已经固化，请勿重复入库。");
            }
            if(oriProgramPkg == null){
               log.warn("未发现匹配的程序包名{}",programPackageName);
               log.warn("文件夹未发现匹配的程序包名{}",dir);
               continue;
            }
            //检查是否在审批过程中
            //根据节点信息查询流程
            boolean active = flowCommonService.isProcessInstanceActive(oriProgramPkg.getProcessInstanceId());
            if(active){
               throw new ServiceException(programPackageName+"正在审批中，请勿重复入库。");
            }
            /*
            if(oriProgramPkg.hasCured()){
               throw new ServiceException(programPackageName+"已经固化，请勿重复入库。");
            }*/

            if(!programPackageIdList.contains(oriProgramPkg.getId())){
               //不在勾选的范围内
               log.info("{}不在勾选范围内",programPackageName);
               continue;
            }
            NcNode newProgramPkg = new NcNode();
@@ -194,7 +227,8 @@
            oriProgramPkg.setIsLastEdition(0);;
            ncNodeService.updateById(oriProgramPkg);

            List<FlowProgramFile> newFlowFiles = new ArrayList<>();
            //List<FlowProgramFile> newFlowFiles = new ArrayList<>();
            List<NcNode> newProgramNodes = new ArrayList<>();
            //查找包下的文件数据,
            entryNameList.stream().filter(s -> s.startsWith(dir)).forEach(entryName -> {
               log.info("{}下的文件：{}",dir,entryName);
@@ -211,9 +245,9 @@
                  BeanUtils.copyProperties(oldProgramNode, newProgramNode);
                  clearBaseProperties(newProgramNode);
                  newProgramNode.setIsLastEdition(1);
                  newProgramNode.setParentId(newProgramPkg.getParentId());
                  newProgramNode.setParentId(newProgramPkg.getId());
                  newProgramNode.setParentIds(newProgramPkg.getParentIds()+","+newProgramPkg.getId());
                  ncNodeService.save(newProgramNode);
                  //ncNodeService.save(newProgramNode);
                  //将回传文件上传，并记录到数据库

                  FlowProgramFile oldFlowFile = flowProgramFileService.getById(newProgramNode.getFlowProgramFileId());
@@ -230,19 +264,21 @@
                  } catch (IOException e) {
                     throw new RuntimeException(e);
                  }

                  newFlowFiles.add(newFlowFile);

                  newProgramNode.setFlowProgramFile(newFlowFile);
                  newProgramNode.setVersionNumber(oldProgramNode.genNewVersionNumber());
                  newProgramNodes.add(newProgramNode);
                  oldProgramNode.setIsLastEdition(0);
                  oldProgramNode.setIsLocked(1);
                  ncNodeService.updateById(oldProgramNode);
               }
            });

            programPackageFileMap.put(newProgramPkg.getId(),newFlowFiles);
            programPackageSubMap.put(newProgramPkg.getId(),newProgramNodes);
         }

      }


      return newProgramPackageNodeList;
   }