From ef4a26bca3552aed5865e5ef3ef2804b8509d31b Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期二, 08 七月 2025 13:57:06 +0800
Subject: [PATCH] 增加param调用接口

---
 blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExchangeService.java |  173 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 161 insertions(+), 12 deletions(-)

diff --git a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExchangeService.java b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExchangeService.java
index 7ead897..e2e2241 100644
--- a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExchangeService.java
+++ b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/program/service/NcProgramExchangeService.java
@@ -12,13 +12,20 @@
 import org.apache.commons.compress.archivers.zip.ZipFile;
 import org.apache.commons.compress.utils.SeekableInMemoryByteChannel;
 import org.apache.commons.io.IOUtils;
+import org.springblade.core.mp.base.BizEntity;
 import org.springblade.core.mp.base.BizServiceImpl;
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.mp.support.Query;
+import org.springblade.core.oss.OssTemplate;
+import org.springblade.core.oss.model.BladeFile;
+import org.springblade.core.redis.cache.BladeRedis;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.core.tool.utils.FileUtil;
 import org.springblade.core.tool.utils.Func;
 import org.springblade.core.tool.utils.IoUtil;
+import org.springblade.mdm.flow.entity.ApproveRecord;
 import org.springblade.mdm.flow.service.CureFlowService;
+import org.springblade.mdm.program.entity.NcNode;
 import org.springblade.mdm.program.entity.NcProgram;
 import org.springblade.mdm.program.entity.NcProgramExchange;
 import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
@@ -32,6 +39,7 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.time.Duration;
 import java.time.LocalDateTime;
 import java.util.*;
 
@@ -46,6 +54,13 @@
 public class NcProgramExchangeService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {
 	private final CureFlowService cureFlowService;
 	private final NcProgramService ncProgramService;
+	private final NcNodeService ncNodeService;
+	private final OssTemplate ossTemplate;
+	private final BladeRedis bladeRedis;
+
+	private String getFileKey(){
+		return "dncexpfile-"+ AuthUtil.getUserId();
+	}
 	/**
 	 * dnc鍥炰紶鏂囦欢涓婁紶锛堣В鏋愬悗淇濆瓨鍏pload琛級
 	 * @param file DNC鍥炰紶鏂囦欢
@@ -54,13 +69,15 @@
 	public List<DncSendBackData> dncSendBackUpload(MultipartFile file) {
 		List<DncSendBackData> list;
 		try {
-			//String fileName = file.getOriginalFilename();
+			BladeFile bfile = ossTemplate.putFile(file);//涓婁紶锛屼緵鍚庣画鍏ュ簱浣跨敤
+			//璁剧疆涓�涓紦瀛橈紝2灏忔椂杩囨湡
+			bladeRedis.setEx(getFileKey(),bfile.getName(), Duration.ofHours(2));
+
 			//InputStream zipFileInputStream = FileExchangeUtil.convertFileToZip(file.getInputStream());
 			InputStream zipFileInputStream = file.getInputStream();//test
 
 			byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
 			list = parseDncZipFromByteArray(bytes);
-
 
 		} catch (IOException e) {
 			log.error("涓婁紶dnc鍥炰紶鏂囦欢澶辫触",e);
@@ -138,11 +155,6 @@
 						}
 
 					}
-					System.out.println("鏂囦欢鍚�: " + entry.getName());
-					System.out.println("澶у皬: " + entry.getSize());
-
-					// 璇诲彇鏂囦欢鍐呭鍒板瓧鑺傛暟缁�
-					ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
 
 				}
 
@@ -164,13 +176,13 @@
 	 * @param ids id鍒楄〃閫楀彿鍒嗛殧
 	 * @return
 	 */
-	public void dncFileAccept(String ids) {
+	public void dncFileAccept(String ids) throws IOException {
 		List<Long> idList = Func.toLongList(ids);
-		List<NcProgram> progList = ncProgramService.listByIds(idList);
+		//
 		NcProgramExchange exchange;
-		//NcProgram program;
-		//NcNode programNode;
-
+		String pkgFileName = bladeRedis.get(getFileKey());
+		updateProgramData(pkgFileName,idList);
+		List<NcProgram> progList = ncProgramService.listByIds(idList);
 		for(NcProgram prog:progList){
 			exchange = new NcProgramExchange();
 			exchange.setName(prog.getName());
@@ -182,6 +194,143 @@
 		}
 
 		cureFlowService.startCure(progList);
+	}
 
+	void updateProgramData(String pkgFileName,List<Long> idList) throws IOException {
+		InputStream inputStream = this.ossTemplate.statFileStream(pkgFileName);
+		byte[] bytes = FileUtil.copyToByteArray(inputStream);
+
+		List<NcNode> nodeList = new ArrayList<>();
+		List<NcProgram> progList = new ArrayList<>();
+
+		try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(bytes);
+			 ZipFile zipFile = new ZipFile(channel)) {
+
+			ZipArchiveEntry entry;
+			Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
+			while (entries.hasMoreElements()) {
+				entry = entries.nextElement();
+
+				String entryName = entry.getName();
+
+				if (entry.isDirectory()) {
+					continue;
+				}
+				if(entryName.equals(NcProgramExportDNCService.PROGRAM_JSON_FILE)){
+
+					try (InputStream insJson = zipFile.getInputStream(entry)) {
+						String jsonStr = IoUtil.readToString(insJson);
+
+						JSONArray jsonArray = JSONArray.parseArray(jsonStr);
+						for(int i=0;i<jsonArray.size();i++){
+							JSONObject jsonObject = jsonArray.getJSONObject(i);
+							NcProgram program = new NcProgram();
+							program.setId(jsonObject.getLong("id"));
+							if(!idList.contains(program.getId())){//涓嶆槸閫夊畾鍏ュ簱鐨�
+								continue;
+							}
+							program.setName(jsonObject.getString("name"));
+
+							program.setCode(jsonObject.getString("code"));
+							program.setDescription(jsonObject.getString("description"));
+							program.setCategory(jsonObject.getString("category"));
+							program.setBindNcNodeId(jsonObject.getLong("bindNcNodeId"));
+							program.setIsLastEdition(jsonObject.getInteger("isLastEdition"));
+							program.setIsLocked(jsonObject.getInteger("isLocked"));
+							program.setIsTest(jsonObject.getInteger("isTest"));
+							program.setMachineCode(jsonObject.getString("machineCode"));
+							program.setNcNodeId(jsonObject.getLong("ncNodeId"));
+							program.setPartNo(jsonObject.getString("partNo"));
+							program.setProcessEdition(jsonObject.getString("processEdition"));
+
+							setBaseProperties(program,jsonObject);
+
+							progList.add(program);
+						}
+
+					}
+				}else if(entryName.equals(NcProgramExportDNCService.NODE_JSON_FILE)){
+					try (InputStream insJson = zipFile.getInputStream(entry)) {
+						String jsonStr = IoUtil.readToString(insJson);
+						JSONArray jsonArray = JSONArray.parseArray(jsonStr);
+						for (int i = 0; i < jsonArray.size(); i++) {
+							JSONObject jsonObject = jsonArray.getJSONObject(i);
+							NcNode node = new NcNode();
+							node.setId(jsonObject.getLong("id"));
+							node.setName(jsonObject.getString("name"));
+							node.setDescription(jsonObject.getString("description"));
+							node.setNodeType(jsonObject.getString("nodeType"));
+							node.setProcessName(jsonObject.getString("processName"));
+							node.setPartNo(jsonObject.getString("partNo"));
+							node.setMachineCode(jsonObject.getString("machineCode"));
+							node.setParentId(jsonObject.getLong("parentId"));
+							node.setParentIds(jsonObject.getString("parentIds"));
+							node.setIsCured(jsonObject.getInteger("isCured"));
+
+							setBaseProperties(node,jsonObject);
+
+							nodeList.add(node);
+						}
+					}
+				}
+
+			}
+
+		}
+
+		//涓婁紶鍘嬬缉鍖呭唴鐨勭▼搴忓苟鏇存柊绋嬪簭鏂囦欢鍦板潃
+		try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(bytes);
+			 ZipFile zipFile = new ZipFile(channel)) {
+
+			ZipArchiveEntry entry;
+			Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
+			while (entries.hasMoreElements()) {
+				entry = entries.nextElement();
+
+				String entryName = entry.getName();
+
+				if (!entry.isDirectory() && !NcProgramExportDNCService.isDataFile(entryName)) {
+
+					for(NcProgram prog:progList){
+						if(prog.getName().equals(entryName)){
+							try (InputStream ncFileStream = zipFile.getInputStream(entry)) {
+								BladeFile bfile = this.ossTemplate.putFile(prog.getName(),ncFileStream);
+								prog.setOssName(bfile.getName());
+								prog.setUrl(bfile.getLink());
+							}
+							break;
+						}
+					}
+
+				}
+			}
+		}
+
+		for(NcNode node:nodeList){
+			NcNode nodeTemp = this.ncNodeService.getById(node.getId());
+			if(nodeTemp == null){
+				ncNodeService.save(node);
+			}else{
+				ncNodeService.updateById(node);
+			}
+		}
+
+		for(NcProgram prog:progList){
+			NcProgram ncTemp = ncProgramService.getById(prog.getId());
+			if(ncTemp == null){
+				ncProgramService.save(prog);
+			}else{
+				ncProgramService.updateById(prog);
+			}
+		}
+
+	}
+
+	void setBaseProperties(BizEntity entity, JSONObject jsonObject){
+		entity.setCreateTime(jsonObject.getDate("createTime"));
+		entity.setUpdateTime(jsonObject.getDate("updateTime"));
+		entity.setStatus(jsonObject.getInteger("status"));
+		entity.setCreateUser(jsonObject.getLong("createUser"));
+		entity.setUpdateUser(jsonObject.getLong("updateUser"));
 	}
 }

--
Gitblit v1.9.3