增加程序文件列表

yangys

2025-06-29 4272f2232fd97d9862bc53e583612c8a4bf82ee4

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

@@ -25,21 +25,31 @@
 */
package org.springblade.mdm.program.service;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.flowable.engine.*;
import org.springblade.core.mp.base.BizServiceImpl;
import org.springblade.core.oss.OssTemplate;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.mdm.basesetting.machine.MachineService;
import org.springblade.mdm.basesetting.machine.entity.Machine;
import org.springblade.mdm.program.entity.NcNode;
import org.springblade.mdm.program.mapper.NcProgramMapper;
import org.springblade.mdm.program.entity.NcProgram;
import org.springblade.mdm.program.vo.DncSendBackData;
import org.springblade.mdm.program.vo.NcProgramUploadVO;
import org.springblade.mdm.program.vo.NcProgramVO;
import org.springblade.mdm.utils.FileContentUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

@@ -53,13 +63,10 @@
@AllArgsConstructor
public class NcProgramService  extends BizServiceImpl<NcProgramMapper, NcProgram> {

   private final ObjectMapper objectMapper;
   private final RepositoryService repositoryService;
   private final RuntimeService runtimeService;
   private final HistoryService historyService;
   private final TaskService taskService;
   private final ProcessEngine processEngine;
   private final MachineService machineService;

   private final OssTemplate ossTemplate;
   private final NcNodeService nodeService;
   /**
    * 查询现有固化的程序，暂定条件：零组件号相同，且是同一机床组
    * @param partNo
@@ -71,21 +78,75 @@
      return this.getBaseMapper().getCuredNcProgram(partNo,machine.getMachineGroupCode());
   }

   /**
    * 查询现有固化的程序，暂定条件：零组件号相同，且是同一机床组
    * @param name 程序名称
    * @return
    */
   public NcProgram getLastNcProgram(String name) {
      return this.getBaseMapper().getLastNcProgram(name);
   }

   /**
    * 上传程序文件到指定节点
    * @param file
    * @param nodeId
    * @param category
    * @param category 文件分类（程序文件/其他文件）
    */
   public void uploadNcFile(MultipartFile file, Long nodeId, String category) {
   public void uploadProgramFile(NcProgramUploadVO uploadVO) {
      MultipartFile file = uploadVO.getFile();
      BladeFile bfile = ossTemplate.putFile(uploadVO.getFile());
      String link = bfile.getLink();
      //保存profame
      NcProgram prog = new NcProgram();
      prog.setName(file.getOriginalFilename());
      prog.setNcNodeId(uploadVO.getNodeId());
      prog.setOssName(bfile.getName());
      prog.setDrawingNo(uploadVO.getDrawingNo());
      boolean isTextFile = false;
      try {
         isTextFile = FileContentUtil.isTextFile(file.getInputStream());
      } catch (IOException e) {
         log.warn("判断是否文本文件异常",e);
      }
      prog.setIsTextFile(isTextFile);
      prog.setUrl(link);
      prog.setCategory(uploadVO.getCategory());

      this.save(prog);
   }

   /**
    * 删除一个程序
    * @param id
    */
   public String getFileContent(Long id) {
      String result  ="";

      NcProgram prog = this.getById(id);
      String fileName = prog.getOssName();
      try (InputStream inputStream = ossTemplate.statFileStream(fileName)) {
         result = IOUtils.toString(inputStream, StandardCharsets.UTF_8);
      } catch (IOException e) {
         throw new RuntimeException(e);
      }
      return result;
   }
   /**
    * 删除一个程序
    * @param id
    */
   public void removeProgram(Long id) {
      NcProgram prog = this.getById(id);
      ossTemplate.removeFile(prog.getOssName());
      this.getBaseMapper().deleteById(id);
   }

   public List<NcProgramVO> listByNode(Long nodeId) {
      NcNode node = nodeService.getById(nodeId);
      List<NcProgram> pList = this.list(Wrappers.lambdaQuery(NcProgram.class).eq(NcProgram::getNcNodeId, nodeId).orderByDesc(NcProgram::getCreateTime));

      List<NcProgramVO> result = new ArrayList<>();

      pList.forEach(prog -> {
         NcProgramVO vo = new NcProgramVO();
         BeanUtils.copyProperties(prog, vo);
         vo.setMachineCode(node.getMachineCode());
         result.add(vo);
      });

      return result;
   }
}