From d7abd9ebd3adf6bd7adac595fd1c070568432acb Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期五, 21 六月 2024 09:01:23 +0800
Subject: [PATCH] 人员考评

---
 performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java |   70 +++++++++++++++++++++++++++++++++++
 1 files changed, 70 insertions(+), 0 deletions(-)

diff --git a/performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java b/performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java
index a90ea43..470fb51 100644
--- a/performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java
+++ b/performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java
@@ -1,10 +1,31 @@
 package com.yuanchu.mom.service.impl;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
+import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.yuanchu.mom.common.GetLook;
+import com.yuanchu.mom.common.PrintChina;
+import com.yuanchu.mom.dto.AuxiliaryOriginalHoursDto;
 import com.yuanchu.mom.pojo.Evaluate;
 import com.yuanchu.mom.mapper.EvaluateMapper;
 import com.yuanchu.mom.service.EvaluateService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.yuanchu.mom.utils.QueryWrappers;
 import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 /**
  * <p>
@@ -17,4 +38,53 @@
 @Service
 public class EvaluateServiceImpl extends ServiceImpl<EvaluateMapper, Evaluate> implements EvaluateService {
 
+    @Resource
+    EvaluateMapper evaluateMapper;
+
+    @Resource
+    GetLook getLook;
+
+    @Override
+    public Map<String, Object> getPage(Page page, Evaluate evaluate) {
+        Map<String, Object> map = new HashMap<>();
+        map.put("head", PrintChina.printChina(Evaluate.class));
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("getPage");
+        if (map1.get("look") == 1) evaluate.setCreateUser(map1.get("userId"));
+        map.put("body", evaluateMapper.getPage(page, QueryWrappers.queryWrappers(evaluate)));
+        return map;
+    }
+
+    //瀵煎嚭鑰冭瘎
+    @Override
+    public void export(String month, String name, String departLims, HttpServletResponse response) throws IOException {
+        List<Evaluate> evaluates = new ArrayList<Evaluate>();
+        //鏌ヨ鍘熷宸ユ椂(浣跨敤鍒嗛〉鏌ヨ)
+        Evaluate evaluate = new Evaluate();
+        if (ObjectUtils.isNotEmpty(month)) {
+            evaluate.setMonth(month);
+        }
+        if (ObjectUtils.isNotEmpty(name)) {
+            evaluate.setName(name);
+        }
+        if (ObjectUtils.isNotEmpty(departLims)) {
+            evaluate.setDepartLims(departLims);
+        }
+        IPage<Evaluate> body = (IPage<Evaluate>) getPage(new Page(-1, -1), evaluate).get("body");
+        evaluates = body.getRecords();
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("UTF-8");
+        // 杩欓噷URLEncoder.encode鍙互闃叉涓枃涔辩爜 褰撶劧鍜宔asyexcel娌℃湁鍏崇郴
+        String fileName = URLEncoder.encode("鑰冭瘎瀵煎嚭", "UTF-8");
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+        try {
+            // 鏂板缓ExcelWriter
+            ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();
+            WriteSheet mainSheet = EasyExcel.writerSheet(0, "鑰冭瘎瀵煎嚭").head(Evaluate.class).build();
+            excelWriter.write(evaluates, mainSheet);
+            // 鍏抽棴娴�
+            excelWriter.finish();
+        } catch (IOException e) {
+            throw new RuntimeException("瀵煎嚭澶辫触");
+        }
+    }
 }

--
Gitblit v1.9.3