From 1911be8941e5fe2705c2c56e74e52bd426468793 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期三, 06 八月 2025 15:26:34 +0800
Subject: [PATCH] 固化流程,建立node;

---
 blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExportDNCService.java |   74 +++++++++++++++++++------------------
 1 files changed, 38 insertions(+), 36 deletions(-)

diff --git a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExportDNCService.java b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExportDNCService.java
index 91342ea..81f1daf 100644
--- a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExportDNCService.java
+++ b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExportDNCService.java
@@ -15,6 +15,8 @@
 import org.springblade.core.mp.support.Query;
 import org.springblade.core.oss.OssTemplate;
 import org.springblade.core.tool.utils.Func;
+import org.springblade.mdm.basesetting.machine.MachineService;
+import org.springblade.mdm.basesetting.machine.entity.Machine;
 import org.springblade.mdm.flow.entity.ApproveRecord;
 import org.springblade.mdm.flow.entity.FlowProgramFile;
 import org.springblade.mdm.flow.service.ApproveRecordService;
@@ -25,6 +27,7 @@
 import org.springblade.mdm.program.entity.NcProgramExchange;
 import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
 import org.springblade.mdm.program.vo.DncSendBackData;
+import org.springblade.mdm.utils.FileContentUtil;
 import org.springframework.stereotype.Service;
 
 import java.io.*;
@@ -44,24 +47,17 @@
 @AllArgsConstructor
 public class NcProgramExportDNCService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {
 
-	private final NcProgramService progService;
+	//private final NcProgramService progService;
+	private final MachineService machineService;
 	private final NcProgramApprovedService approvedService;
 	private final ApproveRecordService approveRecordService;
 	private final NcNodeService ncNodeService;
 	private final OssTemplate ossTemplate;
 	private final FlowProgramFileService flowProgramFileService;
-	public static final String PROGRAM_JSON_FILE = "exp_mdm_nc_program.json";
+	//public static final String PROGRAM_JSON_FILE = "exp_mdm_nc_program.json";
 	public static final String NODE_JSON_FILE = "exp_mdm_nc_node.json";
 	public static final String APPROVE_RECORD_JSON_FILE = "exp_mdm_approve_record.json";
 
-	/**
-	 * 鏄惁鍘嬬缉鍖呭唴鐨勬暟鎹枃浠�
-	 * @param filename 鏂囦欢鍚嶇О
-	 * @return
-	 */
-	public static boolean isDataFile(String filename){
-		return StringUtils.equals(filename, PROGRAM_JSON_FILE) || StringUtils.equals(filename, NODE_JSON_FILE) || StringUtils.equals(filename, APPROVE_RECORD_JSON_FILE);
-	}
 	/**
 	 * 鍒嗛〉鏌ヨ
 	 * @param query 鏌ヨ鍙傛暟
@@ -88,14 +84,8 @@
 
 			for (Long approvedId : approvedIdArr) {
 				NcProgramApproved approved = approvedService.getById(approvedId);
-				programPackageNodeIdList.add(approved.getNodeId());
+				programPackageNodeIdList.add(approved.getNcNodeId());
 
-				/*
-				String filename = prog.getOssName();
-				InputStream inputStream = ossTemplate.statFileStream(filename);
-
-				addInputStreamToZip(zipOut, inputStream, prog.getName());
-				*/
 				addProgramPackageToZip(zipOut,approved);
 			}
 			addDataJson(zipOut, programPackageNodeIdList);
@@ -116,33 +106,36 @@
 		zipOut.putNextEntry(zipEntry);
 		zipOut.closeEntry();
 
-		List<NcNode> programNodes = ncNodeService.lambdaQuery().eq(NcNode::getParentId, approved.getNodeId()).list();
+		List<NcNode> programNodes = ncNodeService.lambdaQuery().eq(NcNode::getParentId, approved.getNcNodeId()).list();
 
 		FlowProgramFile programFile;
+
 		for (NcNode node : programNodes) {
 			String filePathInZip = packageFolder + node.getName();
-			ZipEntry fileEntry = new ZipEntry(filePathInZip);
-
-			zipOut.putNextEntry(fileEntry);
 			programFile = this.flowProgramFileService.getById(node.getFlowProgramFileId());
 
 			InputStream inputStream = ossTemplate.statFileStream(programFile.getOssName());
-			byte[] buffer = new byte[1024];
-			int length;
-			while ((length = inputStream.read(buffer)) >= 0) {
-				zipOut.write(buffer, 0, length);
-			}
-			zipOut.closeEntry();
+			Machine machine = machineService.getByCode(node.getMachineCode());
+			InputStream addedIns = addSendPath(inputStream,machine.getProgSendDir());
+			this.addInputStreamToZip(zipOut,addedIns,filePathInZip);
 		}
 
 	}
 
 	/**
+	 * 鍔犲叆鏈哄簥涓嬪彂璺緞
+	 * @param inputStream
+	 * @return
+	 */
+	InputStream addSendPath(InputStream inputStream,String machineSendPath) throws IOException {
+		return FileContentUtil.insertLine(inputStream,1,machineSendPath);
+
+	}
+	/**
 	 * 瀵煎叆鏁版嵁鏂囦欢
 	 * @param zipOut
 	 */
 	void addDataJson(ZipOutputStream zipOut, List<Long> programPackageNodeIdList) throws IOException {
-		//addProgramDataJson(zipOut, programPackageNodeIdList);
 		addNodeDataJson(zipOut, programPackageNodeIdList);
 		addApproveRecordDataJson(zipOut, programPackageNodeIdList);
 	}
@@ -163,7 +156,7 @@
 			allNodeIds.add(node.getId());
 		}
 		//鍔犲叆绋嬪簭鍖呬笅绾х殑绋嬪簭鑺傜偣
-		List<NcNode> programNodes = ncNodeService.lambdaQuery().in(NcNode::getParentId, pkgNodes.stream().map(NcNode::getId)).list();
+		List<NcNode> programNodes = ncNodeService.lambdaQuery().in(NcNode::getParentId, pkgNodes.stream().map(NcNode::getId).toList()).list();
 
 		allNodeIds.addAll(programNodes.stream().map(NcNode::getId).toList());
 		List<Long>  distinctIds = allNodeIds.stream().distinct().toList();
@@ -175,7 +168,7 @@
 			addSuperProperties(recObj,node);
 			jsonArray.add(recObj);
 		}
-		addInputStreamToZip(zipOut,new ByteArrayInputStream(jsonArray.toJSONString().getBytes(StandardCharsets.UTF_8)),PROGRAM_JSON_FILE);
+		addInputStreamToZip(zipOut,new ByteArrayInputStream(jsonArray.toJSONString().getBytes(StandardCharsets.UTF_8)),NODE_JSON_FILE);
 
 	}
 
@@ -199,6 +192,7 @@
 
 		recObj.put("machineCode", node.getMachineCode());
 		recObj.put("flowProgramFileId", node.getFlowProgramFileId());
+		recObj.put("processInstanceId", node.getProcessInstanceId());
 		return recObj;
 	}
 
@@ -208,10 +202,12 @@
 	 * @param programPackageNodeIdList 绋嬪簭鍖呭悕鐨刬d鍒楄〃
 	 */
 	void addApproveRecordDataJson(ZipOutputStream zipOut, List<Long> programPackageNodeIdList) throws IOException {
-		List<String> instanceIds = this.ncNodeService.lambdaQuery().in(NcNode::getProcessInstanceId, programPackageNodeIdList)
+		/*
+		List<String> instanceIds = this.ncNodeService.lambdaQuery().in(NcNode::getId, programPackageNodeIdList)
 			.list().stream().map(NcNode::getProcessInstanceId).toList();
-
-		List<ApproveRecord> records = approveRecordService.lambdaQuery().in(ApproveRecord::getProcessInstanceId, instanceIds).list();
+*/
+		//List<ApproveRecord> records = approveRecordService.lambdaQuery().in(ApproveRecord::getProcessInstanceId, instanceIds).list();
+		List<ApproveRecord> records = approveRecordService.lambdaQuery().in(ApproveRecord::getNcNodeId, programPackageNodeIdList).list();
 
 		JSONArray jsonArray = new JSONArray();
 		for(ApproveRecord record : records){
@@ -223,9 +219,8 @@
 			recObj.put("operateTime",record.getOperateTime());
 			recObj.put("operateResult",record.getOperateResult());
 			recObj.put("taskName",record.getTaskName());
-			recObj.put("ncProgramId",record.getNcProgramId());
 			recObj.put("processInstanceId",record.getProcessInstanceId());
-
+			recObj.put("ncNodeId",record.getNcNodeId());
 			addSuperProperties(recObj,record);
 
 			jsonArray.add(recObj);
@@ -244,7 +239,14 @@
 		recObj.put("createDept",entity.getCreateDept());
 	}
 
-	public void addInputStreamToZip(ZipOutputStream zipOut, InputStream inputStream, String entryName)
+	/**
+	 * 灏� 杈撳叆娴� 涓殑鍐呭鍐欏叆zip
+	 * @param zipOut zip杈撳嚭娴�
+	 * @param inputStream 杈撳叆娴�
+	 * @param entryName 鏂囦欢鍚�
+	 * @throws IOException
+	 */
+	void addInputStreamToZip(ZipOutputStream zipOut, InputStream inputStream, String entryName)
 		throws IOException {
 		// 鍒涘缓鏂扮殑 ZIP 鏉$洰
 		ZipEntry zipEntry = new ZipEntry(entryName);

--
Gitblit v1.9.3