From f33f56fb2ebfea915b93467698ca6b243ee934ba Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期二, 16 七月 2024 17:32:49 +0800
Subject: [PATCH] 导出完善

---
 src/main/java/com/mindskip/xzs/service/impl/ExamPaperServiceImpl.java |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/mindskip/xzs/service/impl/ExamPaperServiceImpl.java b/src/main/java/com/mindskip/xzs/service/impl/ExamPaperServiceImpl.java
index aa48ad1..bda4cdf 100644
--- a/src/main/java/com/mindskip/xzs/service/impl/ExamPaperServiceImpl.java
+++ b/src/main/java/com/mindskip/xzs/service/impl/ExamPaperServiceImpl.java
@@ -281,17 +281,25 @@
     @Override
     public List<PaperExcelVO> getRandomPaperExcelById(Integer id) {
         List<PaperExcelVO> paperExcel = examPaperMapper.getRandomPaperExcelById(id);
+        // 濡傛灉鏌愯�冪敓瀛樺湪澶氫釜閮ㄩ棬锛屾瘡涓儴闂ㄩ兘鍔犱竴閬嶆暟鎹�
+        List<PaperExcelVO> needAdd = new ArrayList<>();
+        List<PaperExcelVO> needRemove = new ArrayList<>();
         paperExcel = paperExcel.stream().map(e->{
             e.setPaperScore(ExamUtil.scoreToVM(Integer.parseInt(e.getPaperScore())));
             e.setUserScore(ExamUtil.scoreToVM(Integer.parseInt(e.getUserScore())));
             List<Department> userDepartments = userDepartmentMapper.selectDeptByUserId(Integer.parseInt(e.getUserId()));
-            if(userDepartments.size() != 0) {
-                Department department = userDepartments.get(0);
-                e.setDepartmentName(department.getName());
-                e.setDeptId(department.getId());
+            for (Department dept : userDepartments) {
+                PaperExcelVO vo = new PaperExcelVO();
+                BeanUtils.copyProperties(e, vo);
+                vo.setDepartmentName(dept.getName());
+                vo.setDeptId(dept.getId());
+                needAdd.add(vo);
+                needRemove.add(e);
             }
             return e;
         }).collect(Collectors.toList());
+        paperExcel.removeAll(needRemove);
+        paperExcel.addAll(needAdd);
         return paperExcel;
     }
 

--
Gitblit v1.8.0