From b9da8289051937fbbbd47bbda97414ad6c769a90 Mon Sep 17 00:00:00 2001
From: hjl <1657978663@qq.com>
Date: 星期四, 18 七月 2024 14:37:00 +0800
Subject: [PATCH] feat: 代码重构

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/UserController.java |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/UserController.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/UserController.java
index 381e785..abb6c0e 100644
--- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/UserController.java
+++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/UserController.java
@@ -8,6 +8,8 @@
 import com.ruoyi.admin.service.UserService;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.utils.StringUtils;
+import com.ruoyi.common.security.service.TokenService;
+import com.ruoyi.system.api.model.LoginUser;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -34,6 +36,8 @@
 
     @Resource
     private UserService userService;
+    @Resource
+    private TokenService tokenService;
 
     /**
      * 用户信息分页列表
@@ -53,12 +57,19 @@
     public R<IPage<User>> queryPageList(String nickname, String phone, Integer state,
                                         @RequestParam(name = "pageNum", defaultValue = "1") Integer pageNum,
                                         @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize) {
+        LoginUser loginUser = tokenService.getLoginUser();
+        if (null == loginUser) {
+            return R.loginExpire("登录失效!");
+        }
         LambdaQueryChainWrapper<User> query = userService.lambdaQuery();
         // 穆湖匹配规则
         query = StringUtils.isNotBlank(nickname) ? query.like(User::getNickname, nickname) : query;
         query = StringUtils.isNotBlank(phone) ? query.like(User::getPhone, phone) : query;
         // 账号是否启用
         query = null != state ? query.eq(User::getState, state) : query;
+        if (loginUser.getIsFranchisee() && !loginUser.getCityList().isEmpty()) {
+            query.in(User::getCity, loginUser.getCityList());
+        }
         return R.ok(query.orderByDesc(User::getCreateTime).eq(User::getIsDelete, 0)
                 .orderByDesc(User::getCreateTime).page(Page.of(pageNum, pageSize)));
     }

--
Gitblit v1.7.1