工控网分析程序状态问题修复

yangys

2025-09-18 a3048fa6fa72fa3cc5da2c43c59bd000e00c9599

 blade-service/blade-mdm/src/main/java/org/springblade/mdm/task/MachineFileScanTask.java

@@ -1,29 +1,16 @@
package org.springblade.mdm.task;

import com.alibaba.excel.util.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.FileUtil;
import org.springblade.mdm.basesetting.machine.MachineService;
import org.springblade.mdm.basesetting.machine.entity.Machine;
import org.springblade.mdm.gkw.programnode.entity.MachineFile;
import org.springblade.mdm.gkw.programnode.service.MachineFileService;
import org.springblade.mdm.basesetting.machine.service.MachineService;
import org.springblade.mdm.commons.service.ParamService;
import org.springblade.mdm.machinefile.service.MachineFileScanService;
import org.springblade.mdm.machinefile.service.MachineFileService;
import org.springblade.mdm.program.service.ProgramAnnotationService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.FileTime;
import java.util.Date;
import java.util.List;

@Slf4j
@Component
@@ -33,182 +20,26 @@
   private MachineFileService machineFileService;
   @Autowired
   private MachineService machineService;
   @Autowired
   private ParamService paramService;
   @Autowired
   private ProgramAnnotationService programAnnotationService;
   @Autowired
   private MachineFileScanService machineFileScanService;
   // 每5秒执行一次
   //@Scheduled(fixedRate = 1000000)
   //@Scheduled(cron = "0 1 0 * * ?") // 每天上午0点1分执行
   @Scheduled(cron = "0 */5 * * * ?")
   //@Scheduled(cron = "0 */5 * * * ?")
   //@Scheduled(cron = "0 15 19 * * ?") //test
   public void executeEvery5Seconds() {
      scanMachineFile();
   }
   @Scheduled(cron = "${task.cron.machine_file_scan:0 1 0 * * ?}")
   public void execute() {
      String networkType = paramService.getParamValue(ParamService.NETWORK_TYPE,ParamService.NETWORK_TYPE_SHEMI);

   /**
    * 扫描所有机床的配置之目录
    */
   public void scanMachineFile() {
      List<Machine> machines = machineService.lambdaQuery().eq(Machine::getStatus,Machine.STATUS_ENABLE).list();
      for (Machine machine : machines) {

         //扫描新闻界
         try {
            scanDir(machine,MachineFile.DIR_TYPE_REC);
         }catch(Exception e) {
            log.error("REC扫描文件异常,机床"+machine.getCode(),e);
         }
         try {
            scanDir(machine,MachineFile.DIR_TYPE_SEND);
         }catch(Exception e) {
            log.error("SEND扫描文件异常,机床"+machine.getCode(),e);
         }
         try {
            scanDir(machine,MachineFile.DIR_TYPE_TEMP);
         }catch(Exception e) {
            log.error("TEMP扫描文件异常,机床"+machine.getCode(),e);
         }

         //清理数据库，删除文件不存在的数据
         try {
            clearDeletedReccords(machine,MachineFile.DIR_TYPE_REC);
         }catch(Exception e) {
            log.error("REC扫描文件异常,机床"+machine.getCode(),e);
         }
         try {
            clearDeletedReccords(machine,MachineFile.DIR_TYPE_SEND);
         }catch(Exception e) {
            log.error("SEND扫描文件异常,机床"+machine.getCode(),e);
         }
         try {
            clearDeletedReccords(machine,MachineFile.DIR_TYPE_TEMP);
         }catch(Exception e) {
            log.error("TEMP扫描文件异常,机床"+machine.getCode(),e);
         }
      if(!ParamService.NETWORK_TYPE_SHEMI.equals(networkType)){
         //非涉密网，才扫描目录文件
         machineFileScanService.scanMachineFile();
      }

   }


   /**
    * 扫描目录
    * @param machine 机床信息
    * @param dirType 目录类型
    * @throws IOException
    */
   public void scanDir(Machine machine,String dirType) throws IOException {
      String basePath = MachineFileService.getBasePath(machine,dirType);
      if(basePath == null) {
         log.warn("目录类型不匹配:{}",dirType);
         return;
      }
      Path dirPath = Paths.get(basePath);
      if(!dirPath.toFile().exists()){
         log.warn("扫描目录:{} 不存在",dirPath);
         return;
      }
      // 只包含普通文件
      List<Path> files = Files.list(dirPath)
         .filter(Files::isRegularFile).toList();

      byte[] buffer = new byte[2048];

      MachineFile existFileInDb;
      //boolean exists;//文件是否存在于数据库中
      for (Path filePath : files) {

         existFileInDb = null;
         //exists = false;

         MachineFile mf = new MachineFile();

         mf.setTenantId("000000");
         mf.setName(filePath.toFile().getName());
         mf.setDirType(dirType);
         mf.setMachineCode(machine.getCode());

         try {
            BasicFileAttributes attrs = Files.readAttributes(
               filePath,
               BasicFileAttributes.class
            );

            FileTime creationTime = attrs.creationTime();
            Date creationDate = new Date(creationTime.toMillis());
            mf.setFileCreateTime(creationDate);

            FileTime modifyTime = attrs.lastModifiedTime();
            mf.setFileModifyTime(new Date(modifyTime.toMillis()));
            mf.setFileSize(Files.size(filePath));
            try (InputStream inputStream = Files.newInputStream(filePath)) {
               // 使用输入流读取文件内容
               int bytesRead = inputStream.read(buffer);
               mf.setMd5(DigestUtils.md5Hex(buffer));
            } catch (IOException e) {
               log.error("读取文件md5失败",e);
               continue;//有错误,掠过
            }

            //exists = machineFileService.fileExists(mf.getName(),dirType,machine.getCode());

            existFileInDb = machineFileService.getExistsFile(mf.getName(),dirType,machine.getCode());
            System.out.println("文件创建时间: " + creationDate);
         } catch (IOException e) {
            log.error("读取文件信息失败",e);
         }
         if(existFileInDb == null) {
            machineFileService.save(mf);
         }else{
            log.info("文件已如果掠过:{}",filePath.getFileName());
            //已存在咋整。
            existFileInDb.setFileSize(mf.getFileSize());

            existFileInDb.setFileCreateTime(mf.getFileCreateTime());
            existFileInDb.setFileModifyTime(mf.getFileModifyTime());
            if(!StringUtils.equals(existFileInDb.getMd5(),mf.getMd5())){
               //文件内容发生变化了,设置状态为初始状态
               existFileInDb.setStatus(MachineFile.STATUS_NORMAL);
            }
            existFileInDb.setMd5(mf.getMd5());


            machineFileService.updateById(existFileInDb);

         }

      }


   }

   /**
    * 清理已经删除了文件的记录
    * @param machine
    * @param dirType
    * @throws IOException
    */
   void clearDeletedReccords(Machine machine,String dirType) throws IOException {
      List<MachineFile> machineFiles =  this.machineFileService.lambdaQuery().eq(MachineFile::getDirType,dirType)
         .eq(MachineFile::getMachineCode,machine.getCode()).list();

      String basePath = MachineFileService.getBasePath(machine,dirType);
      if(basePath == null) {
         log.warn("目录类型不匹配:{}",dirType);
         return;
      }
      Path dirPath = Paths.get(basePath);
      if(!dirPath.toFile().exists()){
         log.warn("扫描目录:{} 不存在",dirPath);
         return;
      }

      List<Path> files = Files.list(dirPath)
         .filter(Files::isRegularFile).toList();

      for(MachineFile mf : machineFiles){
         long findCount = files.stream().filter(filePath -> filePath.toFile().getName().equals(mf.getName())).count();
         if(findCount == 0){
            //文件夹内没找到，需要删除记录
            mf.markFileDeleted();
            this.machineFileService.updateById(mf);
         }
      }
   }
}