From 5e179f4180fd2dc93623699af614df9a2497ded6 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 25 Sep 2025 02:27:38 +0800
Subject: [PATCH] 1
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/user/UserController.java | 165 +++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 155 insertions(+), 10 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/user/UserController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/user/UserController.java
index 94ce615..d64eae3 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/user/UserController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/user/UserController.java
@@ -23,14 +23,10 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
-import java.util.Collection;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
+import java.util.stream.Collectors;
@RestController
@RequestMapping("/im/user")
@@ -46,12 +42,10 @@
/**
* 获取会员列表
*/
-// @PreAuthorize("@ss.hasPermi('im:user:list')")
@GetMapping("/list")
public TableDataInfo list(UserAccountVo vo)
{
// 创建查询条件包装器
-
LambdaQueryWrapper<UserAccount> queryWrapper = new LambdaQueryWrapper<>();
// 只有当 keyword 不为空时才添加 OR 条件
@@ -71,6 +65,7 @@
queryWrapper
.eq(ObjectUtil.isNotEmpty(vo.getAccountType()), UserAccount::getAccountType, vo.getAccountType())
.eq(ObjectUtil.isNotEmpty(vo.getStatus()), UserAccount::getStatus, vo.getStatus())
+ .eq(ObjectUtil.isNotEmpty(vo.getPosition()), UserAccount::getPosition, vo.getPosition())
.between(ObjectUtil.isAllNotEmpty(vo.getStartTime(), vo.getEndTime()),
UserAccount::getCreateTime, vo.getStartTime(), vo.getEndTime());
@@ -81,7 +76,9 @@
PageInfo<UserAccount> pageInfo = new PageInfo<>(list);
- List<UserAccountOut> toList = ConverterUtil.convertToList(list, UserAccountOut.class);
+ // 转换为输出对象并递归加载下级列表（最多3级）
+ List<UserAccountOut> toList = convertToUserAccountOutWithSubordinates(list, 3);
+
TableDataInfo rspData = new TableDataInfo();
rspData.setCode(HttpStatus.SUCCESS);
rspData.setMsg("查询成功");
@@ -91,6 +88,154 @@
}
/**
+ * 转换用户账户列表并递归加载下级用户
+ * @param userAccounts 用户账户列表
+ * @param maxLevel 最大递归层级
+ * @return 包含下级列表的用户输出对象列表
+ */
+ private List<UserAccountOut> convertToUserAccountOutWithSubordinates(List<UserAccount> userAccounts, int maxLevel) {
+ if (ObjectUtil.isEmpty(userAccounts) || maxLevel <= 0) {
+ return new ArrayList<>();
+ }
+
+ // 先转换基础信息
+ List<UserAccountOut> result = ConverterUtil.convertToList(userAccounts, UserAccountOut.class);
+
+ // 递归加载下级用户
+ loadSubordinateUsersRecursive(result, maxLevel);
+
+ return result;
+ }
+
+ /**
+ * 递归加载多级下级用户
+ * @param userAccountOuts 当前层级的用户列表
+ * @param remainingLevels 剩余递归层级
+ */
+ private void loadSubordinateUsersRecursive(List<UserAccountOut> userAccountOuts, int remainingLevels) {
+ if (ObjectUtil.isEmpty(userAccountOuts) || remainingLevels <= 0) {
+ return;
+ }
+
+ // 加载当前级别的下级用户
+ loadCurrentLevelSubordinates(userAccountOuts);
+
+ // 递归加载下级的下级
+ for (UserAccountOut user : userAccountOuts) {
+ if (ObjectUtil.isNotEmpty(user.getSubordinateList())) {
+ loadSubordinateUsersRecursive(user.getSubordinateList(), remainingLevels - 1);
+ }
+ }
+ }
+
+ /**
+ * 加载当前层级的直接下级用户
+ * @param userAccountOuts 当前层级的用户列表
+ */
+ private void loadCurrentLevelSubordinates(List<UserAccountOut> userAccountOuts) {
+ if (ObjectUtil.isEmpty(userAccountOuts)) {
+ return;
+ }
+
+ // 收集所有用户的账号（用于查询下级）
+ List<String> accounts = userAccountOuts.stream()
+ .map(UserAccountOut::getAccount)
+ .filter(ObjectUtil::isNotEmpty)
+ .distinct()
+ .collect(Collectors.toList());
+
+ if (accounts.isEmpty()) {
+ // 如果没有账号，为所有用户设置空列表
+ userAccountOuts.forEach(user -> user.setSubordinateList(new ArrayList<>()));
+ return;
+ }
+
+ // 批量查询所有直接下级用户
+ LambdaQueryWrapper<UserAccount> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.in(UserAccount::getInvitationAccount, accounts);
+ List<UserAccount> allSubordinates = userAccountService.list(queryWrapper);
+
+ // 转换为输出对象
+ List<UserAccountOut> allSubordinateOuts = ConverterUtil.convertToList(allSubordinates, UserAccountOut.class);
+
+ // 按邀请人账号分组
+ Map<String, List<UserAccountOut>> subordinateMap = allSubordinateOuts.stream()
+ .filter(sub -> ObjectUtil.isNotEmpty(sub.getInvitationAccount()))
+ .collect(Collectors.groupingBy(UserAccountOut::getInvitationAccount));
+
+ // 为每个用户设置下级列表
+ for (UserAccountOut user : userAccountOuts) {
+ if (ObjectUtil.isNotEmpty(user.getAccount())) {
+ List<UserAccountOut> subordinates = subordinateMap.get(user.getAccount());
+ user.setSubordinateList(ObjectUtil.isNotEmpty(subordinates) ? subordinates : new ArrayList<>());
+ } else {
+ user.setSubordinateList(new ArrayList<>());
+ }
+ }
+ }
+
+
+
+ /**
+ * 获取用户的下级树形结构
+ * @param userId 用户ID
+ * @param maxLevel 最大层级深度
+ * @return 用户下级树形结构
+ */
+ @GetMapping("/subordinateTree/{userId}")
+ public AjaxResult getSubordinateTree(@PathVariable Integer userId,
+ @RequestParam(defaultValue = "3") int maxLevel) {
+ try {
+ UserAccount userAccount = userAccountService.getById(userId);
+ if (ObjectUtil.isEmpty(userAccount)) {
+ return AjaxResult.error("用户不存在");
+ }
+
+ // 转换当前用户
+ UserAccountOut userOut = ConverterUtil.convert(userAccount, UserAccountOut.class);
+
+ // 递归加载下级树
+ List<UserAccountOut> userList = new ArrayList<>();
+ userList.add(userOut);
+ loadSubordinateUsersRecursive(userList, maxLevel);
+
+ return AjaxResult.success(userOut);
+ } catch (Exception e) {
+ log.error("获取用户下级树失败", e);
+ return AjaxResult.error("获取下级树失败");
+ }
+ }
+
+ /**
+ * 获取用户的直接下级列表（不分页）
+ */
+ @GetMapping("/directSubordinates/{userId}")
+ public AjaxResult getDirectSubordinates(@PathVariable Integer userId) {
+ try {
+ UserAccount userAccount = userAccountService.getById(userId);
+ if (ObjectUtil.isEmpty(userAccount)) {
+ return AjaxResult.error("用户不存在");
+ }
+
+ // 查询直接下级
+ LambdaQueryWrapper<UserAccount> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.eq(UserAccount::getInvitationAccount, userAccount.getAccount());
+ List<UserAccount> subordinates = userAccountService.list(queryWrapper);
+
+ List<UserAccountOut> result = ConverterUtil.convertToList(subordinates, UserAccountOut.class);
+
+ return AjaxResult.success(result);
+ } catch (Exception e) {
+ log.error("获取直接下级列表失败", e);
+ return AjaxResult.error("获取下级列表失败");
+ }
+ }
+
+
+
+
+
+ /**
* 修改会员
*/
// @PreAuthorize("@ss.hasPermi('im:user:updateUserAccount')")
--
Gitblit v1.9.3