From 2a2f435d52d358cba6e239e29b586777678c2baf Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期二, 14 一月 2025 14:22:25 +0800
Subject: [PATCH] 修改bug

---
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java |  135 +++++++++++++++++++++++++++++++++++++--------
 1 files changed, 111 insertions(+), 24 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java
index aa126fd..e28674b 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/SysUserController.java
@@ -3,6 +3,10 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.ruoyi.account.api.feignClient.AppUserClient;
+import com.ruoyi.account.api.feignClient.AppUserShopClient;
+import com.ruoyi.account.api.model.AppUser;
+import com.ruoyi.account.api.model.AppUserShop;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.utils.StringUtils;
 import com.ruoyi.common.core.web.controller.BaseController;
@@ -33,6 +37,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import java.time.LocalDateTime;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -77,6 +82,12 @@
     
     @Autowired
     private ISysMenuService menuService;
+
+    @Resource
+    private AppUserClient appUserClient;
+
+    @Resource
+    private AppUserShopClient appUserShopClient;
     
     
     
@@ -112,9 +123,6 @@
         user.setUserName(user.getPhonenumber());
         Long userid = tokenService.getLoginUser().getUserid();
         SysUser sysUser1 = userService.getById(userid);
-        if(sysUser1.getRoleType() == 2){
-            user.setObjectId(sysUser1.getObjectId());
-        }
         if(!org.springframework.util.StringUtils.hasLength(user.getNickName())){
             user.setNickName(user.getPhonenumber());
         }
@@ -124,22 +132,64 @@
         if (StringUtils.isNotEmpty(user.getUserName()) && !userService.checkUserNameUnique(user)) {
             return error("登录账号重复");
         }
-        user.setCreateBy(SecurityUtils.getUsername());
-        user.setPassword(SecurityUtils.encryptPassword("a123456"));
-        user.setRoleType(sysUser1.getRoleType());
-        userService.insertUser(user);
-        SysUserRole sysUserRole = new SysUserRole();
-        sysUserRole.setRoleId(user.getRoleId());
-        sysUserRole.setUserId(user.getUserId());
-        userRoleService.insertSysUserRole(sysUserRole);
-        //添加门店员工关系数据
-        if(2 == user.getRoleType()){
+        //门店员工添加数据,需要判断账号是否存在,共用同一个账号
+        if(2 == sysUser1.getRoleType()){
+            AppUser appUser = appUserClient.getAppUserByPhone1(user.getPhonenumber()).getData();
+            if(null == appUser){
+                return error("请先注册小程序账号");
+            }
+            SysUser one = userService.getOne(new LambdaQueryWrapper<SysUser>().eq(SysUser::getUserName, user.getPhonenumber())
+                    .eq(SysUser::getDelFlag, "0").eq(SysUser::getStatus, "0").eq(SysUser::getRoleType, 2));
+            if(null == one){
+                user.setCreateBy(SecurityUtils.getUsername());
+                user.setPassword(SecurityUtils.encryptPassword("a123456"));
+                user.setRoleType(sysUser1.getRoleType());
+                user.setObjectId(sysUser1.getObjectId());
+                if(null != appUser){
+                    user.setAppUserId(appUser.getId());
+                }
+                userService.insertUser(user);
+                one = user;
+                SysUserRole sysUserRole = new SysUserRole();
+                sysUserRole.setRoleId(user.getRoleId());
+                sysUserRole.setUserId(user.getUserId());
+                userRoleService.insertSysUserRole(sysUserRole);
+
+            }
+            //添加小程序店员关系数据
+            if(null != appUser){
+                List<AppUserShop> data = appUserShopClient.getAppUserShop(appUser.getId()).getData();
+                Optional<AppUserShop> first = data.stream().filter(s -> s.getShopId().equals(sysUser1.getObjectId())).findFirst();
+                if(!first.isPresent()){
+                    AppUserShop appUserShop = new AppUserShop();
+                    appUserShop.setAppUserId(appUser.getId());
+                    appUserShop.setShopId(sysUser1.getObjectId());
+                    appUserShop.setRoleType(2);
+                    appUserShopClient.saveAppUserShop(appUserShop);
+                    appUser.setUserType(2);
+                    appUserClient.editAppUserById(appUser);
+                }
+            }
+
+            //添加门店员工关系数据
             UserShop userShop = new UserShop();
-            userShop.setUserId(user.getUserId());
-            userShop.setShopId(user.getObjectId());
+            userShop.setUserId(one.getUserId());
+            userShop.setShopId(sysUser1.getObjectId());
             userShop.setRoleType(2);
             userShop.setRoleId(user.getRoleId());
+            userShop.setDeptId(user.getDeptId());
+            userShop.setNickName(user.getNickName());
+            userShop.setCreateTime(LocalDateTime.now());
             userShopService.save(userShop);
+        }else{
+            user.setCreateBy(SecurityUtils.getUsername());
+            user.setPassword(SecurityUtils.encryptPassword("a123456"));
+            user.setRoleType(sysUser1.getRoleType());
+            userService.insertUser(user);
+            SysUserRole sysUserRole = new SysUserRole();
+            sysUserRole.setRoleId(user.getRoleId());
+            sysUserRole.setUserId(user.getUserId());
+            userRoleService.insertSysUserRole(sysUserRole);
         }
         return AjaxResult.success();
     }
@@ -174,6 +224,17 @@
         ajax.put("posts", postService.selectPostAll());
         if (StringUtils.isNotNull(userId)) {
             SysUser sysUser = userService.selectUserById(userId);
+            if(sysUser.getRoleType() == 2){
+                Long userid = tokenService.getLoginUser().getUserid();
+                SysUser sysUser1 = userService.selectUserById(userid);
+                UserShop one = userShopService.getOne(new LambdaQueryWrapper<UserShop>().eq(UserShop::getUserId, sysUser.getUserId()).eq(UserShop::getShopId, sysUser1.getObjectId()));
+                sysUser.setDeptId(one.getDeptId());
+                sysUser.setNickName(one.getNickName());
+                SysDept sysDept = deptService.selectDeptById(one.getDeptId());
+                sysUser.setDept(sysDept);
+                sysUser.setRoleId(one.getRoleId());
+            }
+
             ajax.put("data", sysUser);
             ajax.put("postIds", postService.selectPostListByUserId(userId));
             ajax.put("roleIds", sysUser.getRoles().stream().map(SysRole::getRoleId).collect(Collectors.toList()));
@@ -228,16 +289,17 @@
         user.setUpdateBy(SecurityUtils.getUsername());
         user.setUpdateTime(new Date());
         userService.updateUser(user);
-    
+        user = userService.getById(user.getUserId());
         //添加门店员工关系数据
         if(2 == user.getRoleType()){
-            userShopService.remove(new LambdaQueryWrapper<UserShop>().eq(UserShop::getUserId, user.getUserId()).eq(UserShop::getShopId, user.getObjectId()));
-            UserShop userShop = new UserShop();
-            userShop.setUserId(user.getUserId());
-            userShop.setShopId(user.getObjectId());
-            userShop.setRoleType(2);
-            userShop.setRoleId(user.getRoleId());
-            userShopService.save(userShop);
+            UserShop one = userShopService.getOne(new LambdaQueryWrapper<UserShop>().eq(UserShop::getUserId, user.getUserId()).eq(UserShop::getShopId, user.getObjectId()));
+            one.setUserId(user.getUserId());
+            one.setShopId(user.getObjectId());
+            one.setRoleType(one.getRoleType());
+            one.setRoleId(user.getRoleId());
+            one.setDeptId(user.getDeptId());
+            one.setNickName(user.getNickName());
+            userShopService.updateById(one);
         }
         return success();
     }
@@ -510,6 +572,8 @@
     public R<SysUser> queryUserByUserName(@RequestBody String userName){
         SysUser user = userService.getOne(Wrappers.lambdaQuery(SysUser.class)
                 .eq(SysUser::getUserName, userName)
+                .eq(SysUser::getDelFlag, "0")
+                .eq(SysUser::getStatus, 0)
                 .last("LIMIT 1"));
         return R.ok(user);
     }
@@ -606,6 +670,11 @@
                 sysUser.setDelFlag("2");
                 userService.updateById(sysUser);
             }
+            if(sysUser.getObjectId().equals(objectId) && collect.size() > 1){
+                Integer integer = collect.stream().filter(s -> !s.equals(sysUser.getObjectId())).findFirst().get();
+                sysUser.setObjectId(integer);
+                userService.updateById(sysUser);
+            }
             userShopService.remove(new LambdaUpdateWrapper<UserShop>().eq(UserShop::getUserId, sysUser.getUserId()).eq(UserShop::getShopId, objectId));
         }
         return R.ok();
@@ -614,7 +683,7 @@
 
 
     @ResponseBody
-    @GetMapping("/switchShop")
+    @PutMapping("/switchShop/{shopId}")
     @ApiOperation(value = "切换门店", tags = {"门店后台-首页"})
     public R<Set<String>> switchShop(@PathVariable("shopId") Integer shopId){
         Long userid = tokenService.getLoginUser().getUserid();
@@ -630,4 +699,22 @@
         Set<String> rolePerms = menuService.selectMenuPermsByRoleId(roleId);
         return R.ok(rolePerms);
     }
+    
+    
+    
+    /**
+     * 通过账号和账号权限查询用户
+     * @param userName
+     * @return
+     */
+    @PostMapping("/queryUserByUserNameAndRoleType")
+    public R<SysUser>  queryUserByUserNameAndRoleType(@RequestParam("roleType") String userName, @RequestParam("roleType") Integer roleType){
+        SysUser user = userService.getOne(Wrappers.lambdaQuery(SysUser.class)
+                .eq(SysUser::getUserName, userName)
+                .eq(SysUser::getDelFlag, "0")
+                .eq(SysUser::getStatus, 0)
+                .eq(SysUser::getRoleType, roleType)
+                .last("LIMIT 1"));
+        return R.ok(user);
+    }
 }

--
Gitblit v1.7.1