From 3a8e1524dcf0eeb610d38123d5f0a3ef838379cd Mon Sep 17 00:00:00 2001
From: linzhijie <19970921lzj>
Date: 星期一, 12 四月 2021 15:44:53 +0800
Subject: [PATCH] Merge branch '20210406-maqv2' into 20210412-youhua

---
 src/main/java/com/ots/project/exam/controller/EntTestPackageController.java |   25 ++++++++++++++++++++-----
 1 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/ots/project/exam/controller/EntTestPackageController.java b/src/main/java/com/ots/project/exam/controller/EntTestPackageController.java
index cf80d92..cd3da43 100644
--- a/src/main/java/com/ots/project/exam/controller/EntTestPackageController.java
+++ b/src/main/java/com/ots/project/exam/controller/EntTestPackageController.java
@@ -1,5 +1,4 @@
 package com.ots.project.exam.controller;
-
 import com.ots.common.enums.UserTypeEnum;
 import com.ots.common.utils.MessageUtils;
 import com.ots.common.utils.StringUtils;
@@ -12,6 +11,7 @@
 import com.ots.framework.web.page.TableDataInfo;
 import com.ots.project.exam.domain.EntDemographyParam;
 import com.ots.project.exam.domain.EntEmailTemplate;
+import com.ots.project.exam.domain.EntTestMember;
 import com.ots.project.exam.domain.EntTestPackage;
 import com.ots.project.exam.domain.SysUserExtend;
 import com.ots.project.exam.service.*;
@@ -26,12 +26,11 @@
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
-
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Objects;
 import java.util.stream.Collectors;
-
 import static com.ots.common.enums.UserTypeEnum.SYS_USER;
 
 /**
@@ -124,7 +123,25 @@
     @PostMapping("/export")
     @ResponseBody
     public AjaxResult export(EntTestPackage entTestPackage) {
+        User sysUser = ShiroUtils.getSysUser();
+        String userType = sysUser.getUserType();
+        List<Long> userIds = new ArrayList<>();
+        userIds.add(sysUser.getUserId());
+
+        if (UserTypeEnum.DIS_USER.getUserType().equals(userType)) {
+            SysUserExtend sysUserExtend = new SysUserExtend();
+            sysUserExtend.setParentUserId(sysUser.getUserId());
+            List<SysUserExtend> sysUserExtends = sysUserExtendService.selectSysUserExtendList(sysUserExtend);
+            for (int i = 0; i < sysUserExtends.size(); i++) {
+                SysUserExtend extend = sysUserExtends.get(i);
+                userIds.add(extend.getUserId());
+            }
+        }
         List<EntTestPackage> list = entTestPackageService.selectEntTestPackageList(entTestPackage);
+        if (!SYS_USER.getUserType().equals(userType)) {
+
+            list = list.stream().filter(p -> userIds.contains(p.getUserId())).collect(Collectors.toList());
+        }
         ExcelUtil<EntTestPackage> util = new ExcelUtil<EntTestPackage>(EntTestPackage.class);
         return util.exportExcel(list, "test_package");
     }
@@ -188,10 +205,8 @@
     public String edit(@PathVariable("id") Long id, ModelMap mmap) {
         EntTestPackage entTestPackage = entTestPackageService.selectEntTestPackageById(id);
         mmap.put("entTestPackage", entTestPackage);
-
         EntDemographyParam entDemographyParam = new EntDemographyParam();
         List<EntDemographyParam> list = entDemographyParamService.selectEntDemographyParamList(entDemographyParam);
-
         String paramCodes = entTestPackage.getParamCodes();
         /**
          * 榛樿鍏ㄩ儴涓篺alse

--
Gitblit v1.9.1