增补上传的文件类型字段

yangys

2025-08-13 c6b49750da5d19383e74e644f4fa1734138aff89

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

@@ -1,5 +1,6 @@
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;
@@ -35,12 +36,15 @@
   // 每5秒执行一次
   //@Scheduled(fixedRate = 1000000)
   //@Scheduled(cron = "0 1 0 * * ?") // 每天上午0点1分执行
   @Scheduled(cron = "0 15 19 * * ?") //test
   @Scheduled(cron = "0 */5 * * * ?")
   //@Scheduled(cron = "0 15 19 * * ?") //test
   public void executeEvery5Seconds() {
      scanMachineFile();
   }


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

      byte[] buffer = new byte[2048];

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

         exists = false;
         existFileInDb = null;
         //exists = false;

         MachineFile mf = new MachineFile();

@@ -129,7 +135,7 @@
            mf.setFileCreateTime(creationDate);

            FileTime modifyTime = attrs.lastModifiedTime();
            mf.setFileModifyDate(new Date(modifyTime.toMillis()));
            mf.setFileModifyTime(new Date(modifyTime.toMillis()));
            mf.setFileSize(Files.size(filePath));
            try (InputStream inputStream = Files.newInputStream(filePath)) {
               // 使用输入流读取文件内容
@@ -140,15 +146,31 @@
               continue;//有错误,掠过
            }

            exists = machineFileService.fileExists(mf.getName(),mf.getMd5(),machine.getCode());
            //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(!exists) {
         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);

         }

      }
@@ -172,16 +194,20 @@
         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 -> {return filePath.toFile().getName().equals(mf.getName());}).count();
         long findCount = files.stream().filter(filePath -> filePath.toFile().getName().equals(mf.getName())).count();
         if(findCount == 0){
            //文件夹内没找到，需要删除记录
            this.machineFileService.save(mf);
            mf.markFileDeleted();
            this.machineFileService.updateById(mf);
         }
      }
   }