射洪市项目管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
fuliqi
2024-12-23 deb49773f332f83dafb78788bc3d9b2b39fa421c 
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java


@@ -1,12 +1,16 @@


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.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.base.Result;


import com.ycl.common.config.SysConfig;


@@ -21,6 +25,9 @@


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.domain.entity.*;


import com.ycl.domain.excel.ProjectExcelTemplate;


import com.ycl.domain.form.DocumentInfoForm;


@@ -30,6 +37,7 @@


import com.ycl.domain.query.ProjectInfoQuery;


import com.ycl.domain.vo.*;


import com.ycl.framework.utils.PageUtil;


import com.ycl.listener.excel.ProjectImportListener;


import com.ycl.mapper.*;


import com.ycl.service.FileService;


import com.ycl.service.ProjectInfoService;


@@ -38,20 +46,25 @@


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;


import org.springframework.web.multipart.MultipartFile;




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;




/**


@@ -546,4 +559,129 @@


        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("解压失败，尝试使用GBK编码解压...");


                    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和busId字段


            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);


        }


    }




    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, "项目名称重复");


    }




}