From 88748094ae17ca3b6423c54d307f0d5c40c8f420 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期六, 02 八月 2025 14:44:13 +0800
Subject: [PATCH] 增加文件完整性校验

---
 blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/service/FlowProgramFileService.java |   38 ++++++++++++++++++++++++++++++++++++++
 1 files changed, 38 insertions(+), 0 deletions(-)

diff --git a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/service/FlowProgramFileService.java b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/service/FlowProgramFileService.java
index a6de868..085a2a3 100644
--- a/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/service/FlowProgramFileService.java
+++ b/blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/service/FlowProgramFileService.java
@@ -23,6 +23,7 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
+import java.util.List;
 
 /**
  * 娴佺▼绋嬪簭澶勭悊锛屽寘鎷笂浼狅紝鏌ヨ灞曠ず
@@ -169,5 +170,42 @@
 		return result;
 	}
 
+	/**
+	 * 楠岃瘉绋嬪簭鏂囦欢鐨勫畬鏁存�э紝浠呭湪缂栧埗鑺傜偣浣跨敤
+	 * @param processInstanceId 娴佺▼瀹炰緥id
+	 */
+	public void checkProgramFiles(String processInstanceId,boolean isPass) {
+		List<FlowProgramFile> flowPrograms = this.lambdaQuery().eq(FlowProgramFile::getProcessInstanceId, processInstanceId).orderByAsc(FlowProgramFile::getCreateTime).list();
+		if(isPass && flowPrograms.isEmpty()){
+			throw new ServiceException("璇蜂笂浼犵▼搴忔枃浠�");
+		}
+		int totalSeg = 0;
+		if(!flowPrograms.isEmpty()){
+			FlowProgramFile progFile = flowPrograms.get(0);
+			totalSeg = getProgramSegCount(progFile.getName());
+		}
 
+		if(totalSeg != flowPrograms.size()){
+			throw new ServiceException("搴斾笂浼�"+totalSeg+"娈电▼搴忥紝瀹為檯涓婁紶"+flowPrograms.size()+"娈�");
+		}
+
+	}
+
+	/**
+	 * 鏍规嵁鏂囦欢鍚嶈幏鍙栫▼搴忔�绘鏍�
+	 * @param filename 鏂囦欢鍚�
+	 * @return
+	 */
+	int getProgramSegCount(String filename){
+		int idx = filename.lastIndexOf(".");
+		String tempstr = filename.substring(0,idx);//鍘绘帀鎵╁睍鍚�
+		//System.out.println(tempstr);
+		idx = tempstr.lastIndexOf("-");
+		tempstr = tempstr.substring(0,idx);
+		//System.out.println(tempstr);
+		idx = tempstr.lastIndexOf("-");
+		tempstr = tempstr.substring(idx+1);
+		//System.out.println(tempstr);
+		return Func.toInt(tempstr);
+	}
 }

--
Gitblit v1.9.3