From 065f6d3555d4125b6045ca2bb66d38030ba125a9 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期一, 15 七月 2024 09:03:39 +0800
Subject: [PATCH] 统计导出(待测试)

---
 src/main/java/com/mindskip/xzs/controller/admin/ExamPaperAnswerController.java |   51 +++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 37 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/mindskip/xzs/controller/admin/ExamPaperAnswerController.java b/src/main/java/com/mindskip/xzs/controller/admin/ExamPaperAnswerController.java
index 5ed2c9f..dc7628d 100644
--- a/src/main/java/com/mindskip/xzs/controller/admin/ExamPaperAnswerController.java
+++ b/src/main/java/com/mindskip/xzs/controller/admin/ExamPaperAnswerController.java
@@ -1,9 +1,13 @@
 package com.mindskip.xzs.controller.admin;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
 import com.alibaba.fastjson.JSONObject;
 import com.github.pagehelper.PageInfo;
 import com.mindskip.xzs.base.BaseApiController;
 import com.mindskip.xzs.base.RestResponse;
+import com.mindskip.xzs.domain.Department;
 import com.mindskip.xzs.domain.ExamPaperAnswer;
 import com.mindskip.xzs.domain.ExamPaperSubject;
 import com.mindskip.xzs.domain.Subject;
@@ -11,16 +15,16 @@
 import com.mindskip.xzs.domain.vo.ExamPaperDataExportVO;
 import com.mindskip.xzs.domain.vo.ExamPaperDataVO;
 import com.mindskip.xzs.domain.vo.ExamPaperStatisticVO;
-import com.mindskip.xzs.service.ExamPaperAnswerService;
-import com.mindskip.xzs.service.ExamPaperSubjectService;
-import com.mindskip.xzs.service.SubjectService;
-import com.mindskip.xzs.service.UserService;
+import com.mindskip.xzs.domain.vo.PaperExcelVO;
+import com.mindskip.xzs.service.*;
 import com.mindskip.xzs.utility.DateTimeUtil;
 import com.mindskip.xzs.utility.ExamUtil;
 import com.mindskip.xzs.utility.PageInfoHelper;
 import com.mindskip.xzs.utility.excel.ExcelUtils;
 import com.mindskip.xzs.viewmodel.admin.paper.ExamPaperAnswerPageRequestVM;
 import com.mindskip.xzs.viewmodel.student.exampaper.ExamPaperAnswerPageResponseVM;
+import lombok.RequiredArgsConstructor;
+import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.ObjectUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -30,12 +34,14 @@
 
 import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
+import java.io.IOException;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
 
+@RequiredArgsConstructor
 @RestController("AdminExamPaperAnswerController")
 @RequestMapping(value = "/api/admin/examPaperAnswer")
 public class ExamPaperAnswerController extends BaseApiController {
@@ -44,14 +50,8 @@
     private final SubjectService subjectService;
     private final UserService userService;
     private final ExamPaperSubjectService examPaperSubjectService;
-
-    @Autowired
-    public ExamPaperAnswerController(ExamPaperAnswerService examPaperAnswerService, SubjectService subjectService, UserService userService, ExamPaperSubjectService examPaperSubjectService) {
-        this.examPaperAnswerService = examPaperAnswerService;
-        this.subjectService = subjectService;
-        this.userService = userService;
-        this.examPaperSubjectService = examPaperSubjectService;
-    }
+    private final ExamPaperService examPaperService;
+    private final DepartmentService departmentService;
 
 
     @RequestMapping(value = "/page", method = RequestMethod.POST)
@@ -140,8 +140,31 @@
 
     @RequestMapping(value = "/export", method = RequestMethod.GET)
     public void export(Integer id, Integer type, HttpServletResponse response) {
-        List<ExamPaperDataExportVO> list =  examPaperAnswerService.dataExport(new ExamPaperDataVO().setId(id).setType(type));
-        ExcelUtils.export(response, "鍒嗘暟缁熻", list, ExamPaperDataExportVO.class);
+
+
+        try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).build()) {
+            // 閮ㄩ棬鑰冭瘯缁熻sheet
+            List<ExamPaperDataExportVO> list =  examPaperAnswerService.dataExport(new ExamPaperDataVO().setId(id).setType(type));
+            WriteSheet writeSheet = EasyExcel.writerSheet("閮ㄩ棬鏁版嵁缁熻").head(ExamPaperDataExportVO.class).build();
+            excelWriter.write(list, writeSheet);
+
+            // 鑰冭瘯璇︽儏sheet锛屼竴涓崟浣嶄竴涓猻heet
+            List<PaperExcelVO> paperDetailList = examPaperService.getPaperExcelById(id);
+            List<Integer> deptIds = paperDetailList.stream().map(PaperExcelVO::getDeptId).collect(Collectors.toList());
+            Map<Integer, List<Department>> deptMap = departmentService.gets(deptIds).stream().collect(Collectors.groupingBy(Department::getId));
+            Map<Integer, List<PaperExcelVO>> deptDataMap = paperDetailList.stream().collect(Collectors.groupingBy(PaperExcelVO::getDeptId));
+            deptDataMap.forEach((deptId, paperList) -> {
+                List<Department> departments = deptMap.get(deptId);
+                String name = "";
+                if (CollectionUtils.isNotEmpty(departments)) {
+                    name = departments.get(0).getName();
+                }
+                WriteSheet writeSheetNext = EasyExcel.writerSheet(name).head(PaperExcelVO.class).build();
+                excelWriter.write(paperList, writeSheetNext);
+            });
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
     }
 
 }

--
Gitblit v1.8.0