From e0d74a95d51fe3182f0205007ee0a39f37e58886 Mon Sep 17 00:00:00 2001
From: yz <yz_0812@outlook.com>
Date: 星期五, 19 二月 2021 17:11:07 +0800
Subject: [PATCH] 修改附件管理

---
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java |  155 +++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 143 insertions(+), 12 deletions(-)

diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java
index 6503635..ae0f802 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java
@@ -1,11 +1,26 @@
 package cc.mrbird.febs.server.hr.service.impl;
 
-import cc.mrbird.febs.server.hr.entity.EmpAccessory;
+import cc.mrbird.febs.common.core.constant.ModuleCode;
+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.mapper.EmpAccessoryMapper;
+import cc.mrbird.febs.server.hr.properties.FebsServerHrProperties;
 import cc.mrbird.febs.server.hr.service.IEmpAccessoryService;
+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.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.transaction.annotation.Propagation;
@@ -15,9 +30,13 @@
 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 org.springframework.web.multipart.MultipartFile;
 
-import java.util.List;
-import java.util.Map;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.io.IOException;
+import java.util.*;
 
 /**
  * 附件管理 Service实现
@@ -29,8 +48,17 @@
 @RequiredArgsConstructor
 @Transactional(propagation = Propagation.SUPPORTS, readOnly = true)
 public class EmpAccessoryServiceImpl extends ServiceImpl<EmpAccessoryMapper, EmpAccessory> implements IEmpAccessoryService {
-
+    private final ILabelService labelService;
     private final EmpAccessoryMapper empAccessoryMapper;
+    private final FebsServerHrProperties properties;
+    private final String operatorId = Optional.ofNullable(FebsUtil.getCurrentUser())
+            .map(u -> u.getUserId().toString())
+            .orElse("1");
+    private final IEmpBaseInfoService empBaseInfoService;
+    /**
+     * 参数2为数据中心ID 参数1为终端ID
+     */
+    private Snowflake snowflake = IdUtil.getSnowflake(ModuleCode.HR_FIlE, 1);
 
     @Override
     public IPage<Map<String,Object>> findEmpAccessorys(QueryRequest request, EmpAccessoryVO vo) {
@@ -93,20 +121,46 @@
         queryWrapper.eq("a.delFlag",0);
         Page<EmpAccessory> page = new Page<>(request.getPageNum(), request.getPageSize());
         String querySql = empAccessoryMapper.getQuerySql();
-        return empAccessoryMapper.selectPageVo(page, queryWrapper,querySql);
+        IPage<Map<String, Object>> mapIPage = empAccessoryMapper.selectPageVo(page, queryWrapper, querySql);
+        return mapIPage;
     }
 
     @Override
     public List<EmpAccessory> findEmpAccessorys(EmpAccessory empAccessory) {
         LambdaQueryWrapper<EmpAccessory> queryWrapper = new LambdaQueryWrapper<>();
-        // TODO 设置查询条件
+        queryWrapper.eq(EmpAccessory::getDelFlag,0);
+        if (null != empAccessory.getLabelid()) {
+            queryWrapper.eq(EmpAccessory::getLabelid, empAccessory.getLabelid());
+        }
+        if (StrUtil.isNotBlank(empAccessory.getFilesname())) {
+            queryWrapper.like(EmpAccessory::getFilesname, empAccessory.getFilesname());
+        }
+        if (null != empAccessory.getEmpid()) {
+            queryWrapper.eq(EmpAccessory::getEmpid, empAccessory.getEmpid());
+        }
         return this.baseMapper.selectList(queryWrapper);
     }
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void createEmpAccessory(EmpAccessory empAccessory) {
-        this.save(empAccessory);
+    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());
+        EmpAccessory accessory = new EmpAccessory();
+        accessory.setCreatetime(new Date());
+        accessory.setCreator(operatorId);
+        accessory.setDelFlag(0);
+        accessory.setEmpid(empId);
+        accessory.setFilesaddress(dict.getStr("newName"));
+        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"));
+        accessory.setLabelid(labelid);
+        accessory.setModifier(operatorId);
+        accessory.setModifytime(new Date());
+        accessory.setVersion(0);
+        this.save(accessory);
     }
 
     @Override
@@ -117,9 +171,86 @@
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void deleteEmpAccessory(EmpAccessory empAccessory) {
-        LambdaQueryWrapper<EmpAccessory> wapper = new LambdaQueryWrapper<>();
-        // TODO 设置删除条件
-        this.remove(wapper);
+    public void deleteEmpAccessory(String accessoryids) {
+        String[] arr = accessoryids.split(",");
+        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
+        wapper.in(EmpAccessory::getAccessoryid,arr);
+        wapper.set(EmpAccessory::getDelFlag,1);
+        wapper.set(EmpAccessory::getModifier,operatorId);
+        wapper.set(EmpAccessory::getModifytime,new Date());
+        this.update(wapper);
+
+        LambdaQueryWrapper<EmpAccessory> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.in(EmpAccessory::getAccessoryid,arr);
+        List<EmpAccessory> list = this.list(queryWrapper);
+        list.parallelStream().forEach(i-> {
+            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(i.getEmpid());
+            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
+            FileUtil.del(properties.getUploadSinglePath()+nameAndCertificateNumb+"/" + i.getFilesaddress());
+        });
+    }
+
+
+    @Override
+    public void download(String accessoryids, HttpServletRequest request, HttpServletResponse response) {
+        String[] arr = accessoryids.split(",");
+        List<File> files = new ArrayList<>();
+        List<String> fileName = new ArrayList<>();
+        Arrays.stream(arr).forEach(i -> {
+            EmpAccessory accessory = this.getById(i);
+            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(accessory.getEmpid());
+            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
+            files.add(new File(properties.getUploadSinglePath() + nameAndCertificateNumb + accessory.getFilesaddress()));
+            fileName.add(accessory.getFilesname());
+        });
+        MyUtil.download(request, response, files, fileName);
+
+    }
+
+    @Override
+    public void mvFiles(String accessoryids,Long labelid,Long empId) {
+        String[] arr = accessoryids.split(",");
+
+        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
+        wapper.in(EmpAccessory::getAccessoryid, arr);
+        wapper.set(EmpAccessory::getModifier,operatorId);
+        wapper.set(EmpAccessory::getModifytime,new Date());
+        wapper.set(EmpAccessory::getLabelid,labelid);
+        this.update(wapper);
+
+    }
+
+    @Override
+    public void singledownload(String empIds, Long labelid, HttpServletRequest request, HttpServletResponse response) throws IOException {
+        String[] arr = empIds.split(",");
+        //主文件夹
+        String mianFileStr = properties.getUploadSinglePath()+snowflake.nextIdStr()+"/";
+        FileUtil.mkdir(mianFileStr);
+        Arrays.stream(arr).forEach(i -> {
+            LambdaQueryWrapper<EmpAccessory> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+            lambdaQueryWrapper.eq(EmpAccessory::getEmpid,i);
+            lambdaQueryWrapper.eq(EmpAccessory::getDelFlag,0);
+            if (!labelid.equals(-1L)){
+                lambdaQueryWrapper.eq(EmpAccessory::getLabelid,labelid);
+            }
+            List<EmpAccessory> accessory = this.list(lambdaQueryWrapper);
+            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(i);
+            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
+            //创建个人文件夹
+            String singlefileStr = mianFileStr+nameAndCertificateNumb;
+            FileUtil.mkdir(singlefileStr);
+            accessory.stream().forEach(a->{
+                Label label = labelService.getById(a.getLabelid());
+                FileUtil.copy(properties.getUploadSinglePath() + nameAndCertificateNumb + a.getFilesaddress(),singlefileStr+label.getLabelname()+"/"+a.getFilesname(),true);
+            });
+        });
+        ZipUtil.zip(mianFileStr);
+        try {
+            MyUtil.downloadFile(new File(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString()),response);
+        } catch (IOException e) {
+            log.error("下载文件异常",e);
+        }
+        FileUtil.del(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString());
+        FileUtil.del(mianFileStr);
     }
 }

--
Gitblit v1.8.0