From dc31792dc1b8ad1658e49ba8fce9a1be924e6fbb Mon Sep 17 00:00:00 2001
From: wu <731351411@qq.com>
Date: 星期四, 30 十一月 2023 08:44:04 +0800
Subject: [PATCH] 理片笼测试修改
---
springboot-vue3/src/main/java/com/example/springboot/security/UserRealm.java | 102 +++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 102 insertions(+), 0 deletions(-)
diff --git a/springboot-vue3/src/main/java/com/example/springboot/security/UserRealm.java b/springboot-vue3/src/main/java/com/example/springboot/security/UserRealm.java
new file mode 100644
index 0000000..8a47614
--- /dev/null
+++ b/springboot-vue3/src/main/java/com/example/springboot/security/UserRealm.java
@@ -0,0 +1,102 @@
+package com.example.springboot.security;
+
+import cn.hutool.core.collection.CollectionUtil;
+import com.example.springboot.entity.MenuList;
+import com.example.springboot.entity.Permission;
+import com.example.springboot.entity.Role;
+import com.example.springboot.entity.RoleMenuList;
+import com.example.springboot.entity.RolePermission;
+import com.example.springboot.entity.User;
+import com.example.springboot.security.constant.SystemConstant;
+import com.example.springboot.security.util.SecurityUtil;
+import com.example.springboot.service.MenuListService;
+import com.example.springboot.service.PermissionService;
+import com.example.springboot.service.RoleMenuListService;
+import com.example.springboot.service.RolePermissionService;
+import com.example.springboot.service.RoleService;
+import com.example.springboot.service.UserService;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.shiro.authc.AuthenticationInfo;
+import org.apache.shiro.authc.AuthenticationToken;
+import org.apache.shiro.authc.DisabledAccountException;
+import org.apache.shiro.authc.SimpleAuthenticationInfo;
+import org.apache.shiro.authc.UnknownAccountException;
+import org.apache.shiro.authc.UsernamePasswordToken;
+import org.apache.shiro.authz.AuthorizationInfo;
+import org.apache.shiro.authz.SimpleAuthorizationInfo;
+import org.apache.shiro.realm.AuthorizingRealm;
+import org.apache.shiro.subject.PrincipalCollection;
+import org.apache.shiro.util.ByteSource;
+import org.springframework.beans.factory.annotation.Autowired;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+@Slf4j
+public class UserRealm extends AuthorizingRealm {
+ @Autowired
+ private UserService userService;
+ @Autowired
+ private RolePermissionService rolePermissionService;
+ @Autowired
+ private PermissionService permissionService;
+ @Autowired
+ private RoleService roleService;
+ @Autowired
+ private MenuListService menuListService;
+ @Autowired
+ private RoleMenuListService roleMenuListService;
+
+ @Override
+ protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
+ // 鎵ц鎺堟潈
+ SimpleAuthorizationInfo authorizationInfo = new SimpleAuthorizationInfo();
+ // 璁剧疆瑙掕壊
+ List<Role> roles = roleService.selectRoles(SecurityUtil.getCurrentUser().getRoleId(), true);
+ if (CollectionUtil.isEmpty(roles)) {
+ return null;
+ }
+ authorizationInfo.addRoles(roles.stream().map(Role::getName).collect(Collectors.toList()));
+ List<RolePermission> rolePermissions = rolePermissionService.lambdaQuery().eq(RolePermission::getRoleId, SecurityUtil.getCurrentUser().getRoleId())
+ .eq(RolePermission::getState, 1).list();
+ if (CollectionUtil.isNotEmpty(rolePermissions)) {
+ Set<Permission> set = new HashSet<>();
+ for (RolePermission rolePermission : rolePermissions) {
+ List<Permission> permissions = permissionService.lambdaQuery().eq(Permission::getId, rolePermission.getPermissionId()).list();
+ set.addAll(permissions);
+ }
+ // 璁剧疆鏉冮檺
+ authorizationInfo.addStringPermissions(set.stream().map(Permission::getName).collect(Collectors.toList()));
+ }
+ return authorizationInfo;
+ }
+
+ @Override
+ protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authenticationToken) {
+ if (authenticationToken.getPrincipal() == null) {
+ return null;
+ }
+ // 鎵ц璁よ瘉
+ UsernamePasswordToken usernamePasswordToken = (UsernamePasswordToken) authenticationToken;
+ User user = userService.selectByUsername(usernamePasswordToken.getUsername());
+ // 鍒ゆ柇鐢ㄦ埛
+ if (user == null) {
+ throw new UnknownAccountException("鐢ㄦ埛涓嶅瓨鍦�!");
+ }
+ if (user.getState() == 0) {
+ throw new DisabledAccountException("璐﹀彿宸茶绂佺敤!");
+ }
+
+ // 璁よ瘉鎴愬姛涔嬪悗璁剧疆瑙掕壊鍏宠仈鐨勮彍鍗�
+ List<RoleMenuList> roleMenuLists = roleMenuListService.lambdaQuery().in(RoleMenuList::getRoleId, user.getRoleId()).list();
+ if (CollectionUtil.isNotEmpty(roleMenuLists)) {
+ List<Long> collect = roleMenuLists.stream().map(RoleMenuList::getMenuListId).collect(Collectors.toList());
+ List<MenuList> menuLists = menuListService.lambdaQuery().in(CollectionUtil.isNotEmpty(collect), MenuList::getId, collect).list();
+ // 璁よ瘉鎴愬姛涔嬪悗璁剧疆瑙掕壊鍏宠仈鐨勮彍鍗�
+ user.setMenuLists(CollectionUtil.isNotEmpty(collect) ? menuLists : null);
+ }
+ return new SimpleAuthenticationInfo(user, user.getPassword(), ByteSource.Util.bytes(SystemConstant.JWT_SECRET_KEY), getName());
+ }
+}
\ No newline at end of file
--
Gitblit v1.8.0