From 55cd1ac1b843cc6c19100d7e23b4276e4a8bc5a3 Mon Sep 17 00:00:00 2001
From: luoyb <412940104@qq.com>
Date: 星期四, 11 三月 2021 22:57:24 +0800
Subject: [PATCH] fix(bug修复): 排序

---
 febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpOccupationalController.java |    7 ++++++-
 1 files changed, 6 insertions(+), 1 deletions(-)

diff --git a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpOccupationalController.java b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpOccupationalController.java
index 87bd791..9034ef7 100644
--- a/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpOccupationalController.java
+++ b/febs-server/febs-server-hr/src/main/java/cc/mrbird/febs/server/hr/controller/EmpOccupationalController.java
@@ -1,6 +1,7 @@
 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.EmpLeaveInfo;
 import cc.mrbird.febs.server.hr.entity.EmpOccupational;
 import cc.mrbird.febs.server.hr.service.IEmpOccupationalService;
@@ -8,6 +9,7 @@
 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;
@@ -58,7 +60,7 @@
     @RequestMapping(value = "export",method= RequestMethod.POST)
     @ControllerEndpoint(operation = "导出用户数据", exceptionMessage = "导出Excel失败")
     public void export(QueryRequest request, EmpOccupational empBaseinfo, HttpServletResponse response) {
-
+        request.setPageSize(this.empOccupationalService.count());
         List<EmpOccupational> records = this.empOccupationalService.findEmpOccupationals(request, empBaseinfo).getRecords();
         ExcelKit.$Export(EmpOccupational.class, response).downXlsx(records, false);
     }
@@ -69,6 +71,9 @@
             this.empOccupationalService.createEmpOccupational(empOccupational);
         } 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