新增和编辑-处理相关文件和审批证据

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/project/service/impl/ProblemManagementEOServiceImpl.java

@@ -1,12 +1,19 @@
 package com.jero.modules.project.service.impl;
 
+import com.jero.modules.oss.entity.OSSFile;
+import com.jero.modules.oss.service.IOSSFileService;
 import com.jero.modules.project.entity.ProblemManagementEO;
 import com.jero.modules.project.mapper.ProblemManagementEOMapper;
 import com.jero.modules.project.service.IProblemManagementEOService;
 import com.jero.modules.system.util.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Date;
+import java.util.UUID;
+
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 
 /**
@@ -18,6 +25,10 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 @Service
 public class ProblemManagementEOServiceImpl extends ServiceImpl<ProblemManagementEOMapper, ProblemManagementEO> implements IProblemManagementEOService {
 
+
+	@Autowired
+	private IOSSFileService iOSSFileService;
+
 	/**
 	 * 保存
 	 *
@@ -28,12 +39,45 @@ public class ProblemManagementEOServiceImpl extends ServiceImpl<ProblemManagemen
     public void add(ProblemManagementEO problemManagementEO) {
 		//相关文件地址链接，进度追踪，审批证据链接
 		setParam(problemManagementEO);
+
+		//处理相关文件和审批证据
+		fileAndApprovalEvidence(problemManagementEO);
+
 		Date now = new Date();
         problemManagementEO.setCreateTime(now);
         problemManagementEO.setUpdateTime(now);
         save(problemManagementEO);
     }
 
+	private void fileAndApprovalEvidence(ProblemManagementEO problemManagementEO) {
+		//处理相关文件
+		if(StringUtils.isNotBlank(problemManagementEO.getFile())){
+			String connectId = UUID.randomUUID().toString().replace("-", "");
+			String fileIds = problemManagementEO.getFile();
+			List<OSSFile> oSSFileList = new ArrayList<>();
+			updateFileInfo(fileIds, oSSFileList,connectId);
+			problemManagementEO.setFile(connectId);
+		}
+		//处理审批证据
+		if(StringUtils.isNotBlank(problemManagementEO.getApprovalEvidence())){
+			String connectId = UUID.randomUUID().toString().replace("-", "");
+			String fileIds = problemManagementEO.getApprovalEvidence();
+			List<OSSFile> oSSFileList = new ArrayList<>();
+			updateFileInfo(fileIds, oSSFileList,connectId);
+			problemManagementEO.setApprovalEvidence(connectId);
+		}
+	}
+
+	private void updateFileInfo(String fileIds, List<OSSFile> oSSFileList,String connectId) {
+		for (String fileId : fileIds.split(",")) {
+			OSSFile ossFile = new OSSFile();
+			ossFile.setId(fileId);
+			ossFile.setConnectId(connectId);
+			oSSFileList.add(ossFile);
+		}
+		iOSSFileService.updateFileInfo(oSSFileList);
+	}
+
 	private void setParam(ProblemManagementEO problemManagementEO) {
 		if(!problemManagementEO.getFileLinkList().isEmpty()){
 			problemManagementEO.setFileLink(StringUtils.join(problemManagementEO.getFileLinkList(),","));
@@ -56,6 +100,10 @@ public class ProblemManagementEOServiceImpl extends ServiceImpl<ProblemManagemen
     public void editById(ProblemManagementEO problemManagementEO) {
 		//相关文件地址链接，进度追踪，审批证据链接
 		setParam(problemManagementEO);
+
+		//处理相关文件和审批证据
+		fileAndApprovalEvidence(problemManagementEO);
+
     	Date now = new Date();
         problemManagementEO.setUpdateTime(now);
     	saveOrUpdate(problemManagementEO);