From 98f527638b35bce0d513a62ce2d954c14b10774d Mon Sep 17 00:00:00 2001
From: 孔祥富 <kongxf@daryun.com>
Date: 星期日, 07 三月 2021 22:39:29 +0800
Subject: [PATCH] 提交

---
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpUnemploymentController.java |   22 +++++++++++++++++++++-
 1 files changed, 21 insertions(+), 1 deletions(-)

diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpUnemploymentController.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpUnemploymentController.java
index 12cc8b5..e142c28 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpUnemploymentController.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpUnemploymentController.java
@@ -1,11 +1,16 @@
 package cc.mrbird.febs.server.hr.controller;
 
+import cc.mrbird.febs.server.hr.annotation.ControllerEndpoint;
+import cc.mrbird.febs.server.hr.constant.MyConstant;
+import cc.mrbird.febs.server.hr.entity.EmpResign;
 import cc.mrbird.febs.server.hr.entity.EmpUnemployment;
 import cc.mrbird.febs.server.hr.service.IEmpUnemploymentService;
 import cc.mrbird.febs.common.core.entity.FebsResponse;
 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 cn.hutool.core.util.StrUtil;
+import com.wuwenze.poi.ExcelKit;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
@@ -14,8 +19,10 @@
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
 import javax.validation.constraints.NotBlank;
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -49,7 +56,14 @@
         Map<String, Object> dataTable = FebsUtil.getDataTable(this.empUnemploymentService.findEmpUnemployments(request, empUnemployment));
         return new FebsResponse().data(dataTable);
     }
-
+    @ApiOperation(value = "导出员工")
+    @RequestMapping(value = "export",method= RequestMethod.POST)
+    @ControllerEndpoint(operation = "导出用户数据", exceptionMessage = "导出Excel失败")
+    public void export(QueryRequest request, EmpUnemployment empBaseinfo, HttpServletResponse response) {
+        request.setPageSize(this.empUnemploymentService.count());
+        List<EmpUnemployment> records = this.empUnemploymentService.findEmpUnemployments(request, empBaseinfo).getRecords();
+        ExcelKit.$Export(EmpUnemployment.class, response).downXlsx(records, false);
+    }
     @PostMapping
     @PreAuthorize("hasAuthority('empUnemployment:add')")
     public void addEmpUnemployment(@Valid EmpUnemployment empUnemployment) throws FebsException {
@@ -57,6 +71,9 @@
             this.empUnemploymentService.createEmpUnemployment(empUnemployment);
         } catch (Exception e) {
             String message = "新增失业金领取表失败";
+            if (StrUtil.equals(e.getMessage(), MyConstant.MESSAGE)){
+                message = e.getMessage();
+            }
             log.error(message, e);
             throw new FebsException(message);
         }
@@ -93,6 +110,9 @@
             this.empUnemploymentService.updateEmpUnemployment(empUnemployment);
         } catch (Exception e) {
             String message = "修改失业金领取表失败";
+            if (StrUtil.equals(e.getMessage(), MyConstant.MESSAGE)){
+                message = e.getMessage();
+            }
             log.error(message, e);
             throw new FebsException(message);
         }

--
Gitblit v1.8.0