From c731b7ce86df3deb69ae853910e56bb374e92017 Mon Sep 17 00:00:00 2001
From: yz <yz_0812@outlook.com>
Date: 星期五, 19 二月 2021 04:57:05 +0800
Subject: [PATCH] 完成附件管理

---
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/IEmpAccessoryService.java         |    5 ++
 febs-common/febs-common-core/src/main/java/cc/mrbird/febs/common/core/utils/MyUtil.java                     |   28 +++++++++++++
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/FilesUploadController.java     |    2 
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java  |    1 
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpAccessoryController.java    |   22 +++++++++++
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/EmpAccessoryServiceImpl.java |   52 +++++++++++++++++++++++++
 6 files changed, 106 insertions(+), 4 deletions(-)

diff --git a/febs-common/febs-common-core/src/main/java/cc/mrbird/febs/common/core/utils/MyUtil.java b/febs-common/febs-common-core/src/main/java/cc/mrbird/febs/common/core/utils/MyUtil.java
index 226d3a1..2b4b383 100644
--- a/febs-common/febs-common-core/src/main/java/cc/mrbird/febs/common/core/utils/MyUtil.java
+++ b/febs-common/febs-common-core/src/main/java/cc/mrbird/febs/common/core/utils/MyUtil.java
@@ -1,6 +1,5 @@
 package cc.mrbird.febs.common.core.utils;
 
-import cc.mrbird.febs.common.core.entity.system.Dic;
 import cc.mrbird.febs.common.core.exception.FebsException;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.lang.Dict;
@@ -9,6 +8,7 @@
 import org.springframework.web.multipart.MultipartFile;
 import cn.hutool.core.io.FileUtil;
 
+import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.*;
@@ -21,6 +21,32 @@
 @Slf4j
 public class MyUtil {
 
+    public static void downloadFile(File file, HttpServletResponse response) throws IOException {
+        InputStream fin = null;
+        ServletOutputStream out = null;
+        try {
+            fin = new FileInputStream(file);
+            out = response.getOutputStream();
+            response.setCharacterEncoding("utf-8");
+            response.setContentType("application/x-download");
+            response.addHeader("Content-Disposition", "attachment;filename=resume.doc");
+
+            byte[] buffer = new byte[1024];
+            int bytesToRead = -1;
+            // 通过循环将读入的Word文件的内容输出到浏览器中
+            while((bytesToRead = fin.read(buffer)) != -1) {
+                out.write(buffer, 0, bytesToRead);
+            }
+        } catch (Exception e) {
+            log.error("下载文件异常" + e);
+            e.printStackTrace();
+        } finally {
+            if(fin != null) fin.close();
+            if(out != null) out.close();
+
+        }
+    }
+
     public static void  download(HttpServletRequest request, HttpServletResponse response, List<File> files , List<String> fileName){
         //设置压缩包的名字
         //解决不同浏览器压缩包名字含有中文时乱码的问题
diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpAccessoryController.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpAccessoryController.java
index d45dad2..a144f95 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpAccessoryController.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpAccessoryController.java
@@ -97,4 +97,26 @@
         }
     }
 
+    @PostMapping("/mvFiles")
+    @PreAuthorize("hasAuthority('empAccessory:mv')")
+    public void mvFiles(@NotBlank(message = "{required}")  String accessoryids,@NotNull(message = "{required}") Long labelid,@NotNull(message = "{required}")  Long empId) throws FebsException {
+        try {
+            this.empAccessoryService.mvFiles(accessoryids,labelid,empId);
+        } catch (Exception e) {
+            String message = "移动文件失败";
+            log.error(message, e);
+            throw new FebsException(message);
+        }
+    }
+    @PostMapping("/singledownload")
+    @PreAuthorize("hasAuthority('empAccessory:singledownload')")
+    public void singledownload(@NotBlank(message = "{required}")  String empIds, @NotNull(message = "{required}")  Long labelid, HttpServletRequest request, HttpServletResponse response) throws FebsException {
+        try {
+            this.empAccessoryService.singledownload(empIds,labelid,request,response);
+        } catch (Exception e) {
+            String message = "下载文件失败";
+            log.error(message, e);
+            throw new FebsException(message);
+        }
+    }
 }
diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/FilesUploadController.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/FilesUploadController.java
index f8d469d..d5d2834 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/FilesUploadController.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/FilesUploadController.java
@@ -103,7 +103,7 @@
         try {
             this.filesUploadService.mvFiles(fileids,folderid);
         } catch (Exception e) {
-            String message = "修改FilesUpload失败";
+            String message = "移动文件失败";
             log.error(message, e);
             throw new FebsException(message);
         }
diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/IEmpAccessoryService.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/IEmpAccessoryService.java
index 6136045..b6a5da9 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/IEmpAccessoryService.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/IEmpAccessoryService.java
@@ -11,6 +11,7 @@
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.validation.constraints.NotNull;
 import java.io.IOException;
 import java.util.List;
 import java.util.Map;
@@ -58,4 +59,8 @@
     void deleteEmpAccessory(String accessoryids);
 
     void download(String accessoryids, HttpServletRequest request, HttpServletResponse response);
+
+    void mvFiles(String accessoryids,Long labelid,Long empId);
+
+    void singledownload(String empIds, Long labelid, HttpServletRequest request, HttpServletResponse response) throws Exception;
 }
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 1442e08..7f21cca 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
@@ -12,11 +12,13 @@
 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;
@@ -46,7 +48,7 @@
 @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())
@@ -205,4 +207,52 @@
 
     }
 
+    @Override
+    public void mvFiles(String accessoryids,Long labelid,Long empId) {
+        String[] arr = accessoryids.split(",");
+
+        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
+        wapper.in(EmpAccessory::getAccessoryid, arr);
+        EmpAccessory empAccessory = new EmpAccessory();
+        empAccessory.setModifier(operatorId);
+        empAccessory.setModifytime(new Date());
+        empAccessory.setLabelid(labelid);
+        empAccessory.setEmpid(empId);
+        this.update(empAccessory,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);
+    }
 }
diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java
index 674daa8..53e37dd 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/service/impl/FilesUploadServiceImpl.java
@@ -159,7 +159,6 @@
         FilesUpload filesUpload = new FilesUpload();
         filesUpload.setModifier(operatorId);
         filesUpload.setModifytime(new Date());
-        filesUpload.setDelFlag(1);
         filesUpload.setFolderid(folderid);
         this.update(filesUpload,wapper);
 

--
Gitblit v1.8.0