|
|
|
@ -1,5 +1,6 @@ |
|
|
|
package com.ccsens.defaultwbs.service; |
|
|
|
|
|
|
|
import cn.hutool.core.bean.BeanUtil; |
|
|
|
import cn.hutool.core.collection.CollectionUtil; |
|
|
|
import cn.hutool.core.lang.Snowflake; |
|
|
|
import cn.hutool.core.util.ObjectUtil; |
|
|
|
@ -96,7 +97,8 @@ public class ImportService implements IImportService { |
|
|
|
* @throws Exception 异常 |
|
|
|
*/ |
|
|
|
@Override |
|
|
|
public void importWbs(String path, Long userId,Long projectId) throws Exception { |
|
|
|
public ProjectVo.ProjectInfo importWbs(String path, Long userId,Long projectId) throws Exception { |
|
|
|
ProjectVo.ProjectInfo projectInfo = new ProjectVo.ProjectInfo(); |
|
|
|
//获取excel文件
|
|
|
|
InputStream is = new FileInputStream(path); |
|
|
|
XSSFWorkbook xssfWorkbook = new XSSFWorkbook(is); |
|
|
|
@ -108,13 +110,15 @@ public class ImportService implements IImportService { |
|
|
|
Map<String, Long> taskMap = new HashMap<>(); |
|
|
|
|
|
|
|
//读取文件
|
|
|
|
readExcel(xssfWorkbook, userId, roleMap, memberMap, taskMap,projectId); |
|
|
|
readExcel(xssfWorkbook, userId, roleMap, memberMap, taskMap,projectId,projectInfo); |
|
|
|
|
|
|
|
return projectInfo; |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
* 读取每个sheet |
|
|
|
*/ |
|
|
|
private void readExcel(XSSFWorkbook wb, Long userId, Map<String, Long> roleMap, Map<String, RoleDto.WbsMember> memberMap, Map<String, Long> taskMap,Long projectId) { |
|
|
|
private void readExcel(XSSFWorkbook wb, Long userId, Map<String, Long> roleMap, Map<String, RoleDto.WbsMember> memberMap, Map<String, Long> taskMap,Long projectId,ProjectVo.ProjectInfo projectInfo) { |
|
|
|
//获取wbsSheet
|
|
|
|
XSSFSheet wbsSheet = wb.getSheet(Constant.WbsExcel.WBS_SHEET); |
|
|
|
if (ObjectUtil.isNull(wbsSheet)) { |
|
|
|
@ -217,6 +221,9 @@ public class ImportService implements IImportService { |
|
|
|
|
|
|
|
saveProjectDto.setUserIdList(userIdSet); |
|
|
|
tall3FeignClient.saveProjectList(saveProjectDto); |
|
|
|
|
|
|
|
//获取项目信息
|
|
|
|
BeanUtil.copyProperties(saveProjectDto,projectInfo); |
|
|
|
} |
|
|
|
|
|
|
|
/** |
|
|
|
|