From 5e873863bc7324894c89b0a0fc77a6202c91cd64 Mon Sep 17 00:00:00 2001
From: wzp <516075396@qq.com>
Date: 星期二, 06 四月 2021 10:35:42 +0800
Subject: [PATCH] 20210406

---
 src/main/java/com/ots/project/exam/controller/EntTestPackageController.java |  113 +++++++++++++++++++-------------------------------------
 1 files changed, 39 insertions(+), 74 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 b8b107b..cb71242 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;
@@ -11,6 +10,7 @@
 import com.ots.framework.web.domain.AjaxResult;
 import com.ots.framework.web.page.TableDataInfo;
 import com.ots.project.exam.domain.EntDemographyParam;
+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.IEntDemographyParamService;
@@ -28,46 +28,31 @@
 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;
 
-/**
- * 娴嬭瘯鍖呯鐞咰ontroller
- *
- * @author ots
- * @date 2019-11-27
- */
 @Controller
 @RequestMapping("/exam/test_package")
 public class EntTestPackageController extends BaseController {
     private String prefix = "exam/test_package";
-
     @Autowired
     private IEntTestPackageService entTestPackageService;
-
     @Autowired
     private IEntDemographyParamService entDemographyParamService;
-
     @Autowired
     private ISysUserExtendService sysUserExtendService;
-
     @Autowired
     private ITReportTemplateService reportTemplateService;
-
     @RequiresPermissions("exam:test_package:view")
     @GetMapping()
     public String test_package() {
         return prefix + "/test_package";
     }
-
-    /**
-     * 鏌ヨ娴嬭瘯鍖呯鐞嗗垪琛�
-     */
+    
     @RequiresPermissions("exam:test_package:list")
     @PostMapping("/list")
     @ResponseBody
@@ -79,7 +64,7 @@
             if (UserTypeEnum.ENT_USER.getUserType().equals(userType)) {
                 entTestPackage.setUserId(sysUser.getUserId());
             }
-            //鍒嗛攢鍟嗗彲浠ョ湅鍒板垎閿�鍟嗕笅鐨勪紒涓氫俊鎭�
+            
             if (UserTypeEnum.DIS_USER.getUserType().equals(userType)) {
                 SysUserExtend sysUserExtend = new SysUserExtend();
                 sysUserExtend.setParentUserId(sysUser.getUserId());
@@ -91,11 +76,7 @@
         }
         return getDataTable(entTestPackageService.selectEntTestPackageList(entTestPackage));
     }
-
-
-    /**
-     * 鏌ヨ娴嬭瘯鍖呯鐞嗗垪琛�
-     */
+    
     @PostMapping("/listTopTestPackage")
     @ResponseBody
     public TableDataInfo listTopTestPackage(EntTestPackage entTestPackage) {
@@ -104,43 +85,53 @@
         if (!SYS_USER.getUserType().equals(sysUser.getUserType())) {
             entTestPackage.setUserId(sysUser.getUserId());
         }
-        return getDataTable(entTestPackageService.selectEntTestPackageMainList(entTestPackage));
-    }
 
-    /**
-     * 瀵煎嚭娴嬭瘯鍖呯鐞嗗垪琛�
-     */
+        return getDataTable(entTestPackageService.selectEntTestPackageList(entTestPackage));
+    }
+    
     @RequiresPermissions("exam:test_package:export")
     @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");
     }
-
-    /**
-     * 鍥炴敹浜烘锛堝洖鏀跺瘑鐮侊級
-     */
+    
     @RequiresPermissions("exam:test_package:recover")
     @PostMapping("/recover/{testId}")
     @ResponseBody
     public AjaxResult recover(@PathVariable Long testId) {
         return toAjax(entTestPackageService.recover(testId));
     }
-
-    /**
-     * 鏂板娴嬭瘯鍖呯鐞�
-     */
+    
     @GetMapping("/add")
     public String add(ModelMap mmap) {
-        //澶氳瑷�閫夋嫨
+        
         EntDemographyParam entDemographyParam = new EntDemographyParam();
         entDemographyParam.setParamStatus("1");
         List<EntDemographyParam> list = entDemographyParamService.selectEntDemographyParamList(entDemographyParam);
         mmap.put("roles", list);
-        //鎶ュ憡妯℃澘缂栫爜  t_report_template
-//        mmap.put("templateList", reportTemplateService.getReportTemplates(ShiroUtils.getSysUser()));
+        
+
         mmap.put("template", getTemplate());
         SysUserExtend sysUserExtend = sysUserExtendService.selectSysUserExtendById(ShiroUtils.getUserId());
         String mailContent = sysUserExtend.getMailContent();
@@ -150,7 +141,6 @@
         mmap.put("hrTemplate", mailContent);
         return prefix + "/add";
     }
-
     @NotNull
     private String getHrTemplate() {
         String template = "浜茬埍鐨剓Company}浜哄姏璧勬簮閮紝\n" +
@@ -176,7 +166,6 @@
                 "喔澿箞喔侧涪喔氞福喔脆竵喔侧福喔ム腹喔佮竸喙夃覆喔副喔∴笧喔编笝喔樴箤喔氞福喔脆俯喔� TAI";
         return template;
     }
-
     @NotNull
     private String getTemplate() {
         String template = "浜茬埍鐨剓FullName},\n" +
@@ -203,19 +192,13 @@
                 "<br>";
         return template;
     }
-
-    /**
-     * 鏍规嵁浜у搧鍖匢D鑾峰彇妯℃澘缂栫爜
-     */
+    
     @PostMapping("/prodTemplate")
     @ResponseBody
     public AjaxResult changeTemplate(Long prodId) {
         return AjaxResult.success(MessageUtils.message("basis.successfulOperation"), reportTemplateService.getReportTemplates(ShiroUtils.getSysUser(), prodId));
     }
-
-    /**
-     * 鏂板淇濆瓨娴嬭瘯鍖呯鐞�
-     */
+    
     @RequiresPermissions("exam:test_package:add")
     @Log(title = "娴嬭瘯鍖呯鐞�", businessType = BusinessType.INSERT)
     @PostMapping("/add")
@@ -223,22 +206,15 @@
     public AjaxResult addSave(EntTestPackage entTestPackage) {
         return toAjax(entTestPackageService.insertEntTestPackage(entTestPackage));
     }
-
-    /**
-     * 淇敼娴嬭瘯鍖呯鐞�
-     */
+    
     @GetMapping("/edit/{id}")
     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
-         */
+        
         if (Objects.nonNull(paramCodes)) {
             String[] split = paramCodes.split(",");
             list.forEach(p -> {
@@ -247,9 +223,7 @@
                 });
             });
         }
-        /**
-         * 瀛樺湪鏁版嵁鍒欐洿鏂颁负true
-         */
+        
         if (Objects.nonNull(paramCodes)) {
             String[] split = paramCodes.split(",");
             list.forEach(p -> {
@@ -261,15 +235,11 @@
             });
         }
         mmap.put("roles", list);
-
-        //鎶ュ憡妯℃澘缂栫爜  t_report_template
+        
         mmap.put("templateList", reportTemplateService.getReportTemplates(ShiroUtils.getSysUser(), entTestPackage.getProdId()));
         return prefix + "/edit";
     }
-
-    /**
-     * 淇敼淇濆瓨娴嬭瘯鍖呯鐞�
-     */
+    
     @RequiresPermissions("exam:test_package:edit")
     @Log(title = "娴嬭瘯鍖呯鐞�", businessType = BusinessType.UPDATE)
     @PostMapping("/edit")
@@ -277,10 +247,7 @@
     public AjaxResult editSave(EntTestPackage entTestPackage) {
         return toAjax(entTestPackageService.updateEntTestPackage(entTestPackage));
     }
-
-    /**
-     * 鍒犻櫎娴嬭瘯鍖呯鐞�
-     */
+    
     @RequiresPermissions("exam:test_package:remove")
     @Log(title = "娴嬭瘯鍖呯鐞�", businessType = BusinessType.DELETE)
     @PostMapping("/remove")
@@ -288,6 +255,4 @@
     public AjaxResult remove(String ids) {
         return toAjax(entTestPackageService.deleteEntTestPackageByIds(ids));
     }
-
-
 }

--
Gitblit v1.9.1