package org.springblade.mdm.flow.task.cure;
|
|
import lombok.AllArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.IOUtils;
|
import org.checkerframework.checker.units.qual.N;
|
import org.flowable.engine.delegate.DelegateExecution;
|
import org.springblade.core.oss.OssTemplate;
|
import org.springblade.core.oss.model.BladeFile;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.utils.DateUtil;
|
import org.springblade.mdm.basesetting.machine.MachineService;
|
import org.springblade.mdm.basesetting.machine.entity.Machine;
|
import org.springblade.mdm.commons.contants.ParamConstants;
|
import org.springblade.mdm.flow.entity.FlowProgramFile;
|
import org.springblade.mdm.flow.service.FlowCommonService;
|
import org.springblade.mdm.flow.service.FlowProgramFileService;
|
import org.springblade.mdm.flow.service.FlowProgramProperties;
|
import org.springblade.mdm.program.entity.NcNode;
|
import org.springblade.mdm.program.service.NcNodeAutoCreateService;
|
import org.springblade.mdm.program.service.NcNodeService;
|
import org.springblade.mdm.program.service.ProgramAnnotationService;
|
import org.springblade.mdm.utils.FileContentUtil;
|
import org.springblade.system.feign.ISysClient;
|
import org.springblade.system.pojo.entity.Dict;
|
import org.springframework.stereotype.Component;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.time.LocalDate;
|
import java.util.Arrays;
|
import java.util.Date;
|
import java.util.List;
|
|
@Slf4j
|
@Component("cureFinishOperateTask")
|
@AllArgsConstructor
|
public class CureFinishOperateTask {
|
private final NcNodeService nodeService;
|
private final NcNodeAutoCreateService ncNodeAutoCreateService;
|
private final FlowCommonService flowCommonService;
|
private final FlowProgramFileService flowProgramFileService;
|
private final ISysClient sysClient;
|
private final OssTemplate ossTemplate;
|
private final ProgramAnnotationService programAnnotationService;
|
private final MachineService machineService;
|
|
public static final String GH = "GH";
|
|
|
/**
|
* 默认有效期间（月数）,2年
|
*/
|
private static final int DEFAULT_VALID_MONTH = 24;
|
private final NcNodeService ncNodeService;
|
|
/**
|
* 固化审批通过处理任务，
|
* @param execution 流程execution
|
*/
|
@Transactional
|
public void execute(DelegateExecution execution) throws IOException {
|
log.info("执行固化程序任务服务,流程实例id={}", execution.getProcessInstanceId());
|
//固化程序检查有效期，
|
dealWithNode(execution.getProcessInstanceId());
|
}
|
|
/**
|
* 处理node状态。
|
* @param processInstanceId 流程实例id
|
*/
|
void dealWithNode(String processInstanceId) throws IOException {
|
//程序包节点和下属程序节点(包括历史节点非最新版本的)从试切挪到固化下面
|
//程序包节点 设置未已固化
|
NcNode pkgNode = nodeService.lambdaQuery().eq(NcNode::getProcessInstanceId, processInstanceId).one();
|
if(pkgNode.isDeviationProgram()){
|
//偏离的程序直接锁定
|
pkgNode.setIsLocked(1);
|
nodeService.updateById(pkgNode);
|
}else {
|
//试切的
|
pkgNode.setIsCured(1);
|
//设置过期日期
|
pkgNode.upgradeVersionNUmber();
|
pkgNode.setExpireDate(calculateExpireDate());
|
nodeService.updateById(pkgNode);
|
|
List<NcNode> historyProgramPackageNodes = nodeService.getNodeHistory(pkgNode);
|
moveNodeToCuredTree(pkgNode, historyProgramPackageNodes, flowCommonService.getProgramProperties(processInstanceId));
|
}
|
}
|
|
/**
|
* 将节点和历史节点挪动到固化的同级节点下
|
* @param pkgNode 要挪动的节点
|
* @param historyProgramPackageNodes 程序包名 历史节点
|
*/
|
void moveNodeToCuredTree(NcNode pkgNode,List<NcNode> historyProgramPackageNodes, FlowProgramProperties programProperties) throws IOException {
|
//创建节点到机床级别.(固化树)
|
NcNode machineNode = ncNodeAutoCreateService.createNodeTreeToMachine(programProperties);
|
|
//机床下现在没有文件了，程序包名升级与机床同级了，改为找到机床同级程序包名更新了
|
this.nodeService.lambdaUpdate().likeRight(NcNode::getParentIds, machineNode.getParentIds())
|
.in(NcNode::getNodeType, Arrays.asList(NcNode.TYPE_PROGRAM_PACKAGE,NcNode.TYPE_PROGRAM_FILE))
|
.set(NcNode::getIsLastEdition,0).set(NcNode::getIsLocked,1).update();
|
|
for(NcNode hisPackageNode : historyProgramPackageNodes){
|
if(!hisPackageNode.getId().equals(pkgNode.getId())){
|
hisPackageNode.setIsLastEdition(0);
|
}
|
hisPackageNode.setParentId(machineNode.getParentId());
|
hisPackageNode.setParentIds(machineNode.getParentIds());//程序包与机床节点同级
|
|
//处理程序包下层的程序节点
|
List<NcNode> programNodes = nodeService.lambdaQuery().eq(NcNode::getParentId,hisPackageNode.getId()).list();
|
for(NcNode programNode : programNodes){
|
programNode.setIsLastEdition(0);
|
programNode.setParentIds(hisPackageNode.getParentIds()+","+hisPackageNode.getId());
|
}
|
nodeService.updateBatchById(programNodes);
|
}
|
nodeService.updateBatchById(historyProgramPackageNodes);
|
|
Machine machine = machineService.getByCode(pkgNode.getMachineCode());
|
List<Dict> annoDicts = programAnnotationService.getAnnotionList();
|
//新的流程文件，需要在包节点下新建，不能用老的
|
List<FlowProgramFile> files = flowProgramFileService.lambdaQuery().eq(FlowProgramFile::getProcessInstanceId,pkgNode.getProcessInstanceId()).list();
|
for(FlowProgramFile flowProgramFile : files){
|
NcNode newProgNode = new NcNode();
|
newProgNode.setName(flowProgramFile.getName());
|
newProgNode.setNodeType(NcNode.TYPE_PROGRAM_FILE);
|
newProgNode.setParentId(pkgNode.getId());
|
newProgNode.setParentIds(pkgNode.getParentIds()+","+pkgNode.getId());
|
newProgNode.setIsLastEdition(1);
|
newProgNode.setVersionNumber(pkgNode.getVersionNumber());
|
newProgNode.setFlowProgramFileId(flowProgramFile.getId());
|
newProgNode.setIsCured(1);
|
ncNodeService.save(newProgNode);
|
|
//修改文件内容更新注释，并更新文件数据的地址
|
///替换注释后总是不行啊
|
setGhAnnotation(flowProgramFile,machine.getMachineGroupCode(),annoDicts);
|
|
flowProgramFileService.updateById(flowProgramFile);
|
}
|
}
|
|
/**
|
* 设置固化注释,实现方式，获取文件字节下，修改后替换文件
|
* @param flowProgramFile oss 文件
|
*
|
*/
|
void setGhAnnotation(FlowProgramFile flowProgramFile,String machineGroup,List<Dict> annoDicts) throws IOException {
|
String ossName = flowProgramFile.getOssName();
|
//
|
try(InputStream ins = ossTemplate.statFileStream(ossName);){
|
byte[] bytes = IOUtils.toByteArray(ins);
|
ByteArrayInputStream byteInputStream = new ByteArrayInputStream(bytes);
|
|
String annoTxt = programAnnotationService.generateAnnotation("GH",machineGroup,annoDicts);
|
int statusLineIndex = 2;
|
String line2 = FileContentUtil.readLineAt(byteInputStream,statusLineIndex);//第三行应该是状态注释
|
byteInputStream.reset();
|
InputStream finishedStream;
|
if(programAnnotationService.isAnnotation(line2,machineGroup,annoDicts)){
|
finishedStream = FileContentUtil.replaceAtLine(byteInputStream,statusLineIndex,annoTxt);
|
}else{
|
finishedStream = FileContentUtil.insertLine(byteInputStream,statusLineIndex,annoTxt);
|
}
|
try(finishedStream) {
|
finishedStream.reset();
|
BladeFile bfile = ossTemplate.putFile(flowProgramFile.getName(), finishedStream);
|
//替换原有的文件地址
|
flowProgramFile.setOssName(bfile.getName());
|
}
|
}
|
}
|
|
/**
|
* 计算有效期
|
* @return
|
*/
|
private Date calculateExpireDate() {
|
R<String> validMonResult = sysClient.getParamValue(ParamConstants.CURE_VALID_MONTH_KEY);
|
|
int month = DEFAULT_VALID_MONTH;
|
if(validMonResult.isSuccess() && validMonResult.getData()!=null){
|
month = Integer.parseInt(validMonResult.getData());
|
}
|
|
LocalDate now = LocalDate.now();
|
return DateUtil.toDate(now.plusMonths(month));
|
}
|
|
}
|
