From 9e28453dba6ecf93c6c2234ba38def7c41cf12f0 Mon Sep 17 00:00:00 2001 From: fuliqi <fuliqi@qq.com> Date: 星期一, 20 一月 2025 18:25:51 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java | 661 ++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 633 insertions(+), 28 deletions(-) diff --git a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java index 13afe3e..4d0d216 100644 --- a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java +++ b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java @@ -1,26 +1,74 @@ package com.ycl.service.impl; +import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.io.IORuntimeException; +import cn.hutool.core.util.CharsetUtil; +import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.StrUtil; +import cn.hutool.core.util.ZipUtil; +import com.alibaba.excel.EasyExcel; +import com.baomidou.mybatisplus.core.conditions.Wrapper; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ycl.common.annotation.DataScope; import com.ycl.common.base.Result; -import com.ycl.domain.entity.File; -import com.ycl.domain.entity.ProjectInfo; -import com.ycl.domain.form.ProjectInfoForm; +import com.ycl.common.config.SysConfig; +import com.ycl.common.constant.Constants; +import com.ycl.common.core.domain.BaseEntity; +import com.ycl.common.enums.business.FileTypeEnum; +import com.ycl.common.enums.business.ImportanceTypeEnum; +import com.ycl.common.enums.business.ProjectCategoryEnum; +import com.ycl.common.exception.base.BaseException; +import com.ycl.common.utils.CopyUtils; +import com.ycl.common.utils.DateUtils; +import com.ycl.common.utils.SecurityUtils; +import com.ycl.common.utils.StringUtils; +import com.ycl.common.utils.bean.BeanUtils; +import com.ycl.common.utils.excel.OutputExcelUtils; +import com.ycl.common.utils.file.FileUploadUtils; +import com.ycl.common.utils.file.FileUtils; +import com.ycl.common.utils.poi.ExcelUtil; +import com.ycl.controller.ProjectInfoController; +import com.ycl.domain.entity.*; +import com.ycl.domain.excel.ProjectExcelTemplate; +import com.ycl.domain.form.*; +import com.ycl.domain.query.ProjectExportQuery; import com.ycl.domain.query.ProjectInfoQuery; import com.ycl.domain.vo.*; import com.ycl.framework.utils.PageUtil; -import com.ycl.mapper.ProjectInfoMapper; +import com.ycl.listener.excel.ProjectImportListener; +import com.ycl.mapper.*; +import com.ycl.service.FileService; import com.ycl.service.ProjectInfoService; +import com.ycl.system.mapper.SysDeptMapper; import lombok.RequiredArgsConstructor; -import org.springframework.beans.BeanUtils; +import org.apache.commons.codec.Charsets; import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; import org.springframework.util.Assert; +import org.springframework.web.multipart.MultipartFile; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.List; -import java.util.Map; +import javax.servlet.ServletOutputStream; +import javax.servlet.http.HttpServletResponse; +import java.io.FileInputStream; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.math.BigDecimal; +import java.math.RoundingMode; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.StandardCopyOption; +import java.text.SimpleDateFormat; +import java.util.*; import java.util.stream.Collectors; +import java.util.stream.Stream; +import java.util.zip.ZipOutputStream; /** * 椤圭洰绠$悊鍩虹淇℃伅琛� 鏈嶅姟瀹炵幇绫� @@ -33,6 +81,18 @@ public class ProjectInfoServiceImpl extends ServiceImpl<ProjectInfoMapper, ProjectInfo> implements ProjectInfoService { private final ProjectInfoMapper projectInfoMapper; + private final ProjectInvestmentFundingMapper investmentFundingMapper; + private final ProjectInvestmentInfoMapper investmentInfoMapper; + private final ProjectInvestmentPolicyComplianceMapper policyMapper; + private final ProjectUnitRegistrationInfoMapper unitRegistrationInfoMapper; + private final FileService fileService; + private final FileMapper fileMapper; + private final PlanMapper planMapper; + private final ProjectInvestmentInfoServiceImpl projectInvestmentInfoServiceImpl; + private final ProjectInvestmentFundingServiceImpl projectInvestmentFundingServiceImpl; + private final ProjectInvestmentPolicyComplianceServiceImpl projectInvestmentPolicyComplianceServiceImpl; + private final ProjectUnitRegistrationInfoServiceImpl projectUnitRegistrationInfoServiceImpl; + private final SysDeptMapper sysDeptMapper; /** * 娣诲姞 @@ -41,13 +101,36 @@ * @return */ @Override + @Transactional(rollbackFor = Exception.class) public Result add(ProjectInfoForm form) { //娣诲姞鍩烘湰淇℃伅 ProjectInfo entity = ProjectInfoForm.getEntityByForm(form, null); -// baseMapper.insert(entity); + Long userId = SecurityUtils.getUserId(); + Long deptId = SecurityUtils.getDeptId(); + entity.setProjectOwnerUnit(deptId); + entity.setCreateBy(userId); + entity.setUpdateBy(userId); + baseMapper.insert(entity); + addPlan(entity.getId()); //娣诲姞璁″垝琛� //娣诲姞鏂囦欢 List<File> fileList = form.getFileList(); - return Result.ok("娣诲姞鎴愬姛"); + fileList.forEach(item -> { + item.setBusId(entity.getId()); + item.setType(FileTypeEnum.PROJECT_INFO); + }); + fileService.saveBatch(fileList); + return Result.ok("娣诲姞鎴愬姛").data(entity.getId()); + } + + public void addPlan(Long projectInfoId) { + Plan plan = new Plan(); + plan.setProjectInfoId(projectInfoId); + plan.setReportStatus(1); + plan.setMonthStatus(1); + plan.setSeasonStatus(1); + plan.setYearStatus(1); + plan.setDeleted(0); + planMapper.insert(plan); } /** @@ -57,13 +140,29 @@ * @return */ @Override + @Transactional(rollbackFor = Exception.class) public Result update(ProjectInfoForm form) { ProjectInfo entity = baseMapper.selectById(form.getId()); - // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊 Assert.notNull(entity, "璁板綍涓嶅瓨鍦�"); - BeanUtils.copyProperties(form, entity); + ProjectInfoForm.getEntityByForm(form, entity); + Long userId = SecurityUtils.getUserId(); + entity.setUpdateBy(userId); + //鏇存柊椤圭洰淇℃伅 baseMapper.updateById(entity); + List<File> fileList = form.getFileList(); + fileList.forEach(item -> { + item.setId(null); + item.setBusId(entity.getId()); + item.setType(FileTypeEnum.PROJECT_INFO); + }); + //鍒犻櫎鍘熸湁鏂囦欢 + QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>(); + fileQueryWrapper.eq("type", FileTypeEnum.PROJECT_INFO.getType()); + fileQueryWrapper.eq("bus_id", entity.getId()); + fileMapper.delete(fileQueryWrapper); + //鏇挎崲鎴愮幇鏈� + fileService.saveBatch(fileList); return Result.ok("淇敼鎴愬姛"); } @@ -86,7 +185,18 @@ * @return */ @Override - public Result removeById(String id) { + @Transactional(rollbackFor = Exception.class) + public Result removeById(Long id) { + investmentFundingMapper.delete(new QueryWrapper<ProjectInvestmentFunding>().eq("project_id", id)); + investmentInfoMapper.delete(new QueryWrapper<ProjectInvestmentInfo>().eq("project_id", id)); + policyMapper.delete(new QueryWrapper<ProjectInvestmentPolicyCompliance>().eq("project_id", id)); + unitRegistrationInfoMapper.delete(new QueryWrapper<ProjectUnitRegistrationInfo>().eq("project_id", id)); + + List<String> types = new ArrayList<>(); + types.add(FileTypeEnum.PROJECT_INFO.getType()); + types.add(FileTypeEnum.INVEST_POLICY.getType()); + types.add(FileTypeEnum.DOCUMENT_INFO.getType()); + fileMapper.delete(new QueryWrapper<File>().eq("bus_id", id).in("type", types)); baseMapper.deleteById(id); return Result.ok("鍒犻櫎鎴愬姛"); } @@ -99,9 +209,71 @@ */ @Override public Result page(ProjectInfoQuery query) { + if (query.getProjectStartTime() != null) { + query.setProjectStartTime(DateUtils.getDayStart(query.getProjectStartTime())); + } + if (query.getProjectEndTime() != null) { + query.setProjectEndTime(DateUtils.getDayEnd(query.getProjectEndTime())); + } + String projectCategory = query.getProjectCategory(); + //瀹炴柦闃舵鏈変袱涓�兼斁鍦╯ql澶勭悊 + if (ProjectCategoryEnum.RESERVE.getType().equals(projectCategory)) { + query.setProjectStatus(ProjectCategoryEnum.RESERVE.getStatus()); + query.setReserveOrPrevious(ProjectCategoryEnum.RESERVE.getCode()); + } else if (ProjectCategoryEnum.PREVIOUS.getType().equals(projectCategory)) { + query.setProjectStatus(ProjectCategoryEnum.PREVIOUS.getStatus()); + query.setReserveOrPrevious(ProjectCategoryEnum.PREVIOUS.getCode()); + } else if (ProjectCategoryEnum.FINISH.getType().equals(projectCategory)) { + query.setProjectStatus(ProjectCategoryEnum.FINISH.getStatus()); + } else if (ProjectCategoryEnum.EXCEPTION.getType().equals(projectCategory)) { + //TODO + //鍏堟煡鍑哄紓甯告祦绋嬫垨鑰呭紓甯歌繘搴︽垨鑰呭紓甯歌鍒掔殑projectId鍜屽紓甯哥绫� + //閫氳繃projectId鏌ュ嚭椤圭洰鏁版嵁 + //琛ュ厖鐩稿簲鐨勫紓甯告暟鎹紙寮傚父绉嶇被銆佸紓甯告祦绋嬭妭鐐圭瓑锛� + List<ProjectVO> list = new ArrayList<>(); + return Result.ok().data(list).total(0); + } + + Long userId = SecurityUtils.getUserId(); + String ancestors = sysDeptMapper.selectAncestors(userId); + String[] ancestorArr = ancestors.split(","); +// for (String ancestor : ancestorArr) { +// if ("101".equals(ancestor)) { // 涓氫富 +// query.setCreateBy(userId); +// query.setRole(0); +// } else if ("102".equals(ancestor)) { // 瀹℃壒 +// query.setUsedStatus(1); +// query.setRole(1); +// } +// } + IPage<ProjectInfoVO> page = PageUtil.getPage(query, ProjectInfoVO.class); - baseMapper.getPage(page, query); - return Result.ok().data(page.getRecords()).total(page.getTotal()); + baseMapper.getPage(query, page); + List<ProjectInfoVO> records = page.getRecords(); + List<ProjectVO> list = new ArrayList<>(); + records.forEach(vo -> { + ProjectInfoVO.transform(vo); + ProjectVO projectVO = new ProjectVO(); + copyToProjectVO(vo, projectVO); + //缈昏瘧椤圭洰闃舵 + String phase = ProjectCategoryEnum.getPhaseByProjectStatus(projectVO.getProjectStatus(), projectVO.getProcessId() != null); + projectVO.setProjectPhase(phase); + list.add(projectVO); + }); + return Result.ok().data(list).total(page.getTotal()); + } + + private void copyToProjectVO(ProjectInfoVO vo, ProjectVO projectVO) { + //蹇界暐null鍊肩殑澶嶅埗 + CopyUtils.copyNoNullProperties(vo, projectVO); + if (vo.getProjectInvestmentInfo() != null) + CopyUtils.copyNoNullProperties(vo.getProjectInvestmentInfo(), projectVO); + if (vo.getProjectUnitRegistrationInfo() != null) + CopyUtils.copyNoNullProperties(vo.getProjectUnitRegistrationInfo(), projectVO); + if (vo.getProjectInvestmentPolicyCompliance() != null) + CopyUtils.copyNoNullProperties(vo.getProjectInvestmentPolicyCompliance(), projectVO); + if (vo.getProjectInvestmentFunding() != null) + CopyUtils.copyNoNullProperties(vo.getProjectInvestmentFunding(), projectVO); } /** @@ -112,8 +284,14 @@ */ @Override public Result detail(Integer id) { - ProjectInfoVO vo = baseMapper.getById(id); - Assert.notNull(vo, "璁板綍涓嶅瓨鍦�"); + ProjectInfo entity = baseMapper.getById(id); + Assert.notNull(entity, "璁板綍涓嶅瓨鍦�"); + ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null); + QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>(); + fileQueryWrapper.eq("type", FileTypeEnum.PROJECT_INFO.getType()); + fileQueryWrapper.eq("bus_id", vo.getId()); + List<File> files = fileMapper.selectList(fileQueryWrapper); + vo.setFileList(files); return Result.ok().data(vo); } @@ -141,17 +319,80 @@ // {"type":"鐪侀噸鐐归」鐩�","count":0,"amount":"0.00","text":"鐪�"}]}} IndexCountVO indexCountVO = new IndexCountVO(); List<IndexProPhaseCountVO> proPhaseCountVO = new ArrayList<>(); - proPhaseCountVO.add(new IndexProPhaseCountVO("鍌ㄥ瑙勫垝闃舵", 0, "0.00", "鍌�")); - proPhaseCountVO.add(new IndexProPhaseCountVO("椤圭洰鍓嶆湡闃舵", 0, "0.00", "鏂�")); - proPhaseCountVO.add(new IndexProPhaseCountVO("瀹炴柦闃舵", 0, "0.00", "寤�")); - proPhaseCountVO.add(new IndexProPhaseCountVO("绔e伐鎶曠敤闃舵", 0, "0.00", "绔�")); - List<IndexImpTypeCountVO> impTypeCountVO = new ArrayList<>(); - impTypeCountVO.add(new IndexImpTypeCountVO("涓�鑸」鐩�", 0, "0.00", "鏅�")); - impTypeCountVO.add(new IndexImpTypeCountVO("鍘块噸鐐归」鐩�", 0, "0.00", "鍘�")); - impTypeCountVO.add(new IndexImpTypeCountVO("甯傞噸鐐归」鐩�", 0, "0.00", "甯�")); - impTypeCountVO.add(new IndexImpTypeCountVO("鐪侀噸鐐归」鐩�", 0, "0.00", "鐪�")); - indexCountVO.setImpTypeCountVO(impTypeCountVO); + List<ProjectVO> projectVOS = baseMapper.homeCount(new BaseEntity()); + List<ProjectVO> reserve = new ArrayList<>(); + List<ProjectVO> previous = new ArrayList<>(); + List<ProjectVO> implement = new ArrayList<>(); + List<ProjectVO> finish = new ArrayList<>(); + List<ProjectVO> normal = new ArrayList<>(); + List<ProjectVO> province = new ArrayList<>(); + List<ProjectVO> city = new ArrayList<>(); + List<ProjectVO> county = new ArrayList<>(); + projectVOS.forEach(item -> { + if (ProjectCategoryEnum.RESERVE.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) { + reserve.add(item); + } else if (ProjectCategoryEnum.PREVIOUS.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) { + previous.add(item); + } else if (ProjectCategoryEnum.IMPLEMENT.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) { + implement.add(item); + } else if (ProjectCategoryEnum.FINISH.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) { + finish.add(item); + } + + if (ImportanceTypeEnum.PROVINCIAL_KEY.getType().equals(item.getImportanceType())) { + province.add(item); + } else if (ImportanceTypeEnum.SUINING_KEY.getType().equals(item.getImportanceType())) { + city.add(item); + } else if (ImportanceTypeEnum.SHEHONG_KEY.getType().equals(item.getImportanceType())) { + county.add(item); + } else if (ImportanceTypeEnum.NORMAL.getType().equals(item.getImportanceType())) { + normal.add(item); + } + }); + //鍗曚綅鍏堥粯璁や负鍏� + proPhaseCountVO.add(new IndexProPhaseCountVO("鍌ㄥ瑙勫垝闃舵", reserve.size(), reserve.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鍌�")); + + proPhaseCountVO.add(new IndexProPhaseCountVO("椤圭洰鍓嶆湡闃舵", previous.size(), previous.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鏂�")); + + proPhaseCountVO.add(new IndexProPhaseCountVO("瀹炴柦闃舵", implement.size(), implement.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "寤�")); + + proPhaseCountVO.add(new IndexProPhaseCountVO("绔e伐鎶曠敤闃舵", finish.size(), finish.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "绔�")); + indexCountVO.setProPhaseCountVO(proPhaseCountVO); + + List<IndexImpTypeCountVO> impTypeCountVO = new ArrayList<>(); + impTypeCountVO.add(new IndexImpTypeCountVO("涓�鑸」鐩�", province.size(), province.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鏅�")); + + impTypeCountVO.add(new IndexImpTypeCountVO("鍘块噸鐐归」鐩�", city.size(), city.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鍘�")); + + impTypeCountVO.add(new IndexImpTypeCountVO("甯傞噸鐐归」鐩�", county.size(), county.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "甯�")); + + impTypeCountVO.add(new IndexImpTypeCountVO("鐪侀噸鐐归」鐩�", normal.size(), normal.stream(). + filter(item -> !StringUtils.isEmpty(item.getTotalInvestment())) + .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP)) + .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鐪�")); + indexCountVO.setImpTypeCountVO(impTypeCountVO); return indexCountVO; @@ -163,4 +404,368 @@ map.put("processExceptionProject", 0); return map; } + + @Override + public Result docDetail(Integer id) { + DocumentInfoForm documentInfoForm = new DocumentInfoForm(); + QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>(); + fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType()); + fileQueryWrapper.eq("bus_id", id); + List<File> files = fileMapper.selectList(fileQueryWrapper); + documentInfoForm.setFileList(files); + return Result.ok().data(documentInfoForm); + } + + @Override + public Result addDoc(DocumentInfoForm form) { + List<File> fileList = form.getFileList(); + fileList.forEach(item -> { + item.setId(null); + item.setBusId(form.getProjectId()); + item.setType(FileTypeEnum.DOCUMENT_INFO); + }); + //鍒犻櫎鍘熸湁鏂囦欢 + QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>(); + fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType()); + fileQueryWrapper.eq("bus_id", form.getProjectId()); + fileMapper.delete(fileQueryWrapper); + //鏇挎崲鎴愮幇鏈� + fileService.saveBatch(fileList); + return Result.ok(); + } + + @Override + public Result getManagerFlag(Integer recordId) { + ProjectInfo projectInfo = baseMapper.selectById(recordId); + // 鍒ゆ柇褰撳墠鐢ㄦ埛id鏄惁鍦ㄤ富绠″垪琛ㄤ腑 + String competentDepartment = projectInfo.getCompetentDepartment(); + if (StringUtils.isEmpty(competentDepartment)) { + return Result.ok().data(false); + } + List<String> list = Arrays.asList(competentDepartment.split(",")); + // 鑾峰緱褰撳墠鐢ㄦ埛id + Long userId = SecurityUtils.getUserId(); + if (list.contains(userId.toString())) { + return Result.ok().data(true); + } else { + return Result.ok().data(false); + } + } + + @Override + public void export(HttpServletResponse response, ProjectExportQuery query) throws IOException { + List<ProjectVO> data = baseMapper.selectProjectDetailByIds(query.getDataIdList()); + List<ProjectExcelTemplate> excelList = new ArrayList<>(); + //瀛楀吀浣滅炕璇� + data.forEach(item -> { + ProjectExcelTemplate excel = new ProjectExcelTemplate(); + BeanUtils.copyProperties(item, excel); + //椤圭洰闃舵 + excel.setProjectPhase(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null)); + excelList.add(excel); + }); + Set<Integer> indexes = OutputExcelUtils.getSelectFields(query.getFieldList(), ProjectExcelTemplate.class); + //涓嶉渶瑕侀檮浠� + if (!query.getRequireFile()) { + try (ServletOutputStream outputStream = response.getOutputStream()) { + EasyExcel.write(outputStream, ProjectExcelTemplate.class).includeColumnIndexes(indexes).sheet("椤圭洰鍒楄〃").doWrite(excelList); + } catch (Exception e) { + log.error(e.getMessage(), e); + throw new BaseException("瀵煎嚭澶辫触锛�" + e.getMessage()); + } + } else { + //闇�瑕侀檮浠�,瀵煎嚭涓簔ip + response.setContentType("application/zip"); + response.setCharacterEncoding(Charsets.UTF_8.name()); + String fileName = URLEncoder.encode("椤圭洰鍒楄〃", Charsets.UTF_8); + response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".zip"); + + java.io.File tempDir = null; + try { + //鏌ュ嚭闄勪欢 + List<File> projectInfoFiles = fileMapper.selectList(new QueryWrapper<File>() + .in("bus_id", data.stream().map(ProjectVO::getId).collect(Collectors.toList())) + .and(q -> q.eq("type", FileTypeEnum.PROJECT_INFO.getType()).or().eq("type", FileTypeEnum.DOCUMENT_INFO.getType())) + ); + List<File> policyFiles = fileMapper.selectList(new QueryWrapper<File>() + .in("bus_id", data.stream().map(ProjectVO::getPolicyId).collect(Collectors.toList())) + .eq("type", FileTypeEnum.INVEST_POLICY.getType()) + ); + projectInfoFiles.addAll(policyFiles); + projectInfoFiles.forEach(file -> { + data.forEach(item -> { + if (FileTypeEnum.PROJECT_INFO.equals(file.getType())) { + if (file.getBusId().equals(item.getId())) { + //瀹℃壒璁″垝涔� + item.setApprovalPlan(file.getOriginalName()); + } + } else if (FileTypeEnum.DOCUMENT_INFO.equals(file.getType())) { + if (file.getBusId().equals(item.getId())) { + //鐩稿叧鏂囦功 + item.setDocuments(file.getOriginalName()); + } + } else if (FileTypeEnum.INVEST_POLICY.equals(file.getType())) { + if (file.getBusId().equals(item.getPolicyId())) { + //绗﹀悎浜т笟鏀跨瓥闄勪欢 + item.setPolicyComplianceAttachment(file.getOriginalName()); + } + } + }); + + }); + // 鍒涘缓涓存椂鐩綍 + tempDir = Files.createTempDirectory("temp").toFile(); + + java.io.File templateDir = new java.io.File(tempDir, "template"); + if (!templateDir.exists()) { + templateDir.mkdirs(); + } + + // 鍒涘缓 Excel 鏂囦欢 + java.io.File excelFile = new java.io.File(templateDir, "excel.xlsx"); + if (!excelFile.exists()) { + excelFile.createNewFile(); + } + + // 鍐欏叆 Excel 妯℃澘鏁版嵁 + try (FileOutputStream fileOutputStream = new FileOutputStream(excelFile, false)) { + EasyExcel.write(fileOutputStream, ProjectExcelTemplate.class).sheet("椤圭洰鍒楄〃").doWrite(excelList); + } + + // 鍒涘缓闄勪欢鐩綍 + java.io.File attachmentDir = new java.io.File(templateDir, "attachment"); + if (!attachmentDir.exists()) { + attachmentDir.mkdirs(); + } + + //灏嗛檮浠跺鍒跺鐩綍涓� + for (File item : projectInfoFiles) { + String url = item.getUrl(); + if (StringUtils.isBlank(url)) continue; + url = url.replaceFirst(Constants.RESOURCE_PREFIX, SysConfig.getProfile()); + // 鐩存帴浣跨敤File绫诲鐞嗙郴缁熻矾寰� + java.io.File sourceFile = new java.io.File(url); + if (!sourceFile.exists() || !sourceFile.isFile()) { + // 澶勭悊鏂囦欢涓嶅瓨鍦ㄦ垨涓嶆槸鏂囦欢鐨勬儏鍐� + continue; // 鎴栬�呰褰曟棩蹇椼�佹姏鍑哄紓甯哥瓑 + } + // 鐩爣鏂囦欢璺緞锛屼娇鐢╝ttachmentDir浣滀负鐖剁洰褰曪紝骞朵繚鎸佸師鏂囦欢鍚� + java.io.File targetFile = new java.io.File(attachmentDir, sourceFile.getName()); + // 澶嶅埗鏂囦欢 + try { + Files.copy(sourceFile.toPath(), targetFile.toPath(), StandardCopyOption.REPLACE_EXISTING); + } catch (IOException e) { + log.error(e.getMessage()); + } + } + // 鎵撳寘 ZIP 鏂囦欢 + java.io.File zipFile = ZipUtil.zip(templateDir); + byte[] zipBytes = Files.readAllBytes(zipFile.toPath()); + + // 灏� ZIP 鏂囦欢鍐欏叆鍝嶅簲 + try (ServletOutputStream outputStream = response.getOutputStream()) { + outputStream.write(zipBytes); + } + } finally { + OutputExcelUtils.deleteDirectoryOrFile(tempDir); + } + } + } + + @Override + public Result updateUsedStatus(Integer id, Integer usedStatus) { + ProjectInfo entity = baseMapper.selectById(id); + // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊 + Assert.notNull(entity, "璁板綍涓嶅瓨鍦�"); + entity.setUsedStatus(usedStatus); + baseMapper.updateById(entity); + return Result.ok("鍒犻櫎鎴愬姛"); + } + + @Transactional + @Override + public void importProject(MultipartFile file) { + java.io.File tempZip = null; + java.io.File tempDir = null; + try { + tempZip = java.io.File.createTempFile("temp", ".zip"); + try (InputStream inputStream = file.getInputStream()) { + Files.copy(inputStream, tempZip.toPath(), StandardCopyOption.REPLACE_EXISTING); + } + + tempDir = Files.createTempDirectory("temp-dir").toFile(); + + try { + ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_UTF_8); + } catch (IORuntimeException e) { + log.error("瑙e帇澶辫触锛屽皾璇曚娇鐢℅BK缂栫爜瑙e帇..."); + ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_GBK); + } + + Path path = tempDir.toPath(); + ProjectImportListener projectImportListener = new ProjectImportListener(); + //闇�瑕佸瓨鍏ユ暟鎹簱 + ArrayList<File> attachments = new ArrayList<>(); + //瀹為檯鐨勬枃浠� + List<java.io.File> files = new ArrayList<>(); + try (Stream<Path> pathStream = Files.walk(path)) { + pathStream.forEach(filePath -> { + if (Files.isDirectory(filePath) && filePath.getFileName().toString().equals("attachment")) { + //闄勪欢澶勭悊 + try (Stream<Path> stream = Files.walk(filePath)) { + stream.forEach(attachmentPath -> { + if (Files.isRegularFile(attachmentPath)) { + files.add(attachmentPath.toFile()); + } + }); + } catch (IOException e) { + log.error(e.getMessage()); + throw new RuntimeException(e); + } + } else if (Files.isRegularFile(filePath) && filePath.getFileName().toString().equals("excel.xlsx")) { + //瀵煎叆鏁版嵁澶勭悊 + try (FileInputStream fileInputStream = new FileInputStream(filePath.toFile())) { + ExcelUtil.importExcel(fileInputStream, ProjectExcelTemplate.class, projectImportListener); + } catch (IOException e) { + log.error(e.getMessage()); + throw new RuntimeException(e); + } + } + }); + } + /* 鎶婇檮浠跺瓨鍏ヤ笂浼犺矾寰� */ + //涓婁紶鏂囦欢璺緞 + String filePath = SysConfig.getUploadPath(); + files.forEach(attachmentFile -> { + try { + //涓婁紶 + String url = FileUploadUtils.uploadIOFile(filePath, attachmentFile); + //瀛樻斁鐨勬枃浠跺悕浼氬姞涓奯鏃堕棿鎴崇殑鍚庣紑 + String newName = FileUtils.getName(url); + attachments.add(new File() + .setUrl(url) + .setName(newName) + .setOriginalName(attachmentFile.getName())); + } catch (IOException e) { + e.printStackTrace(); + } + }); + + Map<String, Long> projectInfoMap = projectImportListener.getProjectInfoMap(); + Map<String, Long> investmentProjectPolicyComplianceMap = projectImportListener.getInvestmentProjectPolicyComplianceMap(); + Map<String, Long> documentsMap = projectImportListener.getDocumentsMap(); + //琛ュ厖type鍜宐usId瀛楁 + attachments.forEach(attachment -> { + String fileName = attachment.getOriginalName(); + if (ObjectUtil.isNotNull(projectInfoMap.get(fileName))) { + attachment.setBusId(projectInfoMap.get(fileName)); + attachment.setType(FileTypeEnum.PROJECT_INFO); + } else if (ObjectUtil.isNotNull(investmentProjectPolicyComplianceMap.get(fileName))) { + attachment.setBusId(investmentProjectPolicyComplianceMap.get(fileName)); + attachment.setType(FileTypeEnum.INVEST_POLICY); + } else if (ObjectUtil.isNotNull(documentsMap.get(fileName))) { + attachment.setBusId(documentsMap.get(fileName)); + attachment.setType(FileTypeEnum.DOCUMENT_INFO); + } + }); + + // 娣诲姞椤圭洰闄勪欢鍏宠仈琛� + fileService.saveBatch(attachments); + } catch (IOException e) { + log.error(e.getMessage()); + throw new RuntimeException(e); + } finally { + deleteDirectoryOrFile(tempZip); + deleteDirectoryOrFile(tempDir); + } + } + + @Override + public Result editProject(ProjectForm form) { + if (ObjectUtil.isNotNull(form.getProjectInfoForm())) { + if (ObjectUtil.isNull(form.getProjectInfoForm().getId())) { + //鏂板 + add(form.getProjectInfoForm()); + } else { + //鏇存柊 + update(form.getProjectInfoForm()); + } + } + if (ObjectUtil.isNotNull(form.getProjectInvestmentInfoForm())) { + if (ObjectUtil.isNull(form.getProjectInvestmentInfoForm().getId())) { + projectInvestmentInfoServiceImpl.add(form.getProjectInvestmentInfoForm()); + } else { + projectInvestmentInfoServiceImpl.update(form.getProjectInvestmentInfoForm()); + } + } + if (ObjectUtil.isNotNull(form.getProjectInvestmentFundingForm())) { + if (ObjectUtil.isNull(form.getProjectInvestmentFundingForm().getId())) { + projectInvestmentFundingServiceImpl.add(form.getProjectInvestmentFundingForm()); + } else { + projectInvestmentFundingServiceImpl.update(form.getProjectInvestmentFundingForm()); + } + } + if (ObjectUtil.isNotNull(form.getProjectInvestmentPolicyComplianceForm())) { + if (ObjectUtil.isNull(form.getProjectInvestmentPolicyComplianceForm().getId())) { + projectInvestmentPolicyComplianceServiceImpl.add(form.getProjectInvestmentPolicyComplianceForm()); + } else { + projectInvestmentPolicyComplianceServiceImpl.update(form.getProjectInvestmentPolicyComplianceForm()); + } + } + if (ObjectUtil.isNotNull(form.getProjectUnitRegistrationInfoForm())) { + if (ObjectUtil.isNull(form.getProjectUnitRegistrationInfoForm().getId())) { + projectUnitRegistrationInfoServiceImpl.add(form.getProjectUnitRegistrationInfoForm()); + } else { + projectUnitRegistrationInfoServiceImpl.update(form.getProjectUnitRegistrationInfoForm()); + } + } + if (ObjectUtil.isNotNull(form.getDocumentInfoForm())) { + if (ObjectUtil.isNull(form.getDocumentInfoForm().getProjectId())) { + return Result.error("璇峰厛淇濆瓨鎶曡祫绠$悊鍩烘湰淇℃伅"); + } else { + addDoc(form.getDocumentInfoForm()); + } + } + + return Result.ok("鎻愪氦鎴愬姛"); + } + + @Override + public Result searchByKey(String wordKey) { + Wrapper wrapper = null; + if (!StringUtils.isEmpty(wordKey)) { + wrapper = Wrappers.<ProjectInfo>lambdaQuery().like(ProjectInfo::getProjectName, wordKey).or().like(ProjectInfo::getProjectCode, wordKey); + } + List<ProjectInfo> list = baseMapper.selectList(wrapper); + return Result.ok().data(list); + } + + private static void deleteDirectoryOrFile(java.io.File file) { + if (ObjectUtil.isNull(file)) { + return; + } + + if (file.isDirectory()) { + java.io.File[] files = file.listFiles(); + if (files != null) { + for (java.io.File f : files) { + deleteDirectoryOrFile(f); + } + } + } + file.delete(); + } + + /** + * 椤圭洰鍚嶇О鏍¢獙閲嶅 + * + * @return + */ + public void checkProjectNameUnique(ProjectInfo projectInfo) { + if (StrUtil.isBlank(projectInfo.getProjectName())) { + throw new BaseException("椤圭洰鍚嶇О涓嶈兘涓虹┖"); + } + Assert.isTrue(baseMapper.selectCount(Wrappers.<ProjectInfo>lambdaQuery().eq(ProjectInfo::getProjectName, projectInfo.getProjectName())) == 0, "椤圭洰鍚嶇О閲嶅"); + } + } -- Gitblit v1.8.0