From bd6b08e65fe29994c6c6b5430639022f2a4f5e5c Mon Sep 17 00:00:00 2001
From: wenminjie <1036699586@qq.com>
Date: 星期二, 14 七月 2026 11:48:08 +0800
Subject: [PATCH] 修复了已知的BUG

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TProjectGroupController.java |   52 +++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 51 insertions(+), 1 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TProjectGroupController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TProjectGroupController.java
index 1051b0a..2fb8e75 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TProjectGroupController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TProjectGroupController.java
@@ -10,6 +10,7 @@
 import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.framework.web.domain.server.Sys;
+import com.ruoyi.framework.web.service.TokenService;
 import com.ruoyi.system.dto.ChoiceDTO;
 import com.ruoyi.system.dto.SysUserDTO;
 import com.ruoyi.system.dto.TProjectTeamDTO;
@@ -30,7 +31,9 @@
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -50,6 +53,8 @@
     private TProjectTeamStaffService projectTeamStaffService;
     @Resource
     private ISysUserService sysUserService;
+    @Resource
+    private TokenService tokenService;
     /**
      * 获取项目组管理列表
      */
@@ -60,11 +65,56 @@
         TProjectTeamQuery query = JSON.parseObject(param, TProjectTeamQuery.class);
         return R.ok(projectTeamService.pageList(query));
     }
+
+    @ApiOperation(value = "获取当前用户的所属项目组")
+    @PostMapping(value = "/api/t_project_myteam/pageList")
+    public R<List<TProjectTeam>> myTeamPageList() {
+        Long userId = tokenService.getLoginUser().getUserId();
+        List<TProjectTeam> list;
+
+        // 判断是否为管理员(超级管理员)
+        if (tokenService.getLoginUser().getUser().isAdmin()) {
+            // 管理员查询全部
+            list = projectTeamService.lambdaQuery()
+                    .eq(TProjectTeam::getStatus, 1)
+                    .list();
+        } else {
+            // 普通用户查询自己的项目组
+            List<String> projetIds = projectTeamStaffService.lambdaQuery()
+                    .eq(TProjectTeamStaff::getUserId, userId)
+                    .list()
+                    .stream()
+                    .map(TProjectTeamStaff::getTeamId)
+                    .collect(Collectors.toList());
+
+            if (projetIds.isEmpty()) {
+                list = new ArrayList<>();
+            } else {
+                list = projectTeamService.lambdaQuery()
+                        .eq(TProjectTeam::getStatus, 1)
+                        .in(TProjectTeam::getId, projetIds)
+                        .list();
+            }
+        }
+
+        return R.ok(list);
+    }
+
     @ApiOperation(value = "新增报告-获取项目组分页列表")
     @PostMapping(value = "/api/t_project_team/pageList1")
     public R<PageInfo<TProjectTeamVO>> pageList1(@RequestBody String param) {
         TProjectTeamQuery query = JSON.parseObject(param, TProjectTeamQuery.class);
-        return R.ok(projectTeamService.pageList(query));
+        Long userId = tokenService.getLoginUser().getUserId();
+        SysUser sysUser = sysUserService.selectUserById(userId);
+        if (!sysUser.getRoleType().equals("1")){
+            List<String> projetIds = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, userId).list()
+                    .stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
+            if (projetIds.isEmpty()){
+                return R.ok(new PageInfo<>());
+            }
+            query.setProjectIds(projetIds);
+        }
+        return R.ok(projectTeamService.pageList1(query));
     }
 
     /**

--
Gitblit v1.7.1