增加工控网rec文件夹移动到temp功能

yangys

2025-08-23 20ab1a89880e5c6041707e5866260d5ab7ecd983

增加工控网rec文件夹移动到temp功能

 blade-service/blade-mdm/src/main/java/org/springblade/mdm/basesetting/machine/MachineService.java

@@ -201,5 +201,11 @@
        return this.lambdaQuery().eq(Machine::getCode, machineCode).one();
    }


    /**
     * èŽ·å–所有启用状态的机床列表
     * @return æœºåºŠåˆ—表
     */
    public List<Machine> getEnableMachines() {
        return lambdaQuery().eq(Machine::getStatus,Machine.STATUS_ENABLE).list();
    }
}

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

@@ -3,8 +3,6 @@
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.commons.service.ParamService;
@@ -14,8 +12,6 @@
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;
@@ -39,9 +35,10 @@
    // æ¯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() {
    @Scheduled(cron = "${task.cron.machine_file_scan:0 1 * * * ?}")
    public void execute() {
        String networkType = paramService.getParamValue(ParamService.NETWORK_TYPE,ParamService.NETWORK_TYPE_SHEMI);

        if(!ParamService.NETWORK_TYPE_SHEMI.equals(networkType)){

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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,235 @@
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.mdm.basesetting.machine.MachineService;
import org.springblade.mdm.basesetting.machine.entity.Machine;
import org.springblade.mdm.commons.service.ParamService;
import org.springblade.mdm.gkw.programnode.entity.MachineFile;
import org.springblade.mdm.gkw.programnode.service.MachineFileService;
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.StandardCopyOption;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.FileTime;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.List;

/**
 * æœºåºŠå›žä¼ æ–‡ä»¶å¤¹rec，目录文件超过机床设定的时间移动到temp文件夹
 */
@Slf4j
@Component
@EnableScheduling
public class ReceiveDirClearTask {
    @Autowired
    private MachineFileService machineFileService;
    @Autowired
    private MachineService machineService;
    @Autowired
    private ParamService paramService;
    /**
     * æ–‡ä»¶é»˜è®¤ä¿å­˜å°æ—¶æ•°
     */
    private static final int DEFAULT_HOUR = 8;
    // æ¯5秒执行一次
    //@Scheduled(fixedRate = 1000000)
    //@Scheduled(cron = "0 1 0 * * ?") // æ¯å¤©ä¸Šåˆ0点1分执行
    //@Scheduled(cron = "0 */3 * * * ?")
    //@Scheduled(cron = "0 15 19 * * ?") //test
    //@Scheduled(cron = "${task.cron.machine_file_scan}")
    @Scheduled(cron = "${task.cron.machine_rec_clean:0 3 * * * ?}")
    public void execute() {
        String networkType = paramService.getParamValue(ParamService.NETWORK_TYPE,ParamService.NETWORK_TYPE_SHEMI);

        if(!ParamService.NETWORK_TYPE_SHEMI.equals(networkType)){
            //非涉密网，才扫描目录文件
            moveOverTimeFiles();
        }

    }

    /**
     * æ‰«ææ•°æ®åº“记录，超时则移动文件
     */
    public void moveOverTimeFiles() {
        List<Machine> machines = machineService.getEnableMachines();
        LocalDateTime now = LocalDateTime.now();
        for (Machine machine : machines) {
            int remainHours = machine.getReceiveDirExpiryHours() != null ?machine.getReceiveDirExpiryHours():DEFAULT_HOUR;
            LocalDateTime earlyTime = now.minusHours(remainHours);
            List<MachineFile> overTimeFiles = this.machineFileService.lambdaQuery().lt(MachineFile::getFileCreateTime,earlyTime)
                .eq(MachineFile::getDirType,MachineFile.DIR_TYPE_REC)
                .eq(MachineFile::getMachineCode,machine.getCode()).list();

            for(MachineFile overTimeFile : overTimeFiles){
                try {
                    moveFileToTemp(overTimeFile, machine);
                }catch(Exception e){
                    log.error("移动rec文件失败:"+overTimeFile.getName(),e);
                }
            }

        }

    }

    /**
     * æ–‡ä»¶ç§»åŠ¨åˆ°temp
     * @param overTimeFile
     */
    private void moveFileToTemp(MachineFile overTimeFile,Machine machine) throws IOException {
        String srcFilepath = MachineFileService.getBasePath(machine,overTimeFile.getDirType())+ File.separator+overTimeFile.getName();
        Path source = Paths.get(srcFilepath);
        if(!source.toFile().exists()){
            return;
        }
        String targetFilepath = MachineFileService.getBasePath(machine,MachineFile.DIR_TYPE_TEMP)+ File.separator+overTimeFile.getName();
        Path target = Paths.get(targetFilepath);

        Path targetDir = target.getParent();
        if (targetDir != null && !Files.exists(targetDir)) {
            Files.createDirectories(targetDir);
        }

        // ä½¿ç”¨ REPLACE_EXISTING é€‰é¡¹æ¥è¦†ç›–已存在的文件
        Files.move(source, target, StandardCopyOption.REPLACE_EXISTING);
    }


    /**
     * æ‰«æç›®å½•
     * @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);
            }
        }
    }
}