From 8b8305b82f23b672023d8a3d1d1d7030af3f110c Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期四, 16 一月 2025 15:44:02 +0800
Subject: [PATCH] 修改bug

---
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java |   60 ++++++++++++++++++++++++++++++++++++++----------------------
 1 files changed, 38 insertions(+), 22 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java
index 3c52cfc..6182a9f 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java
@@ -1,23 +1,7 @@
 package com.ruoyi.system.service.impl;
 
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
-
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ruoyi.system.domain.SysRoleDept;
-import com.ruoyi.system.domain.SysRoleMenu;
-import com.ruoyi.system.domain.SysUserRole;
-import com.ruoyi.system.mapper.SysRoleDeptMapper;
-import com.ruoyi.system.mapper.SysRoleMapper;
-import com.ruoyi.system.mapper.SysRoleMenuMapper;
-import com.ruoyi.system.mapper.SysUserRoleMapper;
-import com.ruoyi.system.service.ISysRoleService;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Service;
-import org.springframework.transaction.annotation.Transactional;
 import com.ruoyi.common.core.constant.UserConstants;
 import com.ruoyi.common.core.exception.ServiceException;
 import com.ruoyi.common.core.utils.SpringUtils;
@@ -26,6 +10,22 @@
 import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.system.api.domain.SysRole;
 import com.ruoyi.system.api.domain.SysUser;
+import com.ruoyi.system.api.model.UserShop;
+import com.ruoyi.system.domain.SysRoleDept;
+import com.ruoyi.system.domain.SysRoleMenu;
+import com.ruoyi.system.domain.SysUserRole;
+import com.ruoyi.system.mapper.SysRoleDeptMapper;
+import com.ruoyi.system.mapper.SysRoleMapper;
+import com.ruoyi.system.mapper.SysRoleMenuMapper;
+import com.ruoyi.system.mapper.SysUserRoleMapper;
+import com.ruoyi.system.service.ISysRoleService;
+import com.ruoyi.system.service.UserShopService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import javax.annotation.Resource;
+import java.util.*;
 
 /**
  * 角色 业务层处理
@@ -45,6 +45,12 @@
 	
 	@Autowired
 	private SysRoleDeptMapper roleDeptMapper;
+	
+	@Resource
+	private UserShopService userShopService;
+	
+	
+	
 	
 	/**
 	 * 根据条件分页查询角色数据
@@ -86,12 +92,22 @@
 	 * @return 权限列表
 	 */
 	@Override
-	public Set<String> selectRolePermissionByUserId(Long userId) {
-		List<SysRole> perms = roleMapper.selectRolePermissionByUserId(userId);
+	public Set<String> selectRolePermissionByUserId(SysUser user) {
 		Set<String> permsSet = new HashSet<>();
-		for (SysRole perm : perms) {
-			if (StringUtils.isNotNull(perm)) {
-				permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(",")));
+		if(user.getRoleType() == 1){
+			List<SysRole> perms = roleMapper.selectRolePermissionByUserId(user.getUserId());
+			for (SysRole perm : perms) {
+				if (StringUtils.isNotNull(perm)) {
+					permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(",")));
+				}
+			}
+		}else{
+			List<UserShop> list = userShopService.list(new LambdaQueryWrapper<UserShop>().eq(UserShop::getUserId, user.getUserId())
+					.eq(UserShop::getShopId, user.getObjectId()));
+			for (UserShop userShop : list) {
+				if (StringUtils.isNotNull(userShop)) {
+					permsSet.add(userShop.getRoleId() + "");
+				}
 			}
 		}
 		return permsSet;

--
Gitblit v1.7.1