package org.springblade.mdm.task;
|
|
import lombok.extern.slf4j.Slf4j;
|
import org.springblade.mdm.basesetting.machine.entity.Machine;
|
import org.springblade.mdm.basesetting.machine.service.MachineService;
|
import org.springblade.mdm.commons.service.ParamService;
|
import org.springblade.mdm.gkw.programnode.vo.ProgramNameVO;
|
import org.springblade.mdm.machinefile.entity.FileSendRecord;
|
import org.springblade.mdm.machinefile.entity.MachineFile;
|
import org.springblade.mdm.machinefile.service.FileSendRecordService;
|
import org.springblade.mdm.machinefile.service.MachineFileService;
|
import org.springblade.mdm.utils.FileContentUtil;
|
import org.springblade.mdm.utils.ProgramFileNameParser;
|
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.time.LocalDateTime;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.Optional;
|
import java.util.stream.Collectors;
|
|
/**
|
* 机床回传rec文件夹内文件异常检查
|
*/
|
@Slf4j
|
@Component
|
@EnableScheduling
|
public class ReceiveDirCheckTask {
|
@Autowired
|
private MachineFileService machineFileService;
|
@Autowired
|
private MachineService machineService;
|
@Autowired
|
private ParamService paramService;
|
@Autowired
|
private FileSendRecordService fileSendRecordService;
|
/**
|
* 文件默认保存小时数，0不限制
|
*/
|
private static final int DEFAULT_HOUR = 0;
|
// 每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_check:0 */10 * * * ?}")
|
public void execute() {
|
String networkType = paramService.getParamValue(ParamService.NETWORK_TYPE,ParamService.NETWORK_TYPE_SHEMI);
|
|
if(!ParamService.NETWORK_TYPE_SHEMI.equals(networkType)){
|
//非涉密网，才扫描目录文件
|
checkFiles();
|
}
|
|
}
|
|
/**
|
* 扫描数据库记录，超时则移动文件
|
*/
|
public void checkFiles() {
|
List<Machine> machines = machineService.getEnableMachines();
|
|
for (Machine machine : machines) {
|
|
List<MachineFile> recFiles = this.machineFileService.lambdaQuery()
|
.eq(MachineFile::getDirType,MachineFile.DIR_TYPE_REC)
|
.eq(MachineFile::getMachineCode,machine.getCode()).list();
|
|
for(MachineFile machineFile : recFiles){
|
try {
|
checkFile(machineFile, recFiles,machine);
|
}catch(Exception e){
|
log.error("检查rec文件失败:"+machineFile.getName(),e);
|
}
|
}
|
|
}
|
|
}
|
|
/**
|
* 文件移动到temp
|
* @param machineFile
|
*/
|
private void checkFile(MachineFile machineFile,List<MachineFile> allFilesInDir,Machine machine) throws IOException {
|
String srcFilepath = MachineFileService.getBasePath(machine,machineFile.getDirType())+ File.separator+machineFile.getName();
|
Path source = Paths.get(srcFilepath);
|
if(!source.toFile().exists()){
|
return;
|
}
|
try(InputStream fileIns = Files.newInputStream(Paths.get(srcFilepath));) {
|
if(!FileContentUtil.isTextFile(fileIns)){
|
//非文本
|
machineFile.setExceptionType(MachineFile.EXCEPTION_NOT_TEXT);
|
}else{
|
ProgramNameVO progNameVO = ProgramFileNameParser.parseProgramName(machineFile.getName());
|
if(!progNameVO.isValidFilename()){
|
machineFile.setExceptionType(MachineFile.EXCEPTION_BAD_FILENAME);
|
}else {
|
String prefix = progNameVO.logicProgramName()+"-";
|
long matchCount = allFilesInDir.stream().filter(file -> file.getName().startsWith(prefix)).count();
|
if(matchCount != progNameVO.getSegmentCount()){//文件段数缺失
|
machineFile.setExceptionType(MachineFile.EXCEPTION_LOST_FILES);
|
}else{
|
//检查是否匹配下发记录的段数
|
//正负3秒作为查询时间
|
Date beginTime = new Date(machineFile.getFileCreateTime().getTime()-3000);
|
Date endTime = new Date(machineFile.getFileCreateTime().getTime()+3000);
|
Optional<FileSendRecord> optFile = fileSendRecordService.lambdaQuery()
|
.eq(FileSendRecord::getMachineCode,machineFile.getMachineCode())
|
.likeRight(FileSendRecord::getName,prefix).between(FileSendRecord::getCreateTime,beginTime,endTime).oneOpt();
|
|
if(optFile.isPresent()){
|
//确实下发过，比对总段数是否相同
|
FileSendRecord sendFile = optFile.get();
|
ProgramNameVO sendProgNameVO = ProgramFileNameParser.parseProgramName(sendFile.getName());
|
if(progNameVO.getSegmentCount() != sendProgNameVO.getSegmentCount()){
|
//段数不匹配
|
machineFile.setExceptionType(MachineFile.EXCEPTION_NOT_MATCH_SEND);
|
}else{
|
machineFile.setExceptionType(MachineFile.EXCEPTION_OK);
|
}
|
}
|
}
|
}
|
}
|
|
machineFileService.updateById(machineFile);
|
}
|
}
|
}
|
