lbqms.git - Gitblit

			@@ -4,6 +4,7 @@
			import cn.hutool.core.codec.Base64;
			import cn.hutool.core.collection.CollUtil;
			import cn.hutool.core.collection.CollectionUtil;
			import cn.hutool.core.io.FileUtil;
			import cn.hutool.core.io.IoUtil;
			import cn.hutool.core.util.ObjectUtil;
			import lombok.RequiredArgsConstructor;
			@@ -28,6 +29,7 @@
			import org.flowable.engine.repository.*;
			import org.flowable.task.api.history.HistoricTaskInstance;
			import org.springframework.stereotype.Service;
			import org.springframework.transaction.annotation.Transactional;
			import org.springframework.web.multipart.MultipartFile;

			import java.io.IOException;
			@@ -36,6 +38,7 @@
			import java.util.ArrayList;
			import java.util.List;
			import java.util.stream.Collectors;
			import java.util.zip.ZipEntry;
			import java.util.zip.ZipInputStream;

			/**
			@@ -265,59 +268,69 @@
			* @param file 文件
			* @param categoryCode 分类
			*/
			@SneakyThrows
			@Override
			public boolean deployByFile(MultipartFile file, String categoryCode) {
			try {
			WfCategory wfCategory = wfCategoryService.queryByCategoryCode(categoryCode);
			if (wfCategory == null) {
			throw new ServiceException("流程分类不存在");
			}
			// 文件名 = 流程名称-流程key
			String filename = file.getOriginalFilename();
			assert filename != null;
			String[] splitFilename = filename.substring(0, filename.lastIndexOf(".")).split("-");
			if (splitFilename.length < 2) {
			throw new ServiceException("流程分类不能为空(文件名 = 流程名称-流程key)");
			}
			//流程名称
			String processName = splitFilename[0];
			//流程key
			String processKey = splitFilename[1];
			// 文件后缀名
			String suffix = filename.substring(filename.lastIndexOf(".") + 1).toUpperCase();
			InputStream inputStream = file.getInputStream();
			Deployment deployment;
			if (FlowConstant.ZIP.equals(suffix)) {
			DeploymentBuilder builder = repositoryService.createDeployment();
			deployment = builder.addZipInputStream(new ZipInputStream(inputStream))
			.tenantId(TenantHelper.getTenantId())
			.name(processName).key(processKey).category(categoryCode).deploy();
			} else {
			String[] list = ResourceNameUtil.BPMN_RESOURCE_SUFFIXES;
			boolean flag = false;
			for (String str : list) {
			if (filename.contains(str)) {
			flag = true;
			break;
			}
			}
			if (flag) {
			@Transactional(rollbackFor = Exception.class)
			public void deployByFile(MultipartFile file, String categoryCode) {

			WfCategory wfCategory = wfCategoryService.queryByCategoryCode(categoryCode);
			if (wfCategory == null) {
			throw new ServiceException("流程分类不存在");
			}
			// 文件后缀名
			String suffix = FileUtil.extName(file.getOriginalFilename());
			InputStream inputStream = file.getInputStream();
			if (FlowConstant.ZIP.equalsIgnoreCase(suffix)) {
			ZipInputStream zipInputStream = null;
			try {
			zipInputStream = new ZipInputStream(inputStream);
			ZipEntry zipEntry;
			while ((zipEntry = zipInputStream.getNextEntry()) != null) {
			String filename = zipEntry.getName();
			String[] splitFilename = filename.substring(0, filename.lastIndexOf(".")).split("-");
			//流程名称
			String processName = splitFilename[0];
			//流程key
			String processKey = splitFilename[1];
			DeploymentBuilder builder = repositoryService.createDeployment();
			deployment = builder.addInputStream(filename, inputStream)
			Deployment deployment = builder.addInputStream(filename, zipInputStream)
			.tenantId(TenantHelper.getTenantId())
			.name(processName).key(processKey).category(categoryCode).deploy();
			} else {
			throw new ServiceException("文件类型上传错误！");
			ProcessDefinition definition = QueryUtils.definitionQuery().deploymentId(deployment.getId()).singleResult();
			repositoryService.setProcessDefinitionCategory(definition.getId(), categoryCode);
			zipInputStream.closeEntry();
			}
			} catch (IOException e) {
			throw new RuntimeException(e);
			} finally {
			if (zipInputStream != null) {
			zipInputStream.close();
			}
			}
			// 更新分类
			ProcessDefinition definition = QueryUtils.definitionQuery().deploymentId(deployment.getId()).singleResult();
			repositoryService.setProcessDefinitionCategory(definition.getId(), categoryCode);

			return true;
			} catch (IOException e) {
			e.printStackTrace();
			throw new ServiceException("部署失败" + e.getMessage());
			} else {
			String originalFilename = file.getOriginalFilename();
			String bpmnResourceSuffix = ResourceNameUtil.BPMN_RESOURCE_SUFFIXES[0];
			if (originalFilename.contains(bpmnResourceSuffix)) {
			// 文件名 = 流程名称-流程key
			String[] splitFilename = originalFilename.substring(0, originalFilename.lastIndexOf(".")).split("-");
			if (splitFilename.length < 2) {
			throw new ServiceException("文件名 = 流程名称-流程KEY");
			}
			//流程名称
			String processName = splitFilename[0];
			//流程key
			String processKey = splitFilename[1];
			DeploymentBuilder builder = repositoryService.createDeployment();
			Deployment deployment = builder.addInputStream(originalFilename, inputStream)
			.tenantId(TenantHelper.getTenantId())
			.name(processName).key(processKey).category(categoryCode).deploy();
			// 更新分类
			ProcessDefinition definition = QueryUtils.definitionQuery().deploymentId(deployment.getId()).singleResult();
			repositoryService.setProcessDefinitionCategory(definition.getId(), categoryCode);
			} else {
			throw new ServiceException("文件类型上传错误！");
			}
			}

			}
			}