From f31ae33e37895b9517ee6ace6bd703f4bbb47e81 Mon Sep 17 00:00:00 2001
From: value <z1292839451@163.com>
Date: 星期日, 23 六月 2024 23:44:30 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 performance-server/src/main/java/com/yuanchu/mom/service/impl/EvaluateServiceImpl.java |   51 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 49 insertions(+), 2 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 6d9ec81..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,8 +1,16 @@
 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;
@@ -11,7 +19,12 @@
 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;
 
 /**
@@ -36,8 +49,42 @@
         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)));
+        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