From 87895ab581b2f44e3492e4395dce5fc7e8341363 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期六, 16 八月 2025 17:00:48 +0800
Subject: [PATCH] 替换注化

---
 blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/task/cure/CureFinishOperateTask.java |  165 +++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 141 insertions(+), 24 deletions(-)

diff --git a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/task/cure/CureFinishOperateTask.java b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/task/cure/CureFinishOperateTask.java
index 2d2ece0..a3152e3 100644
--- a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/task/cure/CureFinishOperateTask.java
+++ b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/task/cure/CureFinishOperateTask.java
@@ -2,15 +2,36 @@
 
 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.entity.NcProgram;
 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
@@ -20,61 +41,157 @@
 	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
 	 */
-	public void execute(DelegateExecution execution) {
+	@Transactional
+	public void execute(DelegateExecution execution) throws IOException {
 		log.info("鎵ц鍥哄寲绋嬪簭浠诲姟鏈嶅姟,娴佺▼瀹炰緥id={}", execution.getProcessInstanceId());
 		//鍥哄寲绋嬪簭妫�鏌ユ湁鏁堟湡锛�
-		//1.灏嗘祦绋嬭缃�
 		dealWithNode(execution.getProcessInstanceId());
-
 	}
 
 	/**
 	 * 澶勭悊node鐘舵�併��
 	 * @param processInstanceId 娴佺▼瀹炰緥id
 	 */
-	void dealWithNode(String processInstanceId) {
+	void dealWithNode(String processInstanceId) throws IOException {
 		//绋嬪簭鍖呰妭鐐瑰拰涓嬪睘绋嬪簭鑺傜偣(鍖呮嫭鍘嗗彶鑺傜偣闈炴渶鏂扮増鏈殑)浠庤瘯鍒囨尓鍒板浐鍖栦笅闈�
 		//绋嬪簭鍖呰妭鐐� 璁剧疆鏈凡鍥哄寲
 		NcNode pkgNode =  nodeService.lambdaQuery().eq(NcNode::getProcessInstanceId, processInstanceId).one();
-		pkgNode.setIsCured(1);
+		if(pkgNode.isDeviationProgram()){
+			//鍋忕鐨勭▼搴忕洿鎺ラ攣瀹�
+			pkgNode.setIsLocked(1);
+			nodeService.updateById(pkgNode);
+		}else {
+			//璇曞垏鐨�
+			pkgNode.setIsCured(1);
+			//璁剧疆杩囨湡鏃ユ湡
+			pkgNode.upgradeVersionNUmber();
+			pkgNode.setExpireDate(calculateExpireDate());
+			nodeService.updateById(pkgNode);
 
-		List<NcNode> historyNodes = nodeService.getNodeHistorys(pkgNode);
-		moveNodeToCuredTree(pkgNode,historyNodes,flowCommonService.getProgramProperties(processInstanceId));
-
+			List<NcNode> historyProgramPackageNodes = nodeService.getNodeHistory(pkgNode);
+			moveNodeToCuredTree(pkgNode, historyProgramPackageNodes, flowCommonService.getProgramProperties(processInstanceId));
+		}
 	}
 
 	/**
 	 * 灏嗚妭鐐瑰拰鍘嗗彶鑺傜偣鎸姩鍒板浐鍖栫殑鍚岀骇鑺傜偣涓�
-	 * @param historyNodes
+	 * @param pkgNode 瑕佹尓鍔ㄧ殑鑺傜偣
+	 * @param historyProgramPackageNodes 绋嬪簭鍖呭悕 鍘嗗彶鑺傜偣
 	 */
-	void moveNodeToCuredTree(NcNode pkgNode,List<NcNode> historyNodes, FlowProgramProperties programProperties) {
-		NcNode node = historyNodes.get(0);
+	void moveNodeToCuredTree(NcNode pkgNode,List<NcNode> historyProgramPackageNodes, FlowProgramProperties programProperties) throws IOException {
 		//鍒涘缓鑺傜偣鍒版満搴婄骇鍒�.(鍥哄寲鏍�)
 		NcNode machineNode = ncNodeAutoCreateService.createNodeTreeToMachine(programProperties);
 
-		//绉诲姩鍒板浐鍖栨爲涓嬬殑鏈哄簥鑺傜偣涓嬪眰
-		String newParentIds = machineNode.getParentIds()+","+machineNode.getId();
-		for(NcNode hisNode : historyNodes){
-			if(!hisNode.getId().equals(pkgNode.getId())){
-				hisNode.setIsLastEdition(0);
+		//鏈哄簥涓嬬幇鍦ㄦ病鏈夋枃浠朵簡锛岀▼搴忓寘鍚嶅崌绾т笌鏈哄簥鍚岀骇浜嗭紝鏀逛负鎵惧埌鏈哄簥鍚岀骇绋嬪簭鍖呭悕鏇存柊浜�
+		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);
 			}
-			hisNode.setParentId(machineNode.getParentId());
-			hisNode.setParentIds(newParentIds);
+			hisPackageNode.setParentId(machineNode.getParentId());
+			hisPackageNode.setParentIds(machineNode.getParentIds());//绋嬪簭鍖呬笌鏈哄簥鑺傜偣鍚岀骇
 
 			//澶勭悊绋嬪簭鍖呬笅灞傜殑绋嬪簭鑺傜偣
-			List<NcNode> pragramNodes = nodeService.lambdaQuery().eq(NcNode::getParentId,hisNode.getId()).list();
-			for(NcNode pragramNode : pragramNodes){
-				pragramNode.setParentIds(hisNode.getParentIds()+","+hisNode.getId());
+			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(pragramNodes);
+			nodeService.updateBatchById(programNodes);
 		}
-		nodeService.updateBatchById(historyNodes);
+		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));
+	}
+
 }

--
Gitblit v1.9.3