當(dāng)前位置：首頁(yè) > 前端技术 > javascript >内容正文

javascript

一篇搞定 SpringBoot+Mybatis+Shiro 实现多角色权限管理

發(fā)布時(shí)間：2024/3/13 javascript 23 豆豆

生活随笔收集整理的這篇文章主要介紹了一篇搞定 SpringBoot+Mybatis+Shiro 实现多角色权限管理小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

初衷：我在網(wǎng)上想找整合springboot+mybatis+shiro并且多角色認(rèn)證的博客，發(fā)現(xiàn)找了好久也沒有找到想到的，現(xiàn)在自己會(huì)了，就打算寫個(gè)博客分享出去,希望能幫到你。
原創(chuàng)不易，請(qǐng)點(diǎn)贊支持！

該項(xiàng)目不會(huì)將過多基礎(chǔ)，直接實(shí)戰(zhàn)，比較使用于有一點(diǎn)基礎(chǔ)的，并且想整合springboot+mybatis+shiro的朋友們。

代碼和數(shù)據(jù)庫(kù)sql都放在github,鏈接如下：https://github.com/zhiyuwyu/springboot-mybatis-shiro.git

一篇搞定 SpringBoot+Mybatis+Shiro 實(shí)現(xiàn)多角色權(quán)限管理

- 1、了解需求
- - 1.1、了解頁(yè)面
  - 1.2、需求
- 2、準(zhǔn)備數(shù)據(jù)庫(kù)環(huán)境
- 3、編寫代碼
- - 3.1、新建SpringBoot 工程項(xiàng)目
  - 3.2、添加如下依賴
  - 3.3、編寫代碼連接數(shù)據(jù)庫(kù)并測(cè)試
  - - 3.3.1、配置數(shù)據(jù)庫(kù)信息
    - 3.3.2、編寫實(shí)體類 entity
    - 3.3.3、mapper
    - 3.3.4、service
  - 3.4、編寫頁(yè)面
  - 3.5、編寫 shiro 的有關(guān)配置
  - 3.6、編寫Controller層代碼
- 4、測(cè)試
- - - 4.0、游客訪問
    - 4.1、測(cè)試無(wú)用戶登錄
    - 4.2、測(cè)試密碼不正確登錄
    - 4.3、測(cè)試 rootA用戶正確登錄
    - 4.4、測(cè)試adminA用戶正確登錄
    - 4.5、測(cè)試userA用戶正確登錄
- 5、結(jié)語(yǔ)

1、了解需求

1.1、了解頁(yè)面

登錄頁(yè)面如下

首頁(yè)頁(yè)面如下

分別點(diǎn)擊添加、刪除、查詢、測(cè)試超鏈接，展示的內(nèi)容如下

1.2、需求

首頁(yè)頁(yè)面必須登錄成功之后才能訪問
所有用戶、游客等都可訪問登錄頁(yè)面、測(cè)試頁(yè)面，無(wú)需登錄
擁有 root 角色的用戶可以訪問所有頁(yè)面，包括添加頁(yè)面、刪除頁(yè)面、查詢頁(yè)面、測(cè)試頁(yè)面等
擁有admin 角色的用戶可以訪問添加頁(yè)面，查詢頁(yè)面、測(cè)試頁(yè)面，除了刪除頁(yè)面不能訪問
擁有 user 角色的用戶可以訪問查詢頁(yè)面、測(cè)試頁(yè)面，除了添加頁(yè)面、刪除頁(yè)面不能訪問

2、準(zhǔn)備數(shù)據(jù)庫(kù)環(huán)境

新建一個(gè)test數(shù)據(jù)庫(kù),創(chuàng)建兩個(gè)表（role、user）并插入數(shù)據(jù)，sql 如下

/*Navicat Premium Data TransferSource Server : LocalHostSource Server Type : MySQLSource Server Version : 50731Source Host : localhost:3306Source Schema : testTarget Server Type : MySQLTarget Server Version : 50731File Encoding : 65001Date: 12/07/2021 21:08:51 */SET NAMES utf8mb4; SET FOREIGN_KEY_CHECKS = 0;-- ---------------------------- -- Table structure for role -- ---------------------------- DROP TABLE IF EXISTS `role`; CREATE TABLE `role` (`id` int(2) NOT NULL AUTO_INCREMENT,`name` varchar(100) CHARACTER SET utf8 COLLATE utf8_unicode_ci NOT NULL,`remark` varchar(255) CHARACTER SET utf8 COLLATE utf8_unicode_ci NULL DEFAULT NULL,PRIMARY KEY (`id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 4 CHARACTER SET = utf8 COLLATE = utf8_unicode_ci ROW_FORMAT = Dynamic;-- ---------------------------- -- Records of role -- ---------------------------- INSERT INTO `role` VALUES (1, 'root', '超級(jí)管理員'); INSERT INTO `role` VALUES (2, 'admin', '管理員'); INSERT INTO `role` VALUES (3, 'user', '普通用戶');-- ---------------------------- -- Table structure for user -- ---------------------------- DROP TABLE IF EXISTS `user`; CREATE TABLE `user` (`id` int(10) NOT NULL AUTO_INCREMENT,`username` varchar(100) CHARACTER SET utf8 COLLATE utf8_unicode_ci NOT NULL,`password` varchar(255) CHARACTER SET utf8 COLLATE utf8_unicode_ci NOT NULL,`role_id` int(3) NOT NULL,PRIMARY KEY (`id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 5 CHARACTER SET = utf8 COLLATE = utf8_unicode_ci ROW_FORMAT = Dynamic;-- ---------------------------- -- Records of user -- ---------------------------- INSERT INTO `user` VALUES (1, 'rootA', '123456', 1); INSERT INTO `user` VALUES (2, 'adminA', '123456', 2); INSERT INTO `user` VALUES (3, 'userA', '123456', 3); INSERT INTO `user` VALUES (4, 'userB', '123456', 3);SET FOREIGN_KEY_CHECKS = 1;

role 表數(shù)據(jù)如下：

user 表數(shù)據(jù)如下：

3、編寫代碼

3.1、新建SpringBoot 工程項(xiàng)目

3.2、添加如下依賴

全部依賴如下

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-thymeleaf</artifactId> </dependency>  <dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId> </dependency>  <dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId> </dependency>  <dependency><groupId>org.mybatis.spring.boot</groupId><artifactId>mybatis-spring-boot-starter</artifactId><version>2.2.0</version> </dependency>  <dependency><groupId>org.apache.shiro</groupId><artifactId>shiro-spring</artifactId><version>1.7.1</version> </dependency>  <dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-test</artifactId><scope>test</scope> </dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><version>1.16.14</version> </dependency>

3.3、編寫代碼連接數(shù)據(jù)庫(kù)并測(cè)試

3.3.1、配置數(shù)據(jù)庫(kù)信息

把 application.properties 文件修改成 application.yml，并添加如下內(nèi)容

spring:datasource:username: root password: 123456 url: jdbc:mysql://localhost:3306/test # 打印sql語(yǔ)句 logging:level:com:huang:shiro1:mapper: debug

3.3.2、編寫實(shí)體類 entity

新建一個(gè)entity 包，分別添加下面兩個(gè)實(shí)體類

User.java

package com.huang.springbootmybatisshiro.entity; import lombok.Data; import java.io.Serializable; /*** (User)實(shí)體類*/ @Data @NoArgsConstructor @AllArgsConstructor public class User implements Serializable {private static final long serialVersionUID = 227751358530931042L;private Integer id;private String username;private String password;private Integer roleId;private Role role; }

Role.java

package com.huang.springbootmybatisshiro.entity; import lombok.Data; import java.io.Serializable; /*** (Role)實(shí)體類*/ @Data public class Role implements Serializable {private static final long serialVersionUID = -76407922564857637L;private Integer id;private String name;private String remark;}

3.3.3、mapper

新建一個(gè) mapper 包，分別創(chuàng)建下面兩個(gè)文件

UserMapper.java

public interface UserMapper {/*** 根據(jù) Username 查詢單條數(shù)據(jù)** @param username* @return*/User queryByUsername(@Param("username") String username);}

UserMapper.xml

<?xml version="1.0" encoding="UTF-8" ?> <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"> <mapper namespace="com.huang.springbootmybatisshiro.mapper.UserMapper"><resultMap type="com.huang.springbootmybatisshiro.entity.User" id="UserMap"><result property="id" column="id" jdbcType="INTEGER"/><result property="username" column="username" jdbcType="VARCHAR"/><result property="password" column="password" jdbcType="VARCHAR"/><result property="roleId" column="role_id" jdbcType="INTEGER"/></resultMap><resultMap id="UserMapWithRole" type="com.huang.springbootmybatisshiro.entity.User" extends="UserMap"><collection property="role" ofType="com.huang.springbootmybatisshiro.entity.Role"><id property="id" column="rid"></id><result property="name" column="rname"></result><result property="remark" column="rremark"></result></collection></resultMap><select id="queryByUsername" resultMap="UserMapWithRole">selectu.*,r.id rid,r.name rname,r.remark rremarkfrom test.user u,test.role rwhere u.role_id=r.id<if test="username != null and username != ''">and username = #{username}</if></select></mapper>

注意：

UserMapper.xml 和UserMapper.java 文件寫在同一目錄下，需在pom.xml文件添加如下內(nèi)容

<build><resources><resource><directory>src/main/java</directory><includes><include>**/*.xml</include></includes></resource><resource><directory>src/main/resources</directory></resource></resources><plugins><plugin><groupId>org.springframework.boot</groupId><artifactId>spring-boot-maven-plugin</artifactId></plugin></plugins> </build>

并且在啟動(dòng)類中添加注解@mapperscan 全局掃描 mapper 文件

@SpringBootApplication @MapperScan(basePackages = "com.huang.springbootmybatisshiro.mapper") public class SpringbootMybatisShiroApplication {public static void main(String[] args) {SpringApplication.run(SpringbootMybatisShiroApplication.class, args);} }

3.3.4、service

新建一個(gè)包service，并添加 Userservice.java 文件如下

@Service public class UserService {@AutowiredUserMapper userMapper;public User queryByUsername(String username) {return userMapper.queryByUsername(username);} }

3.3.4 測(cè)試是否可以正常獲取數(shù)據(jù)庫(kù)信息

@SpringBootTest class SpringbootMybatisShiroApplicationTests {@AutowiredUserService userService;@Testvoid contextLoads() {User rootA = userService.queryByUsername("rootA");System.out.println("=====================================");System.out.println("rootA = " + rootA);}}

如果輸入如下數(shù)據(jù)則成功，就可以進(jìn)行下一步了

3.4、編寫頁(yè)面

在resources目錄下新建文件夾templates（有就不用了），在templates 下添加如下頁(yè)面

add.html 添加頁(yè)面
delete.html 刪除頁(yè)面
index.html 首頁(yè)頁(yè)面
login.html 登錄頁(yè)面
query.html 查詢頁(yè)面
test.html 測(cè)試頁(yè)面
unauthorized.html 未授權(quán)頁(yè)面

add.html

<!DOCTYPE html > <html lang="en" xmlns:th="http://www.thymeleaf.org"> <head><meta charset="UTF-8"><title>add</title> </head> <body> <h1>添加頁(yè)面</h1> </body> </html>

delete.html

<!DOCTYPE html> <html lang="en"> <head><meta charset="UTF-8"><title>update</title> </head> <body> <h1>刪除頁(yè)面</h1> </body> </html>

index.html

<!DOCTYPE html > <html lang="en" xmlns:th="http://www.thymeleaf.org"> <head><meta charset="UTF-8"><title>首頁(yè)</title> </head> <body> <h1>首頁(yè)</h1> 進(jìn)入用戶添加頁(yè)面： <a href="add">添加頁(yè)面</a><br/> 進(jìn)入用戶刪除頁(yè)面： <a href="delete">刪除頁(yè)面</a><br/> 進(jìn)入用戶查詢頁(yè)面： <a href="query">查詢頁(yè)面</a><br/> 進(jìn)入用戶測(cè)試頁(yè)面： <a href="test">測(cè)試頁(yè)面</a><br/> </body> </html>

<!DOCTYPE html> <html lang="en" xmlns:th="http://www.thymeleaf.org"> <head><meta charset="UTF-8"><title>login</title> </head> <body> <h1>登錄頁(yè)面</h1> <h3 th:text="${msg}" style="color: red"></h3> <form action="login" method="post">用戶名：<input type="text" name="username"><br>密碼：<input type="text" name="password"><br><input type="submit" value="submit"> </form> </body> </html>

query.html

<!DOCTYPE html> <html lang="en"> <head><meta charset="UTF-8"><title>query</title> </head> <body> <h1>查詢頁(yè)面</h1> </body> </html>

test.html

<!DOCTYPE html > <html lang="en" xmlns:th="http://www.thymeleaf.org"> <head><meta charset="UTF-8"><title>測(cè)試頁(yè)面</title> </head> <body> <h1>測(cè)試頁(yè)面</h1> </body> </html>

unauthorized.html。當(dāng)訪問不夠權(quán)限的頁(yè)面時(shí)會(huì)跳轉(zhuǎn)到該頁(yè)面

<!DOCTYPE html> <html lang="en"> <head><meta charset="UTF-8"><title>unauthorized</title> </head> <body> <h1>你未授權(quán)，請(qǐng)聯(lián)系管理員</h1> </body> </html>

3.5、編寫 shiro 的有關(guān)配置

新建一個(gè)config 包，添加以下文件

CustomRealm.java

/*** @Author: Zhiyu* @Date: 2021/7/13 10:40* @Description: 主要用于用戶數(shù)據(jù)和shiro的交互工作*/ public class CustomRealm extends AuthorizingRealm {@AutowiredUserService userService;/*** 授權(quán)：給當(dāng)前用戶授權(quán)，以便能訪問** @param principals* @return*/@Overrideprotected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {// principal 就是下面的方法 doGetAuthenticationInfo 中的return new SimpleAuthenticationInfo(user, user.getPassword(), "") 中的第一個(gè)參數(shù) user 賦值的User principal = (User) principals.getPrimaryPrincipal();System.out.println("principal = " + principal);// 給資源進(jìn)行授權(quán)SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();// 授權(quán)該用戶的本身角色的權(quán)限info.addRole(principal.getRole().getName());return info;}/*** 認(rèn)證：能不能登錄等認(rèn)證** @param authenticationToken* @return* @throws AuthenticationException*/@Overrideprotected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authenticationToken) throws AuthenticationException {System.out.println("-------------進(jìn)入了認(rèn)證邏輯---------------");// token 中存儲(chǔ)著 subject.login(token) 中傳過來的數(shù)據(jù)UsernamePasswordToken token = (UsernamePasswordToken) authenticationToken;String username = token.getUsername();// 這里假設(shè)，數(shù)據(jù)庫(kù)中的 username 字段是唯一字段，可以作為唯一標(biāo)識(shí)，實(shí)際開發(fā)中可以適當(dāng)修改，換湯不換藥// 根據(jù) username 去數(shù)據(jù)庫(kù)查詢用戶信息// 數(shù)據(jù)庫(kù)查詢回來的數(shù)據(jù)User user = userService.queryByUsername(username);if (user == null) {// 用戶名不存在// return null; shiro 底層會(huì)拋出UnknowAccountExceptionthrow new UnknownAccountException();}// 第一個(gè)參數(shù) user: 代表傳值,及保存用戶的信息，后面會(huì)用到// 第二個(gè)參數(shù) 填真正的密碼，shiro 會(huì)幫我們做密碼驗(yàn)證，無(wú)需我們自己做密碼驗(yàn)證邏輯return new SimpleAuthenticationInfo(user, user.getPassword(), "");} }

ShiroConfig.java

/*** @Author: Zhiyu* @Date: 2021/7/13 10:40* @Description: shiro 的配置* 完成下面三件事* 1.創(chuàng)建 ShiroFilterFactoryBean* 2.DefaultWebSecurityManager* 3.創(chuàng)建Realm并關(guān)聯(lián)*/ @Configuration public class ShiroConfig {@Bean(name = "customRealm")public CustomRealm customRealm() {return new CustomRealm();}@Bean(name = "securityManager")public SecurityManager securityManager(CustomRealm customRealm) {DefaultWebSecurityManager manager = new DefaultWebSecurityManager();manager.setRealm(customRealm);return manager;}/*** 過濾器配置** @param securityManager* @return*/@Beanpublic ShiroFilterFactoryBean shiroFilterFactoryBean(SecurityManager securityManager) {ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();shiroFilterFactoryBean.setSecurityManager(securityManager);/*** Shiro內(nèi)置過濾器，可以實(shí)現(xiàn)權(quán)限相關(guān)的攔截器* 常用的過濾器：* anon: 無(wú)需認(rèn)證（登錄）可以訪問* authc: 必須認(rèn)證才可以訪問* user: 如果使用rememberMe的功能可以直接訪問* perms：該資源必須得到資源權(quán)限才可以訪問* roles: 該資源必須得到角色權(quán)限才可以訪問*/// 1. 權(quán)限相關(guān)的攔截器（什么路徑需要什么權(quán)限）LinkedHashMap<String, String> filterMap = new LinkedHashMap<>();filterMap.put("/delete", "roles[root]"); //roles[root] 意思是訪問/delete 需要角色 root// roles[admin,root]意思是訪問/add 需要角色 admin或者root。// 如果不配置 RoleFilter，解決多角色and關(guān)系，則roles[admin,root]意思就是訪問/add 需要 admin和root兩個(gè)角色同時(shí)。filterMap.put("/add", "roles[admin,root]");// anon，意識(shí)是/test、/login / 無(wú)需認(rèn)證（登錄）可以訪問filterMap.put("/test", "anon");filterMap.put("/login", "anon");filterMap.put("/", "anon");// authc 其余的訪問都必須認(rèn)證才可以訪問，filterMap.put("/*", "authc");shiroFilterFactoryBean.setFilterChainDefinitionMap(filterMap);// 添加shiro 權(quán)限過濾器// 2. 配置自定義 or 角色認(rèn)證，把自定義過濾器配置進(jìn)去即可LinkedHashMap<String, Filter> filters = new LinkedHashMap<>();filters.put("roles", new RoleFilter());shiroFilterFactoryBean.setFilters(filters);// 3. 修改默認(rèn)的登錄頁(yè)面和未授權(quán)頁(yè)面// 即訪問需要登錄有頁(yè)面時(shí)會(huì)跳轉(zhuǎn)到 /toLogin 請(qǐng)求// 即訪問需要不夠權(quán)限的時(shí)候頁(yè)面時(shí)會(huì)跳轉(zhuǎn)到 /toLogin 請(qǐng)求shiroFilterFactoryBean.setLoginUrl("/unauthorized");// 修改調(diào)整的登錄頁(yè)面shiroFilterFactoryBean.setUnauthorizedUrl("/unauthorized");// 設(shè)置未授權(quán)提示頁(yè)面return shiroFilterFactoryBean;} }

RoleFilter.java

/*** @Author: Zhiyu* @Date: 2021/7/13 11:08* @Description: 重寫Shiro自帶角色權(quán)限過濾器* shiro自帶的方法同一權(quán)限只能分配一個(gè)角色，默認(rèn)所個(gè)角色的時(shí)候是 and 關(guān)系，不是 or 關(guān)系* 所以重寫重寫Shiro自帶角色權(quán)限過濾器解決多角色的時(shí)候是 or 關(guān)系*/ public class RoleFilter extends RolesAuthorizationFilter {@Overridepublic boolean isAccessAllowed(ServletRequest request, ServletResponse response, Object mappedValue)throws IOException {final Subject subject = getSubject(request, response);final String[] rolesArray = (String[]) mappedValue;if (rolesArray == null || rolesArray.length == 0) {// 無(wú)指定角色時(shí)，無(wú)需檢查，允許訪問return true;}for (String roleName : rolesArray) {if (subject.hasRole(roleName)) {return true;}}return false;} }

3.6、編寫Controller層代碼

新建一個(gè)controller包，添加 UserController.java 文件，內(nèi)容如下

@Controller public class UserController {/*** 使用 shiro 編寫認(rèn)證(登錄)邏輯* 1. 獲取 Subject* 2. 封裝用戶數(shù)據(jù)* 3. 執(zhí)行登錄方法*/@PostMapping("/login")public String login(String username, String password, Model model) {System.out.println("username = " + username);System.out.println("password = " + password);// 1.獲取 SubjectSubject subject = SecurityUtils.getSubject();// 2. 封裝用戶數(shù)據(jù)UsernamePasswordToken token = new UsernamePasswordToken(username, password);String msg = "登錄成功";try {// 3. 執(zhí)行登錄方法, 到 CustomRealm 類的doGetAuthenticationInfo中去執(zhí)行認(rèn)證邏輯subject.login(token);} catch (UnknownAccountException uae) {msg = "未知賬戶";} catch (IncorrectCredentialsException ice) {msg = "密碼不正確";} catch (LockedAccountException lae) {msg = "賬戶已鎖定";} catch (ExcessiveAttemptsException eae) {msg = "用戶名或密碼錯(cuò)誤次數(shù)過多";} catch (AuthenticationException ae) {msg = "用戶名或密碼不正確！";}model.addAttribute("msg", msg);if (subject.isAuthenticated()) {// 登錄成功，跳轉(zhuǎn)到 index.htmlreturn "redirect:/index";} else {token.clear();return "login";}} }

7、編寫 WebMvcConfig, 配置無(wú)邏輯的訪問頁(yè)面跳轉(zhuǎn)

在config 包下新建 WebMvcConfig.java ，代碼如下

@Configuration public class WebMvcConfig implements WebMvcConfigurer {// 這里配置一些無(wú)邏輯處理的頁(yè)面請(qǐng)求@Overridepublic void addViewControllers(ViewControllerRegistry registry) {/*** registry.addViewController("/add").setViewName("add");* 的意識(shí)等價(jià)于在controller 層的* @RequestMapping("/add")* public String add() {* return "add"; // 跳轉(zhuǎn)到 add.html 頁(yè)面* }*** 所以registry.addViewController("/add").setViewName("add");* 意思是：訪問 /add 就會(huì)跳轉(zhuǎn)到 add.html 頁(yè)面* 下面的以此類推**/registry.addViewController("/add").setViewName("add");registry.addViewController("/delete").setViewName("delete");registry.addViewController("/query").setViewName("query");registry.addViewController("/toLogin").setViewName("login");registry.addViewController("/").setViewName("login");registry.addViewController("/unauthorized").setViewName("unauthorized");registry.addViewController("/index").setViewName("index");registry.addViewController("/test").setViewName("test");} }

啟動(dòng)項(xiàng)目進(jìn)行測(cè)試

4、測(cè)試

User 表數(shù)據(jù)如下

4.0、游客訪問

游客直接在瀏覽器輸入下面的地址

http://localhost:8080/add
http://localhost:8080/delete
http://localhost:8080/query

都是下面會(huì)跳轉(zhuǎn)到登錄頁(yè)面

除了訪問http://localhost:8080/test 可以正確跳轉(zhuǎn)