luoyb
2021-06-18 543941f345319f29f92d9be2529d41868bc3cf80
febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java
@@ -1,11 +1,14 @@
package cc.mrbird.febs.server.hr.service.impl;
import cc.mrbird.febs.common.core.constant.ModuleCode;
import cc.mrbird.febs.common.core.entity.QueryRequest;
import cc.mrbird.febs.common.core.exception.FebsException;
import cc.mrbird.febs.common.core.utils.FebsUtil;
import cc.mrbird.febs.common.core.utils.MyUtil;
import cc.mrbird.febs.common.core.utils.SequenceUtil;
import cc.mrbird.febs.server.hr.entity.*;
import cc.mrbird.febs.server.hr.entity.EmpAccessory;
import cc.mrbird.febs.server.hr.entity.EmpBaseInfo;
import cc.mrbird.febs.server.hr.entity.Label;
import cc.mrbird.febs.server.hr.feign.IRemoteDeptService;
import cc.mrbird.febs.server.hr.mapper.EmpAccessoryMapper;
import cc.mrbird.febs.server.hr.properties.FebsServerHrProperties;
@@ -13,37 +16,31 @@
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import cc.mrbird.febs.server.hr.service.ILabelService;
import cc.mrbird.febs.server.hr.vo.EmpAccessoryVO;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.Dict;
import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.weaver.ast.Or;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.annotation.Propagation;
import lombok.RequiredArgsConstructor;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import cc.mrbird.febs.common.core.entity.QueryRequest;
import cc.mrbird.febs.common.core.entity.constant.StringConstant;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.io.*;
import java.util.*;
import java.util.logging.Logger;
import java.util.function.Consumer;
/**
 * 附件管理 Service实现
@@ -74,7 +71,7 @@
        if (StrUtil.isNotBlank(vo.getBasic())) {
            queryWrapper.and(p -> {
                p.like("a.empNumb", vo.getBasic())
                        .or().like("a.deptName", vo.getBasic())
                        .or().like("a.allDeptName", vo.getBasic())
                        .or().like("a.certificateNumb", vo.getBasic())
                        .or().like("a.empName", vo.getBasic())
                        .or().like("a.jobName", vo.getBasic());
@@ -90,35 +87,22 @@
            queryWrapper.like("a.certificateNumb", vo.getCertificateNumb());
        }
        if (StrUtil.isNotBlank(vo.getDeptName())) {
            queryWrapper.like("a.deptName", vo.getDeptName());
            queryWrapper.like("a.allDeptName", vo.getDeptName());
        }
        if (StrUtil.isNotBlank(vo.getEntryDate())) {
            queryWrapper.between("a.entryDate", vo.getEntryDate().split(",")[0], vo.getEntryDate().split(",")[1]);
        }
        if (StrUtil.isNotBlank(vo.getAge())) {
            String[] arr = vo.getAge().split(",");
            for (String s : arr) {
                if (StrUtil.equals(s, "18-29")) {
                    queryWrapper.or().between("a.age", 18, 29);
                }
                if (StrUtil.equals(s, "30-39")) {
                    queryWrapper.or().between("a.age", 30, 39);
                }
                if (StrUtil.equals(s, "40-49")) {
                    queryWrapper.or().between("a.age", 40, 49);
                }
                if (StrUtil.equals(s, "50-59")) {
                    queryWrapper.or().between("a.age", 50, 59);
                }
                if (StrUtil.equals(s, "60-69")) {
                    queryWrapper.or().between("a.age", 60, 69);
                }
                if (StrUtil.equals(s, "69-999")) {
                    queryWrapper.or().ge("a.age", 69);
                }
            }
        if (StringUtils.isNotBlank(vo.getAge())) {
            String[] ages = vo.getAge().split(",");
            Consumer<QueryWrapper<EmpAccessoryVO>> consumer = wrapper -> {
                for (int i = 0; i < ages.length; i++) {
                    String ageBtn = ages[i];
                    wrapper.or().between("a.Age", ageBtn.split("-")[0], ageBtn.split("-")[1]);
                }
            };
            queryWrapper.and(consumer);
        }
        if (StrUtil.isNotBlank(vo.getArchivesStatus())) {
            queryWrapper.in("a.archivesStatus", vo.getArchivesStatus().split(","));
@@ -152,10 +136,55 @@
        }
        if (StringUtils.isNotBlank(vo.getEntryType())) {
            queryWrapper.in("a.entryType", vo.getEntryType().split(","));
        }        if (StringUtils.isNotBlank(vo.getCertificateList())) {
            String[] certificates = vo.getCertificateList().split(",");
            Consumer<QueryWrapper<EmpAccessoryVO>> consumer = new Consumer<QueryWrapper<EmpAccessoryVO>>() {
                @Override
                public void accept(QueryWrapper<EmpAccessoryVO> wrapper) {
                    for (int i = 0; i < certificates.length; i++) {
                        String ageBtn = certificates[i];
                        wrapper.or().like("a.certificateList", ageBtn);
                    }
                }
            };
            queryWrapper.and(consumer);
        }
        queryWrapper.in("c.dept_Id", remoteDeptService.userRightDepts().split(StringConstant.COMMA));
        queryWrapper.orderByAsc("a.deptName,a.empName,a.certificateNumb");
        queryWrapper.inSql("c.dept_Id", remoteDeptService.userRightDepts());
        if (StrUtil.isNotBlank(vo.getFileName())){
            queryWrapper.like("acc.filesName", vo.getFileName());
        }
        if (StrUtil.equals(vo.getField(),"empName")){
            if (StrUtil.equals(vo.getOrder(),"descending")){
                queryWrapper.orderByDesc("a.empName");
            }else{
                queryWrapper.orderByAsc("a.empName");
            }
        }else if(StrUtil.equals(vo.getField(),"empNumb")){
            if (StrUtil.equals(vo.getOrder(),"descending")){
                queryWrapper.orderByDesc("a.empNumb");
            }else{
                queryWrapper.orderByAsc("a.empNumb");
            }
        }else if(StrUtil.equals(vo.getField(),"deptName")){
            if (StrUtil.equals(vo.getOrder(),"descending")){
                queryWrapper.orderByDesc("deptName");
            }else{
                queryWrapper.orderByAsc("deptName");
            }
        }else if(StrUtil.equals(vo.getField(),"jobName")){
            if (StrUtil.equals(vo.getOrder(),"descending")){
                queryWrapper.orderByDesc("a.jobName");
            }else{
                queryWrapper.orderByAsc("a.jobName");
            }
        }else if(StrUtil.equals(vo.getField(),"certificateNumb")){
            if (StrUtil.equals(vo.getOrder(),"descending")){
                queryWrapper.orderByDesc("a.certificateNumb");
            }else{
                queryWrapper.orderByAsc("a.certificateNumb");
            }
        }
        queryWrapper.groupBy(" a.delFlag,a.empName,a.empStatus,a.empId,a.certificateNumb,a.empNumb,deptName,a.jobName");
        Page<EmpAccessory> page = new Page<>(request.getPageNum(), request.getPageSize());
       // String querySql = empAccessoryMapper.getQuerySql();
       //List<Label> labelList = labelService.findLabels(new Label());
@@ -188,13 +217,13 @@
    public void createEmpAccessory(Long labelid, MultipartFile file, Long empId) throws FebsException, IOException {
        EmpBaseInfo empBaseInfo = empBaseInfoService.getById(empId);
        String nameAndCertificateNumb = empBaseInfo.getEmpName() + "_" + empBaseInfo.getCertificateNumb() + "/";
        Dict dict = MyUtil.filesUpload(file, properties.getUploadSinglePath() + nameAndCertificateNumb, snowflake.nextIdStr());
        Dict dict = MyUtil.filesUpload(file, properties.getUploadSinglePath() + nameAndCertificateNumb);
        EmpAccessory accessory = new EmpAccessory();
        accessory.setCreatetime(new Date());
        accessory.setCreator(FebsUtil.getUserId());
        accessory.setDelFlag(0);
        accessory.setEmpid(empId);
        accessory.setFilesaddress(dict.getStr("newName"));
        accessory.setFilesaddress(dict.getStr("fileName"));
        accessory.setFilesformat(StrUtil.isBlank(dict.getStr("suffix")) ? dict.getStr("suffix") : dict.getStr("suffix").substring(1));
        accessory.setAccessoryid(SequenceUtil.generateId(0L, ModuleCode.HR_FIlE));
        accessory.setFilesname(dict.getStr("fileName"));
@@ -208,12 +237,25 @@
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateEmpAccessory(EmpAccessory empAccessory) throws FebsException {
        if ( empAccessory.getFilesname().indexOf(".")<=0){
        if ( empAccessory.getFilesname().lastIndexOf(".")<=0){
            throw  new FebsException("文件名格式错误");
        }
        EmpAccessory accessory = this.getById(empAccessory.getAccessoryid());
        EmpBaseInfo empBaseInfo = empBaseInfoService.getById(accessory.getEmpid());
        String nameAndCertificateNumb = empBaseInfo.getEmpName() + "_" + empBaseInfo.getCertificateNumb() + "/";
        if (FileUtil.exist(properties.getUploadSinglePath() + nameAndCertificateNumb+empAccessory.getFilesname())){
            throw  new FebsException("文件已经存在");
        }
        File file = new File(properties.getUploadSinglePath() + nameAndCertificateNumb+accessory.getFilesaddress());
        FileUtil.rename(file,empAccessory.getFilesname(),true);
        LambdaUpdateWrapper<EmpAccessory> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(EmpAccessory::getAccessoryid,empAccessory.getAccessoryid())
                .set(EmpAccessory::getFilesname,empAccessory.getFilesname()).set(EmpAccessory::getFilesformat,  empAccessory.getFilesname().substring( empAccessory.getFilesname().indexOf(".")+1));
                .set(EmpAccessory::getFilesname,empAccessory.getFilesname())
                .set(EmpAccessory::getFilesaddress,empAccessory.getFilesname())
                .set(EmpAccessory::getFilesformat,  empAccessory.getFilesname().substring( empAccessory.getFilesname().lastIndexOf(".")+1));
        this.update(wrapper);
    }
@@ -252,7 +294,7 @@
            if (fileName.parallelStream().filter(filter -> StrUtil.equals(accessory.getFilesname(), filter)).count() == 0) {
                fileName.add(accessory.getFilesname());
            } else {
                fileName.add(accessory.getFilesname() + fileName.parallelStream().filter(filter -> StrUtil.equals(accessory.getFilesname(), filter)).count() + 1);
                fileName.add(accessory.getFilesname().substring(0,accessory.getFilesname().lastIndexOf(".")) + fileName.parallelStream().filter(filter -> StrUtil.equals(accessory.getFilesname(), filter)).count()+"."+accessory.getFilesformat());
            }
        });
        MyUtil.download(request, response, files, fileName);
@@ -330,7 +372,11 @@
            File[] childlist = FileUtil.ls(fileids+"/"+file.getName());
            for (File file1 : childlist) {
                EmpAccessory accessory = new EmpAccessory();
                if (file1.getName().indexOf("体检")>=0){
                if (file1.getName().indexOf("外皮")>=0||file1.getName().indexOf("外批")>=0){
                    accessory.setLabelid(142445285230593L);
                }else if (file1.getName().indexOf("心电图")>=0){
                    accessory.setLabelid(142445218121729L);
                }else if (file1.getName().indexOf("体检")>=0){
                    accessory.setLabelid(142445218121729L);
                }else if (file1.getName().indexOf("合同")>=0){
                    accessory.setLabelid(142445285230593L);
@@ -345,7 +391,7 @@
                }else if (file1.getName().indexOf("证")>=0){
                    accessory.setLabelid(143073812170753L);
                }else{
                    accessory.setLabelid(142445180372993L);
                    accessory.setLabelid(143073866696705L);
                }
                accessory.setCreatetime(new Date());
@@ -353,8 +399,8 @@
                accessory.setDelFlag(0);
                accessory.setEmpid(empBaseInfo.get(0).getEmpId());
                accessory.setFilesaddress(file1.getName());
                if (file1.getName().indexOf(".") > 0) {
                    accessory.setFilesformat( file1.getName().substring(file1.getName().indexOf(".")+1));
                if (file1.getName().lastIndexOf(".") > 0) {
                    accessory.setFilesformat( file1.getName().substring(file1.getName().lastIndexOf(".")+1));
                }
                accessory.setAccessoryid(SequenceUtil.generateId(0L, ModuleCode.HR_FIlE));
                accessory.setFilesname(file1.getName());
@@ -366,4 +412,34 @@
        }
    }
    @Override
    public void getImage(String accessoryid, HttpServletResponse response) throws Exception {
        EmpAccessory empAccessory = this.getById(accessoryid);
        if (StrUtil.isBlank(empAccessory.getFilesaddress())) {
            return;
        }
        EmpBaseInfo empBaseInfo = empBaseInfoService.getById(empAccessory.getEmpid());
        String nameAndCertificateNumb = empBaseInfo.getEmpName() + "_" + empBaseInfo.getCertificateNumb() + "/";
        String path = properties.getUploadSinglePath() + nameAndCertificateNumb + empAccessory.getFilesaddress();
        try (InputStream inputStream = new FileInputStream(path); OutputStream out = response.getOutputStream()) {
            //byte数组用于存放图片字节数据
            byte[] buff = new byte[inputStream.available()];
            inputStream.read(buff);
            inputStream.close();
            if (empAccessory.getFilesformat().equals("tif")){
                //设置发送到客户端的响应内容类型
                response.setContentType("image/tiff");
            }else if (empAccessory.getFilesformat().equals("bmp")){
                response.setContentType("application/x-bmp");
            }else if (empAccessory.getFilesformat().equals("jpg")){
                response.setContentType("image/jpeg");
            }else if (empAccessory.getFilesformat().equals("png")){
                response.setContentType("image/png");
            }else if (empAccessory.getFilesformat().equals("gif")){
                response.setContentType("image/gif");
            }
            out.write(buff);
        }
    }
}