From a230f6653b7c87c3529bd5640886c22d688509c4 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期一, 26 五月 2025 14:37:48 +0800
Subject: [PATCH] 修改接口

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TExperimentSchemeController.java |   48 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 47 insertions(+), 1 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TExperimentSchemeController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TExperimentSchemeController.java
index 9073e78..fef862b 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TExperimentSchemeController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TExperimentSchemeController.java
@@ -51,8 +51,9 @@
     private final TExperimentDispatchService experimentDispatchService;
     private final TProjectProposalService projectProposalService;
     private final TExperimentDispatchGroupService experimentDispatchGroupService;
+    private final TProjectTeamStaffService projectTeamStaffService;
     @Autowired
-    public TExperimentSchemeController(TExperimentSchemeService experimentSchemeService, TokenService tokenService, ISysUserService sysUserService, SysUserMapper sysUserMapper, TExperimentSchemePersonService experimentSchemePersonService, TExperimentDispatchParticipantsService experimentDispatchParticipantsService, TExperimentDispatchService experimentDispatchService, TProjectProposalService projectProposalService, TExperimentDispatchGroupService experimentDispatchGroupService) {
+    public TExperimentSchemeController(TExperimentSchemeService experimentSchemeService, TokenService tokenService, ISysUserService sysUserService, SysUserMapper sysUserMapper, TExperimentSchemePersonService experimentSchemePersonService, TExperimentDispatchParticipantsService experimentDispatchParticipantsService, TExperimentDispatchService experimentDispatchService, TProjectProposalService projectProposalService, TExperimentDispatchGroupService experimentDispatchGroupService, TProjectTeamStaffService projectTeamStaffService) {
         this.experimentSchemeService = experimentSchemeService;
         this.tokenService = tokenService;
         this.sysUserService = sysUserService;
@@ -62,6 +63,7 @@
         this.experimentDispatchService = experimentDispatchService;
         this.projectProposalService = projectProposalService;
         this.experimentDispatchGroupService = experimentDispatchGroupService;
+        this.projectTeamStaffService = projectTeamStaffService;
     }
 
     /**
@@ -72,6 +74,37 @@
     @PostMapping(value = "/api/t-experiment-scheme/pageList")
     public R<PageInfo<TExperimentSchemeVO>> pageList(@RequestBody String param) {
         TExperimentSchemeQuery query = JSON.parseObject(param, TExperimentSchemeQuery.class);
+        // 获取当前用户
+        Long userId = tokenService.getLoginUser().getUserId();
+        Integer roleType = tokenService.getLoginUser().getUser().getRoleType();
+        if (roleType != 1){
+            query.setUserId(userId);
+            if(roleType ==2){
+                // 查询项目组
+                TProjectTeamStaff projectTeamStaff = projectTeamStaffService.getOne(Wrappers.lambdaQuery(TProjectTeamStaff.class)
+                        .eq(TProjectTeamStaff::getUserId, userId));
+                // 查询项目的工艺工程师id
+                TProjectTeamStaff teamStaff = projectTeamStaffService.getOne(Wrappers.lambdaQuery(TProjectTeamStaff.class)
+                        .eq(TProjectTeamStaff::getTeamId, projectTeamStaff.getTeamId())
+                        .eq(TProjectTeamStaff::getRoleType, 3)
+                        .last("LIMIT 1"));
+                // 查询实验参与人员
+                List<TExperimentDispatchParticipants> experimentDispatchParticipants = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
+                        .eq(TExperimentDispatchParticipants::getUserId, teamStaff.getUserId()));
+                if(!CollectionUtils.isEmpty(experimentDispatchParticipants)){
+                    List<String> dispatchIds = experimentDispatchParticipants.stream().map(TExperimentDispatchParticipants::getDispatchId).distinct().collect(Collectors.toList());
+                    query.setDispatchIds(dispatchIds);
+                }
+            }else {
+                // 查询实验参与人员
+                List<TExperimentDispatchParticipants> experimentDispatchParticipants = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
+                        .eq(TExperimentDispatchParticipants::getUserId, userId));
+                if (!CollectionUtils.isEmpty(experimentDispatchParticipants)) {
+                    List<String> dispatchIds = experimentDispatchParticipants.stream().map(TExperimentDispatchParticipants::getDispatchId).distinct().collect(Collectors.toList());
+                    query.setDispatchIds(dispatchIds);
+                }
+            }
+        }
         return R.ok(experimentSchemeService.pageList(query));
     }
     /**
@@ -82,6 +115,19 @@
     @PostMapping(value = "/api/t-experiment-scheme/auditPageList")
     public R<PageInfo<TExperimentSchemeVO>> auditPageList(@RequestBody String param) {
         TExperimentSchemeQuery query = JSON.parseObject(param, TExperimentSchemeQuery.class);
+        // 获取当前用户
+        Long userId = tokenService.getLoginUser().getUserId();
+        Integer roleType = tokenService.getLoginUser().getUser().getRoleType();
+        if (roleType != 1){
+            query.setUserId(userId);
+            // 查询实验参与人员
+            List<TExperimentDispatchParticipants> experimentDispatchParticipants = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
+                    .eq(TExperimentDispatchParticipants::getUserId, userId));
+            if(!CollectionUtils.isEmpty(experimentDispatchParticipants)){
+                List<String> dispatchIds = experimentDispatchParticipants.stream().map(TExperimentDispatchParticipants::getDispatchId).distinct().collect(Collectors.toList());
+                query.setDispatchIds(dispatchIds);
+            }
+        }
         return R.ok(experimentSchemeService.auditPageList(query));
     }
 

--
Gitblit v1.7.1