SpringBoot整合Shiro，通过用户、角色、权限三者关联实现权限管理

avatar 2019年01月24日15:45:45 8 8910 views
博主分享免费Java教学视频，B站账号：Java刘哥 ，长期提供技术问题解决、项目定制：本站商品点此
一套系统，除了个人网站，一般都要有多种角色，每种角色必须严格控制它的权限。

Shiro是一种轻量级的安全框架，主要是做登录验证，权限检查，相对 Spring Security 是要简单很多，源码也很清晰。

本文通过实战做一个介绍。

一、准备


1.pom.xml


本文采用 SpringBoot2.0.5 + MyBatis + MyBatis Plus

下面是 Shiro 的依赖
  1. <!--Shiro-->
  2. <dependency>
  3.   <groupId>org.apache.shiro</groupId>
  4.   <artifactId>shiro-spring</artifactId>
  5.   <version>1.4.0</version>
  6. </dependency>



其他依赖就不贴，如果缺少，可以在下面评论取索取，或者在 Maven 上搜


2.数据库


5张表

① user 表
  1. CREATE TABLE `sens_user` (
  2.   `user_id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
  3.   `login_enable` varchar(255) DEFAULT NULL,
  4.   `login_error` int(11) DEFAULT NULL,
  5.   `login_last` datetime DEFAULT NULL,
  6.   `user_avatar` varchar(255) DEFAULT NULL,
  7.   `user_desc` varchar(255) DEFAULT NULL,
  8.   `user_display_name` varchar(255) DEFAULT NULL,
  9.   `user_email` varchar(255) DEFAULT NULL,
  10.   `user_name` varchar(255) DEFAULT NULL,
  11.   `user_pass` varchar(255) DEFAULT NULL,
  12.   `user_site` varchar(255) DEFAULT NULL,
  13.   `status` int(1) DEFAULT NULL,
  14.   `create_time` datetime DEFAULT NULL,
  15.   PRIMARY KEY (`user_id`)
  16. ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4





②  role 表
  1. CREATE TABLE `sens_rbac_role` (
  2.   `id` int(11) NOT NULL,
  3.   `role` varchar(255) DEFAULT NULL,
  4.   `description` varchar(255) DEFAULT NULL
  5. ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4





③ permission 表
  1. CREATE TABLE `sens_rbac_permission` (
  2.   `id` int(11) NOT NULL AUTO_INCREMENT,
  3.   `name` varchar(255) DEFAULT NULL,
  4.   `permission` varchar(255) DEFAULT NULL,
  5.   PRIMARY KEY (`id`)
  6. ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4





④ user_role_ref 表
  1. CREATE TABLE `sens_rbac_user_role_ref` (
  2.   `user_id` int(20) DEFAULT NULL,
  3.   `role_id` int(11) DEFAULT NULL
  4. ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4





⑤ role_permission_ref 表
  1. CREATE TABLE `sens_rbac_role_permission_ref` (
  2.   `role_id` int(11) DEFAULT NULL,
  3.   `permission_id` int(11) DEFAULT NULL
  4. ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4







备注：

本来还有一张菜单表，不属于本文的范畴，就不介绍了。

跟权限有关的，表名加前缀 rbac_，项目名叫 sens，表名加前缀 sens_


二、基础代码(Model，Dao和Service层)


1.Model 实体


User.java
  1. package com.liuyanzhao.sens.entity;
  3. import com.baomidou.mybatisplus.annotations.TableId;
  4. import com.baomidou.mybatisplus.annotations.TableName;
  5. import com.baomidou.mybatisplus.enums.IdType;
  6. import com.fasterxml.jackson.annotation.JsonIgnore;
  7. import lombok.Data;
  8. import javax.validation.constraints.Email;
  9. import javax.validation.constraints.NotBlank;
  10. import java.io.Serializable;
  11. import java.util.Date;
  13. /**
  14.  * <pre>
  15.  *     博主信息
  16.  * </pre>
  17.  *
  18.  * @author : saysky
  19.  * @date : 2017/11/14
  20.  */
  21. @Data
  22. @TableName("sens_user")
  23. public class User implements Serializable {
  25.     private static final long serialVersionUID = -5144055068797033748L;
  27.     /**
  28.      * 编号
  29.      */
  30.     @TableId(type = IdType.AUTO)
  31.     private Long userId;
  33.     /**
  34.      * 用户名
  35.      */
  36.     @NotBlank(message = "用户名不能为空")
  37. //    @JsonIgnore
  38.     private String userName;
  40.     /**
  41.      * 显示名称
  42.      */
  43.     private String userDisplayName;
  45.     /**
  46.      * 密码
  47.      */
  48. //    @JsonIgnore
  49.     private String userPass;
  51.     /**
  52.      * 邮箱
  53.      */
  54.     @Email(message = "邮箱格式不正确")
  55.     private String userEmail;
  57.     /**
  58.      * 头像
  59.      */
  60.     private String userAvatar;
  62.     /**
  63.      * 说明
  64.      */
  65.     private String userDesc;
  67.     /**
  68.      * 个人主页
  69.      */
  70.     private String userSite;
  72.     /**
  73.      * 是否禁用登录
  74.      */
  75. //    @JsonIgnore
  76.     private String loginEnable = "true";
  78.     /**
  79.      * 最后一次登录时间
  80.      */
  81. //    @JsonIgnore
  82.     private Date loginLast;
  84.     /**
  85.      * 登录错误次数记录
  86.      */
  87. //    @JsonIgnore
  88.     private Integer loginError = 0;
  90.     /**
  91.      * 0 正常
  92.      * 1 禁用
  93.      * 2 已删除
  94.      */
  95.     private Integer status = 0;
  97.     /**
  98.      * 注册时间
  99.      */
  100.     private Date createTime;
  101. }



Role.java
  1. package com.liuyanzhao.sens.entity;
  3. import com.baomidou.mybatisplus.annotations.TableField;
  4. import com.baomidou.mybatisplus.annotations.TableName;
  5. import lombok.Data;
  7. import java.io.Serializable;
  9. /**
  10.  * @author liuyanzhao
  11.  */
  12. @Data
  13. @TableName("sens_rbac_role")
  14. public class Role implements Serializable  {
  16.     private static final long serialVersionUID = 2233979806802117985L;
  18.     /**
  19.      * ID
  20.      */
  21.     private Integer id;
  23.     /**
  24.      * 角色名称：admin，author，subscriber
  25.      */
  26.     private String role;
  28.     /**
  29.      * 描述：管理员，作者，订阅者
  30.      */
  31.     private String description;
  33.     /**
  34.      * 该角色对应的用户数量，非数据库字段
  35.      */
  36.     @TableField(exist = false)
  37.     private Integer count;
  39. }



Permisison.java
  1. package com.liuyanzhao.sens.entity;
  3. import com.baomidou.mybatisplus.annotations.TableName;
  4. import lombok.Data;
  6. import java.io.Serializable;
  8. /**
  9.  * @author liuyanzhao
  10.  */
  11. @Data
  12. @TableName("sens_rbac_permission")
  13. public class Permission implements Serializable  {
  15.     private static final long serialVersionUID = 2233979806802117985L;
  17.     /**
  18.      * ID
  19.      */
  20.     private Integer id;
  22.     /**
  23.      * 权限名称
  24.      */
  25.     private String name;
  27.     /**
  28.      * 权限：user:list，user:add等
  29.      */
  30.     private String permission;
  34. }



UserRoleRef.java
  1. package com.liuyanzhao.sens.entity;
  3. import com.baomidou.mybatisplus.annotations.TableName;
  4. import lombok.Data;
  6. import java.io.Serializable;
  8. @Data
  9. @TableName("sens_rbac_user_role_ref")
  10. public class UserRoleRef implements Serializable {
  12.     private static final long serialVersionUID = -1528306395712515362L;
  14.     /**
  15.      * 用户Id
  16.      */
  17.     private Long userId;
  19.     /**
  20.      * 角色Id
  21.      */
  22.     private Integer roleId;
  24. }



RolePermission.java
  1. package com.liuyanzhao.sens.entity;
  3. import com.baomidou.mybatisplus.annotations.TableName;
  4. import lombok.Data;
  6. import java.io.Serializable;
  8. @Data
  9. @TableName("sens_rbac_role_permission_ref")
  10. public class RolePermissionRef implements Serializable {
  12.     private static final long serialVersionUID = -4015823675140925791L;
  14.     /**
  15.      * 角色Id
  16.      */
  17.     private Integer roleId;
  19.     /**
  20.      * 权限Id
  21.      */
  22.     private Integer permissionId;
  25. }


2.Mapper类


UserMapper.java
  1. package com.liuyanzhao.sens.mapper;
  3. import com.baomidou.mybatisplus.mapper.BaseMapper;
  4. import com.baomidou.mybatisplus.plugins.pagination.Pagination;
  5. import com.liuyanzhao.sens.entity.User;
  6. import org.apache.ibatis.annotations.Mapper;
  7. import org.apache.ibatis.annotations.Param;
  9. import java.util.List;
  11. /**
  12.  * @author liuyanzhao
  13.  */
  14. @Mapper
  15. public interface UserMapper extends BaseMapper<User> {
  17.     /**
  18.      * 查询所有
  19.      *
  20.      * @return 用户列表
  21.      */
  22.     List<User> findAll(Pagination page);
  24.     /**
  25.      * 根据角色Id获得用户
  26.      *
  27.      * @param roleId 角色Id
  28.      * @param pagination 分页信息
  29.      * @return 用户列表
  30.      */
  31.     List<User> findByRoleId(@Param("roleId") Integer roleId, Pagination pagination);
  32.     /**
  33.      * 根据用户名获得用户
  34.      *
  35.      * @param userName 用户名
  36.      * @return 用户
  37.      */
  38.     User findByUserName(String userName);
  40.     /**
  41.      * 根据用户邮箱获得用户
  42.      *
  43.      * @param userEmail 邮箱
  44.      * @return 用户
  45.      */
  46.     User findByUserEmail(String userEmail);
  48.     /**
  49.      * 根据用户Id获得用户
  50.      * @param userId 用户Id
  51.      * @return 用户
  52.      */
  53.     User findByUserId(Long userId);
  55. }



RoleMapper.java
  1. package com.liuyanzhao.sens.mapper;
  3. import com.baomidou.mybatisplus.mapper.BaseMapper;
  4. import com.liuyanzhao.sens.entity.Role;
  5. import org.apache.ibatis.annotations.Mapper;
  7. import java.util.List;
  10. /**
  11.  * @author liuyanzhao
  12.  */
  13. @Mapper
  14. public interface RoleMapper extends BaseMapper<Role> {
  17.     /**
  18.      * 根据用户Id获得角色列表
  19.      *
  20.      * @param userId 用户Id
  21.      * @return 角色列表
  22.      */
  23.     List<Role> findByUserId(Long userId);
  25.     /**
  26.      * 根据名称获得角色
  27.      *
  28.      * @param roleName 角色名
  29.      * @return 角色
  30.      */
  31.     Role findByRoleName(String roleName);
  33.     /**
  34.      * 获得角色列表
  35.      *
  36.      * @return 角色列表
  37.      */
  38.     List<Role> findAll();
  40.     /**
  41.      * 删除用户和角色管理
  42.      *
  43.      * @param userId 用户ID
  44.      * @return 影响行数
  45.      */
  46.     Integer deleteByUserId(Long userId);
  48.     /**
  49.      * 统计某个角色的用户数
  50.      *
  51.      * @param roleId 角色Id
  52.      * @return 用户数
  53.      */
  54.     Integer countUserByRoleId(Integer roleId);
  55. }



PermissionMapper.java
  1. package com.liuyanzhao.sens.mapper;
  3. import com.baomidou.mybatisplus.mapper.BaseMapper;
  4. import com.liuyanzhao.sens.entity.Permission;
  5. import org.apache.ibatis.annotations.Mapper;
  7. import java.util.List;
  10. /**
  11.  * @author liuyanzhao
  12.  */
  13. @Mapper
  14. public interface PermissionMapper extends BaseMapper<Permission> {
  16.     /**
  17.      * 根据角色Id获得权限列表
  18.      *
  19.      * @param roleId 角色Id
  20.      * @return 权限列表
  21.      */
  22.     List<Permission> findByRoleId(Integer roleId);
  24.     /**
  25.      * 获得权限列表
  26.      *
  27.      * @return 权限列表
  28.      */
  29.     List<Permission> findAll();
  31. }



备注：

关于UserRoleRefMapper.java 和 RolePermissionRefMapper.java 本文没有涉及到，不贴了


3.Mapper  xml


UserMapper.xml
  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.liuyanzhao.sens.mapper.UserMapper">
  4.   <resultMap id="BaseResultMap" type="com.liuyanzhao.sens.entity.User">
  5.     <id column="user_id" jdbcType="INTEGER" property="userId"/>
  6.     <result column="user_name" jdbcType="VARCHAR" property="userName"/>
  7.     <result column="user_display_name" jdbcType="VARCHAR" property="userDisplayName"/>
  8.     <result column="user_pass" jdbcType="VARCHAR" property="userPass"/>
  9.     <result column="user_email" jdbcType="VARCHAR" property="userEmail"/>
  10.     <result column="user_avatar" jdbcType="VARCHAR" property="userAvatar"/>
  11.     <result column="user_desc" jdbcType="VARCHAR" property="userDesc"/>
  12.     <result column="user_site" jdbcType="VARCHAR" property="userSite"/>
  13.     <result column="login_enable" jdbcType="VARCHAR" property="loginEnable"/>
  14.     <result column="login_last" jdbcType="TIMESTAMP" property="loginLast"/>
  15.     <result column="login_error" jdbcType="INTEGER" property="loginError"/>
  16.     <result column="status" jdbcType="INTEGER" property="status"/>
  17.     <result column="create_time" jdbcType="TIMESTAMP" property="createTime"/>
  18.   </resultMap>
  20.   <sql id="all_columns">
  21.    user_id, user_name, user_display_name, user_pass, user_email,
  22.    user_avatar, user_desc, user_site, login_enable, login_last,
  23.    login_error, status, create_time
  24.    </sql>
  26.   <sql id="normal">0</sql>
  28.   <sql id="all_columns_with_table_name">
  29.     sens_user.user_id, sens_user.user_name, sens_user.user_display_name,
  30.     sens_user.user_pass, sens_user.user_email, sens_user.user_avatar,
  31.     sens_user.user_desc, sens_user.user_site, sens_user.login_enable,
  32.     sens_user.login_last, sens_user.login_error, sens_user.status, sens_user.create_time
  33.   </sql>
  35.   <sql id="tb">`sens_user`</sql>
  37.   <sql id="all_values">
  38.         #{userId}, #{userName}, #{userDisplayName}, #{userEmail}, #{userPass},
  39.         #{userAvatar}, #{userDesc}, #{userSite}, #{loginEnable}, #{loginLast},
  40.         #{loginError}, #{status}, #{createTime}
  41.     </sql>
  43.   <select id="findAll" resultMap="BaseResultMap">
  44.     SELECT
  45.     <include refid="all_columns"/>
  46.     FROM
  47.     <include refid="tb"/>
  48.   </select>
  50.   <select id="findByRoleId" resultType="com.liuyanzhao.sens.entity.User">
  51.     SELECT
  52.     <include refid="all_columns_with_table_name"/>
  53.     FROM sens_rbac_user_role_ref,
  54.     sens_user WHERE sens_rbac_user_role_ref.role_id = #{roleId} AND
  55.     sens_rbac_user_role_ref.user_id = sens_user.user_id
  56.   </select>
  58.   <select id="findByUserName" resultType="com.liuyanzhao.sens.entity.User">
  59.     SELECT
  60.     <include refid="all_columns"/>
  61.     FROM
  62.     <include refid="tb"/>
  63.     WHERE user_name = #{value} AND status = <include refid="normal"/> LIMIT 1
  64.   </select>
  66.   <select id="findByUserEmail" resultType="com.liuyanzhao.sens.entity.User">
  67.     SELECT
  68.     <include refid="all_columns"/>
  69.     FROM
  70.     <include refid="tb"/>
  71.     WHERE user_email = #{value} AND status = <include refid="normal"/> LIMIT 1
  72.   </select>
  75.   <select id="findByUserId" resultType="com.liuyanzhao.sens.entity.User">
  76.     SELECT
  77.     <include refid="all_columns"/>
  78.     FROM
  79.     <include refid="tb"/>
  80.     WHERE user_id = #{value}
  81.   </select>
  82. </mapper>



RoleMapper.xml
  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.liuyanzhao.sens.mapper.RoleMapper">
  4.   <resultMap id="BaseResultMap" type="com.liuyanzhao.sens.entity.Role">
  5.     <id column="id" jdbcType="INTEGER" property="id"/>
  6.     <result column="role" jdbcType="VARCHAR" property="role"/>
  7.     <result column="description" jdbcType="VARCHAR" property="description"/>
  8.   </resultMap>
  10.   <sql id="all_columns">
  11.     id, role, description
  12.   </sql>
  14.   <sql id="tb">`sens_rbac_role`</sql>
  16.   <sql id="all_values">
  17.         #{id}, #{role}, #{description}
  18.   </sql>
  20.   <sql id="all_values_with_table_name">
  21.     sens_rbac_role.id, sens_rbac_role.role, sens_rbac_role.description
  22.   </sql>
  23.   <select id="findByUserId" resultType="com.liuyanzhao.sens.entity.Role">
  24.     select
  25.     <include refid="all_values_with_table_name"/>
  26.     FROM<include refid="tb"/>, sens_rbac_user_role_ref
  27.     where sens_rbac_user_role_ref.user_id = #{value} AND
  28.     sens_rbac_role.id = sens_rbac_user_role_ref.role_id
  29.   </select>
  31.   <select id="findByRoleName" resultType="com.liuyanzhao.sens.entity.Role">
  32.     SELECT
  33.     <include refid="all_columns"/>
  34.     FROM
  35.     <include refid="tb"/>
  36.     WHERE role = #{value} LIMIT 1
  37.   </select>
  39.   <select id="findAll" resultType="com.liuyanzhao.sens.entity.Role">
  40.     SELECT
  41.     <include refid="all_columns"/>
  42.     FROM
  43.     <include refid="tb"/>
  44.   </select>
  46.   <delete id="deleteByUserId">
  47.     DELETE FROM sens_rbac_user_role_ref
  48.     WHERE user_id = #{value}
  49.   </delete>
  51.   <select id="countUserByRoleId" resultType="java.lang.Integer">
  52.     SELECT count(*) FROM sens_rbac_user_role_ref,sens_user
  53.      WHERE sens_rbac_user_role_ref.role_id = #{value} AND
  54.      sens_rbac_user_role_ref.user_id = sens_user.user_id
  55.   </select>
  56. </mapper>



PermissionMapper.xml
  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.liuyanzhao.sens.mapper.PermissionMapper">
  4.   <resultMap id="BaseResultMap" type="com.liuyanzhao.sens.entity.Permission">
  5.     <id column="id" jdbcType="INTEGER" property="id"/>
  6.     <result column="name" jdbcType="VARCHAR" property="name"/>
  7.     <result column="permission" jdbcType="VARCHAR" property="permission"/>
  8.   </resultMap>
  10.   <sql id="all_columns">
  11.     id, `name`, permission
  12.   </sql>
  14.   <sql id="all_values_with_table_name">
  15.     sens_rbac_permission.id, sens_rbac_permission.`name`, sens_rbac_permission.permission
  16.   </sql>
  18.   <sql id="tb">`sens_rbac_permission`</sql>
  20.   <sql id="all_values">
  21.         #{id}, #{name}, #{permission}
  22.   </sql>
  24.   <select id="findByRoleId" resultType="com.liuyanzhao.sens.entity.Permission">
  25.     select
  26.     <include refid="all_values_with_table_name"/>
  27.     FROM<include refid="tb"/>, sens_rbac_role_permission_ref
  28.     where sens_rbac_role_permission_ref.role_id = #{value} AND
  29.     sens_rbac_permission.id = sens_rbac_role_permission_ref.permission_id
  30.   </select>
  33.   <select id="findAll" resultType="com.liuyanzhao.sens.entity.Permission">
  34.     SELECT <include refid="all_columns"/>
  35.     FROM <include refid="tb"/>
  36.   </select>
  37. </mapper>



备注：

关于UserRoleRefMapper.xml 和 RolePermissionRefMapper.xml 本文没有涉及到，不贴了



4. Service 接口

UserService.java
  1. package com.liuyanzhao.sens.service;
  3. import com.baomidou.mybatisplus.plugins.Page;
  4. import com.liuyanzhao.sens.entity.User;
  6. import java.util.Date;
  8. /**
  9.  * <pre>
  10.  *     用户业务逻辑接口
  11.  * </pre>
  12.  *
  13.  * @author : saysky
  14.  * @date : 2017/11/14
  15.  */
  16. public interface UserService {
  18.     /**
  19.      * 新增/修改用户
  20.      *
  21.      * @param user user
  22.      * @return Role
  23.      */
  24.     User saveByUser(User user);
  25.     /**
  26.      * 根据用户名获得用户
  27.      *
  28.      * @param userName 用户名
  29.      * @return 用户
  30.      */
  31.     User findByUserName(String userName);
  34.     /**
  35.      * 根据邮箱查找用户
  36.      *
  37.      * @param userEmail 邮箱
  38.      * @return User
  39.      */
  40.     User findByEmail(String userEmail);
  42.     /**
  43.      * 根据用户Id获得用户
  44.      *
  45.      * @param userId 用户名
  46.      * @return 用户
  47.      */
  48.     User findByUserId(Long userId);
  50.     /**
  51.      * 修改禁用状态
  52.      *
  53.      * @param enable enable
  54.      */
  55.     void updateUserLoginEnable(User user, String enable);
  57.     /**
  58.      * 修改最后登录时间
  59.      *
  60.      * @param lastDate 最后登录时间
  61.      * @return User
  62.      */
  63.     User updateUserLoginLast(User user, Date lastDate);
  65.     /**
  66.      * 增加登录错误次数
  67.      *
  68.      * @return 登录错误次数
  69.      */
  70.     Integer updateUserLoginError(User user);
  72.     /**
  73.      * 修改用户的登录状态为正常
  74.      *
  75.      * @return User
  76.      */
  77.     User updateUserLoginNormal(User user);
  81. }



RoleService.java
  1. package com.liuyanzhao.sens.service;
  3. import com.liuyanzhao.sens.entity.Role;
  4. import com.liuyanzhao.sens.entity.User;
  6. import java.util.List;
  8. /**
  9.  * <pre>
  10.  *     角色逻辑接口
  11.  * </pre>
  12.  *
  13.  * @author : saysky
  14.  * @date : 2017/11/14
  15.  */
  16. public interface RoleService {
  18.     /**
  19.      * 新增/修改角色
  20.      *
  21.      * @param role role
  22.      * @return Role
  23.      */
  24.     Role saveByRole(Role role);
  26.     /**
  27.      * 根据编号删除
  28.      *
  29.      * @param roleId roleId
  30.      * @return Role
  31.      */
  32.     void removeByRoleId(Integer roleId);
  34.     /**
  35.      * 删除某个用户的所有关联
  36.      *
  37.      * @param userId 用户Id
  38.      */
  39.     void removeByUserId(Long userId);
  40.     /**
  41.      * 根据编号查询单个权限
  42.      *
  43.      * @param roleId roleId
  44.      * @return Role
  45.      */
  46.     Role findByRoleId(Integer roleId);
  48.     /**
  49.      * 根据编号查询单个权限
  50.      *
  51.      * @param roleName roleName
  52.      * @return Role
  53.      */
  54.     Role findByRoleName(String roleName);
  56.     /**
  57.      * 根据用户Id获得角色列表
  58.      *
  59.      * @param userId 用户Id
  60.      * @return 角色列表
  61.      */
  62.     List<Role> listRolesByUserId(Long userId);
  64.     /**
  65.      * 获得所有的角色
  66.      *
  67.      * @return 角色列表
  68.      */
  69.     List<Role> findAll();
  71. }



PermissionService.java
  1. package com.liuyanzhao.sens.service;
  3. import com.liuyanzhao.sens.entity.Permission;
  5. import java.util.List;
  7. /**
  8.  * <pre>
  9.  *     权限逻辑接口
  10.  * </pre>
  11.  *
  12.  * @author : saysky
  13.  * @date : 2017/11/14
  14.  */
  15. public interface PermissionService {
  17.     /**
  18.      * 新增/修改权限
  19.      *
  20.      * @param permission permission
  21.      * @return Permission
  22.      */
  23.     Permission saveByPermission(Permission permission);
  25.     /**
  26.      * 根据编号删除
  27.      *
  28.      * @param permissionId permissionId
  29.      * @return Permission
  30.      */
  31.     void removeByPermissionId(Integer permissionId);
  33.     /**
  34.      * 根据编号查询单个权限
  35.      *
  36.      * @param permissionId permissionId
  37.      * @return Permission
  38.      */
  39.     Permission findByPermissionId(Integer permissionId);
  41.     /**
  42.      * 根据角色Id获得权限列表
  43.      *
  44.      * @param roleId 角色Id
  45.      * @return 权限列表
  46.      */
  47.     List<Permission> listPermissionsByRoleId(Integer roleId);
  49. }


5.Service 实现


UserServiceImpl.java
  1. package com.liuyanzhao.sens.service.impl;
  3. import com.baomidou.mybatisplus.plugins.Page;
  4. import com.liuyanzhao.sens.entity.Role;
  5. import com.liuyanzhao.sens.mapper.UserMapper;
  6. import com.liuyanzhao.sens.entity.User;
  7. import com.liuyanzhao.sens.model.enums.CommentStatusEnum;
  8. import com.liuyanzhao.sens.model.enums.TrueFalseEnum;
  9. import com.liuyanzhao.sens.model.enums.UserStatusEnum;
  10. import com.liuyanzhao.sens.service.RoleService;
  11. import com.liuyanzhao.sens.service.UserService;
  12. import com.liuyanzhao.sens.utils.RedisUtil;
  13. import org.springframework.beans.factory.annotation.Autowired;
  14. import org.springframework.cache.annotation.CacheEvict;
  15. import org.springframework.cache.annotation.Cacheable;
  16. import org.springframework.stereotype.Service;
  17. import org.springframework.transaction.annotation.Transactional;
  19. import java.util.Collections;
  20. import java.util.Date;
  21. import java.util.List;
  22. import java.util.Objects;
  24. /**
  25.  * <pre>
  26.  *     用户业务逻辑实现类
  27.  * </pre>
  28.  *
  29.  * @author : saysky
  30.  * @date : 2017/11/14
  31.  */
  32. @Service
  33. public class UserServiceImpl implements UserService {
  35.     private static final String USERS_CACHE_NAME = "users";
  37.     @Autowired(required = false)
  38.     private UserMapper userMapper;
  41.     @Autowired(required = false)
  42.     private RoleService roleService;
  44.     @Override
  45.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  46.     public User saveByUser(User user) {
  47.         if (user != null && user.getUserId() != null) {
  48.             userMapper.updateById(user);
  49.         } else {
  50.             userMapper.insert(user);
  51.         }
  52.         return user;
  53.     }
  55.     @Override
  56.     @Cacheable(value = USERS_CACHE_NAME, key = "'users_name_'+#userName", unless = "#result == null")
  57.     public User findByUserName(String userName) {
  58.         return userMapper.findByUserName(userName);
  59.     }
  61.     @Override
  62.     @Cacheable(value = USERS_CACHE_NAME, key = "'users_email_'+#userEmail", unless = "#result == null")
  63.     public User findByEmail(String userEmail) {
  64.         return userMapper.findByUserEmail(userEmail);
  65.     }
  67.     @Override
  68.     @Cacheable(value = USERS_CACHE_NAME, key = "'users_id_'+#userName", unless = "#result == null")
  69.     public User findByUserId(Long userId) {
  70.         return userMapper.findByUserId(userId);
  71.     }
  73.     @Override
  74.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  75.     public void updateUser(User user) {
  76.         userMapper.updateById(user);
  77.     }
  79.     /**
  80.      * 修改禁用状态
  81.      *
  82.      * @param enable enable
  83.      */
  84.     @Override
  85.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  86.     public void updateUserLoginEnable(User user, String enable) {
  87.         //如果是修改为正常, 重置错误次数
  88.         if (Objects.equals(TrueFalseEnum.TRUE.getDesc(), enable)) {
  89.             user.setLoginError(0);
  90.         }
  91.         user.setLoginEnable(enable);
  92.         this.updateUser(user);
  93.     }
  95.     /**
  96.      * 修改最后登录时间
  97.      *
  98.      * @param lastDate 最后登录时间
  99.      * @return User
  100.      */
  101.     @Override
  102.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  103.     public User updateUserLoginLast(User user, Date lastDate) {
  104.         user.setLoginLast(lastDate);
  105.         this.updateUser(user);
  106.         return user;
  107.     }
  109.     /**
  110.      * 增加登录错误次数
  111.      *
  112.      * @return 登录错误次数
  113.      */
  114.     @Override
  115.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  116.     public Integer updateUserLoginError(User user) {
  117.         user.setLoginError((user.getLoginError() == null ? 0 : user.getLoginError()) + 1);
  118.         this.updateUser(user);
  119.         return user.getLoginError();
  120.     }
  122.     /**
  123.      * 修改用户的状态为正常
  124.      *
  125.      * @return User
  126.      */
  127.     @Override
  128.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  129.     public User updateUserLoginNormal(User user) {
  130.         user.setLoginEnable(TrueFalseEnum.TRUE.getDesc());
  131.         user.setLoginError(0);
  132.         user.setLoginLast(new Date());
  133.         this.updateUser(user);
  134.         return user;
  135.     }
  137.     @Override
  138.     @CacheEvict(value = USERS_CACHE_NAME, allEntries = true, beforeInvocation = true)
  139.     public void updateUserStatus(Long userId, Integer status) {
  140.         User user = findByUserId(userId);
  141.         if (user != null) {
  142.             user.setStatus(status);
  143.             userMapper.updateById(user);
  144.         }
  145.     }
  147. }



RoleServiceImpl.java
  1. package com.liuyanzhao.sens.service.impl;
  3. import com.liuyanzhao.sens.entity.Role;
  4. import com.liuyanzhao.sens.mapper.RoleMapper;
  5. import com.liuyanzhao.sens.service.RoleService;
  6. import org.springframework.beans.factory.annotation.Autowired;
  7. import org.springframework.cache.annotation.CacheEvict;
  8. import org.springframework.cache.annotation.Cacheable;
  9. import org.springframework.stereotype.Service;
  11. import java.util.List;
  13. /**
  14.  * <pre>
  15.  *     角色业务逻辑实现类
  16.  * </pre>
  17.  *
  18.  * @author : saysky
  19.  * @date : 2017/11/14
  20.  */
  21. @Service
  22. public class RoleServiceImpl implements RoleService {
  24.     private static final String ROLES_CACHE_NAME = "roles";
  26.     @Autowired(required = false)
  27.     private RoleMapper roleMapper;
  29.     @Override
  30.     @CacheEvict(value = ROLES_CACHE_NAME, allEntries = true, beforeInvocation = true)
  31.     public Role saveByRole(Role role) {
  32.         if (role != null && role.getId() != null) {
  33.             roleMapper.updateById(role);
  34.         } else {
  35.             roleMapper.insert(role);
  36.         }
  37.         return role;
  38.     }
  40.     @Override
  41.     @CacheEvict(value = ROLES_CACHE_NAME, allEntries = true, beforeInvocation = true)
  42.     public void removeByRoleId(Integer roleId) {
  43.         roleMapper.deleteById(roleId);
  44.     }
  46.     @Override
  47.     @CacheEvict(value = ROLES_CACHE_NAME, allEntries = true, beforeInvocation = true)
  48.     public void removeByUserId(Long userId) {
  49.         roleMapper.deleteByUserId(userId);
  50. }
  52.     @Override
  53.     @Cacheable(value = ROLES_CACHE_NAME, key = "'roles_id_'+#roleId", unless = "#result == null")
  54.     public Role findByRoleId(Integer roleId) {
  55.         return roleMapper.selectById(roleId);
  56.     }
  58.     @Override
  59.     @Cacheable(value = ROLES_CACHE_NAME, key = "'roles_name_'+#roleName", unless = "#result == null")
  60.     public Role findByRoleName(String roleName) {
  61.         return roleMapper.findByRoleName(roleName);
  62.     }
  64.     @Override
  65.     @Cacheable(value = ROLES_CACHE_NAME, key = "'roles_uid_'+#userId", unless = "#result == null")
  66.     public List<Role> listRolesByUserId(Long userId) {
  67.         return roleMapper.findByUserId(userId);
  68.     }
  70.     @Override
  71.     @Cacheable(value = ROLES_CACHE_NAME, key = "'roles_all'")
  72.     public List<Role> findAll() {
  73.         //获得角色
  74.         List<Role> roles = roleMapper.findAll();
  75.         //封装count
  76.         roles.forEach(role -> role.setCount(roleMapper.countUserByRoleId(role.getId())));
  77.         return roles;
  78.     }
  80. }



PermissionServiceImpl.java
  1. package com.liuyanzhao.sens.service;
  3. import com.liuyanzhao.sens.entity.Permission;
  5. import java.util.List;
  7. /**
  8.  * <pre>
  9.  *     权限逻辑接口
  10.  * </pre>
  11.  *
  12.  * @author : saysky
  13.  * @date : 2017/11/14
  14.  */
  15. public interface PermissionService {
  17.     /**
  18.      * 新增/修改权限
  19.      *
  20.      * @param permission permission
  21.      * @return Permission
  22.      */
  23.     Permission saveByPermission(Permission permission);
  25.     /**
  26.      * 根据编号删除
  27.      *
  28.      * @param permissionId permissionId
  29.      * @return Permission
  30.      */
  31.     void removeByPermissionId(Integer permissionId);
  33.     /**
  34.      * 根据编号查询单个权限
  35.      *
  36.      * @param permissionId permissionId
  37.      * @return Permission
  38.      */
  39.     Permission findByPermissionId(Integer permissionId);
  41.     /**
  42.      * 根据角色Id获得权限列表
  43.      *
  44.      * @param roleId 角色Id
  45.      * @return 权限列表
  46.      */
  47.     List<Permission> listPermissionsByRoleId(Integer roleId);
  49. }


三、整合 Shiro


1.MyShiroRealm.java
  1. package com.liuyanzhao.sens.config;
  3. import cn.hutool.core.date.DateUnit;
  4. import cn.hutool.core.date.DateUtil;
  5. import cn.hutool.core.lang.Validator;
  6. import cn.hutool.core.util.ObjectUtil;
  7. import cn.hutool.extra.servlet.ServletUtil;
  8. import cn.hutool.http.HtmlUtil;
  9. import com.liuyanzhao.sens.entity.Permission;
  10. import com.liuyanzhao.sens.entity.Role;
  11. import com.liuyanzhao.sens.entity.User;
  12. import com.liuyanzhao.sens.model.dto.JsonResult;
  13. import com.liuyanzhao.sens.model.dto.LogsRecord;
  14. import com.liuyanzhao.sens.model.enums.CommonParamsEnum;
  15. import com.liuyanzhao.sens.model.enums.ResultCodeEnum;
  16. import com.liuyanzhao.sens.model.enums.TrueFalseEnum;
  17. import com.liuyanzhao.sens.service.PermissionService;
  18. import com.liuyanzhao.sens.service.RoleService;
  19. import com.liuyanzhao.sens.service.UserService;
  20. import com.liuyanzhao.sens.utils.LocaleMessageUtil;
  21. import com.liuyanzhao.sens.utils.Md5Util;
  22. import lombok.extern.slf4j.Slf4j;
  23. import org.apache.commons.lang3.StringUtils;
  24. import org.apache.shiro.authc.*;
  25. import org.apache.shiro.authz.AuthorizationInfo;
  26. import org.apache.shiro.authz.SimpleAuthorizationInfo;
  27. import org.apache.shiro.realm.AuthorizingRealm;
  28. import org.apache.shiro.subject.PrincipalCollection;
  29. import org.apache.shiro.util.ByteSource;
  30. import org.springframework.beans.factory.annotation.Autowired;
  32. import java.util.Date;
  33. import java.util.List;
  35. /**
  36.  * 默认的realm
  37.  *
  38.  * @author 言曌
  39.  * @date 2018/9/1 上午10:47
  40.  */
  41. @Slf4j
  42. public class MyShiroRealm extends AuthorizingRealm {
  44.     @Autowired
  45.     private UserService userService;
  47.     @Autowired
  48.     private RoleService roleService;
  50.     @Autowired
  51.     private PermissionService permissionService;
  53.     @Autowired
  54.     private LocaleMessageUtil localeMessageUtil;
  56.     /**
  57.      * 认证信息(身份验证) Authentication 是用来验证用户身份
  58.      */
  59.     @Override
  60.     protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
  61.         log.info("认证-->MyShiroRealm.doGetAuthenticationInfo()");
  62.         //1.验证用户名
  63.         User user = null;
  64.         String loginName = (String) token.getPrincipal();
  65.         if (Validator.isEmail(loginName)) {
  66.             user = userService.findByEmail(loginName);
  67.         } else {
  68.             user = userService.findByUserName(loginName);
  69.         }
  70.         if (user == null) {
  71.             //用户不存在
  72.             log.info("用户不存在! 登录名:{}, 密码:{}", loginName, token.getCredentials());
  73.             return null;
  74.         }
  76.         //2.首先判断是否已经被禁用已经是否已经过了10分钟
  77.         Date loginLast = DateUtil.date();
  78.         if (null != user.getLoginLast()) {
  79.             loginLast = user.getLoginLast();
  80.         }
  81.         Long between = DateUtil.between(loginLast, DateUtil.date(), DateUnit.MINUTE);
  82.         if (StringUtils.equals(user.getLoginEnable(), TrueFalseEnum.FALSE.getDesc()) && (between < CommonParamsEnum.TEN.getValue())) {
  83.             log.info("账号已锁定! 登录名:{}, 密码:{}", loginName, token.getCredentials());
  84.             throw new LockedAccountException("账号被锁定");
  85.         }
  86.         userService.updateUserLoginLast(user, DateUtil.date());
  88.         //3.封装authenticationInfo，准备验证密码
  89.         SimpleAuthenticationInfo authenticationInfo = new SimpleAuthenticationInfo(
  90.                 user, // 用户名
  91.                 user.getUserPass(), // 密码
  92.                 ByteSource.Util.bytes("sens"), // 盐
  93.                 getName() // realm name
  94.         );
  95.         System.out.println("realName:" + getName());
  96.         return authenticationInfo;
  97.     }
  100.     @Override
  101.     protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
  102.         log.info("授权-->MyShiroRealm.doGetAuthorizationInfo()");
  104.         SimpleAuthorizationInfo authorizationInfo = new SimpleAuthorizationInfo();
  105.         User user = (User) principals.getPrimaryPrincipal();
  107.         List<Role> roles = roleService.listRolesByUserId(user.getUserId());
  108.         for (Role role : roles) {
  109.             authorizationInfo.addRole(role.getRole());
  110.             List<Permission> permissions = permissionService.listPermissionsByRoleId(role.getId());
  111.             for (Permission p : permissions) {
  112.                 authorizationInfo.addStringPermission(p.getPermission());
  113.             }
  114.         }
  115.         return authorizationInfo;
  116.     }
  117. }



上面一个方法是自定义认证方法，查询用户是否存在，然后把密码等信息放到 authenticationInfo 里，shiro 然后进行验证。

下面这个方法是被已登录用户封装角色列表和权限列表。



2.ShiroConfig.java
  1. package com.liuyanzhao.sens.config;
  3. import org.apache.shiro.authc.credential.HashedCredentialsMatcher;
  4. import org.apache.shiro.mgt.SecurityManager;
  5. import org.apache.shiro.realm.Realm;
  6. import org.apache.shiro.spring.LifecycleBeanPostProcessor;
  7. import org.apache.shiro.spring.security.interceptor.AuthorizationAttributeSourceAdvisor;
  8. import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
  9. import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
  10. import org.springframework.aop.framework.autoproxy.DefaultAdvisorAutoProxyCreator;
  11. import org.springframework.context.annotation.Bean;
  12. import org.springframework.context.annotation.Configuration;
  13. import org.springframework.context.annotation.DependsOn;
  15. import java.util.ArrayList;
  16. import java.util.LinkedHashMap;
  17. import java.util.List;
  18. import java.util.Map;
  20. /**
  21.  * @author 言曌
  22.  * @date 2018/8/20 上午6:19
  23.  */
  25. @Configuration
  26. public class ShiroConfig {
  28.     @Bean
  29.     public ShiroFilterFactoryBean shiroFilter(SecurityManager securityManager) {
  30.         System.out.println("ShiroConfiguration.shirFilter()");
  31.         ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();
  32.         shiroFilterFactoryBean.setSecurityManager(securityManager);
  33.         //拦截器.
  34.         Map<String,String> filterChainDefinitionMap = new LinkedHashMap<String,String>();
  35.         // 配置不会被拦截的链接 顺序判断
  36.         filterChainDefinitionMap.put("/static/**", "anon");
  37.         filterChainDefinitionMap.put("/upload/**", "anon");
  38.         filterChainDefinitionMap.put("/favicon.ico", "anon");
  39.         filterChainDefinitionMap.put("/favicon.ico", "anon");
  40.         filterChainDefinitionMap.put("/admin/login", "anon");
  41.         filterChainDefinitionMap.put("/admin/getLogin", "anon");
  42.         //配置退出 过滤器,其中的具体的退出代码Shiro已经替我们实现了
  43.         filterChainDefinitionMap.put("/logout", "logout");
  44.         //<!-- 过滤链定义，从上向下顺序执行，一般将/**放在最为下边 -->:这是一个坑呢，一不小心代码就不好使了;
  45.         //<!-- authc:所有url都必须认证通过才可以访问; anon:所有url都都可以匿名访问-->
  46.         filterChainDefinitionMap.put("/admin/**", "authc");
  47.         filterChainDefinitionMap.put("/backup/**", "authc");
  48.         filterChainDefinitionMap.put("/**", "anon");
  49.         shiroFilterFactoryBean.setFilterChainDefinitionMap(filterChainDefinitionMap);
  51.         // 如果不设置默认会自动寻找Web工程根目录下的"/login"页面
  52.         shiroFilterFactoryBean.setLoginUrl("/admin/login");
  53.         // 登录成功后要跳转的链接
  54.         shiroFilterFactoryBean.setSuccessUrl("/");
  55.         //未授权界面;
  56.         shiroFilterFactoryBean.setUnauthorizedUrl("/403");
  58.         return shiroFilterFactoryBean;
  60.     }
  62.     @Bean
  63.     public SecurityManager securityManager() {
  64.         DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
  65.         securityManager.setRealm(myShiroRealm());
  66.         return securityManager;
  67.     }
  69.     /**
  70.      * 身份认证realm; (这个需要自己写，账号密码校验；权限等)
  71.      *
  72.      * @return MyShiroRealm
  73.      */
  74.     @Bean
  75.     public MyShiroRealm myShiroRealm() {
  76.         MyShiroRealm myShiroRealm = new MyShiroRealm();
  77.         myShiroRealm.setCredentialsMatcher(hashedCredentialsMatcher());
  78.         return myShiroRealm;
  79.     }
  82.     /**
  83.      * 凭证匹配器
  84.      * （由于我们的密码校验交给Shiro的SimpleAuthenticationInfo进行处理了
  85.      *  所以我们需要修改下doGetAuthenticationInfo中的代码;
  86.      * ）
  87.      * @return
  88.      */
  89.     @Bean
  90.     public HashedCredentialsMatcher hashedCredentialsMatcher(){
  91.         HashedCredentialsMatcher hashedCredentialsMatcher = new HashedCredentialsMatcher();
  92.         hashedCredentialsMatcher.setHashAlgorithmName("md5");//散列算法:这里使用MD5算法;
  93.         hashedCredentialsMatcher.setHashIterations(2);//散列的次数，md5(md5("xxx"))
  94.         return hashedCredentialsMatcher;
  95.     }
  98.     /**
  99.      *  开启shiro aop注解支持，不开启无法使用 shiro 的注解.
  100.      *  使用代理方式;所以需要开启代码支持;
  101.      * @param securityManager
  102.      * @return
  103.      */
  104.     /** * Shiro生命周期处理器 * @return */
  105.     @Bean
  106.     public LifecycleBeanPostProcessor lifecycleBeanPostProcessor(){
  107.         return new LifecycleBeanPostProcessor();
  108.     }
  110.     @Bean
  111.     @DependsOn({"lifecycleBeanPostProcessor"})
  112.     public DefaultAdvisorAutoProxyCreator advisorAutoProxyCreator(){
  113.         DefaultAdvisorAutoProxyCreator advisorAutoProxyCreator = new DefaultAdvisorAutoProxyCreator();
  114.         advisorAutoProxyCreator.setProxyTargetClass(true);
  115.         return advisorAutoProxyCreator;
  116.     }
  117.     @Bean
  118.     public AuthorizationAttributeSourceAdvisor authorizationAttributeSourceAdvisor(){
  119.         AuthorizationAttributeSourceAdvisor authorizationAttributeSourceAdvisor = new AuthorizationAttributeSourceAdvisor();
  120.         authorizationAttributeSourceAdvisor.setSecurityManager(securityManager());
  121.         return authorizationAttributeSourceAdvisor;
  122.     }
  123. }


四、登录和退出


下面是登录的后台代码

主要是做一个 subject.login() 的操作，然后 try catch，根据抛的异常，返回指定的信息或者跳转。

这里说明一下，登录成功后，不需要添加 session，shiro 有自己的 SessionDao 创建 session，但是那个 key，可能不知道。

后台获得用户直接从 subject.getPrincipal(); 里拿，前台可以用对应的模板引擎取，或者向后台发请求获得当前用户，建议不用自己弄 session，容易出现两个 session 过期时间不同，尤其是记住密码后。


1.登录代码

  1. /**
  2.    * 验证登录信息
  3.    *
  4.    * @param loginName 登录名：邮箱／用户名
  5.    * @param loginPwd  loginPwd 密码
  6.    * @return JsonResult JsonResult
  7.    */
  8.   @PostMapping(value = "/doLogin")
  9.   @ResponseBody
  10.   public JsonResult getLogin(@ModelAttribute("loginName") String loginName,
  11.                              @ModelAttribute("loginPwd") String loginPwd) {
  13.       Subject subject = SecurityUtils.getSubject();
  14.       UsernamePasswordToken token = new UsernamePasswordToken(loginName, loginPwd);
  15.       try {
  16.           subject.login(token);
  17.           if (subject.isAuthenticated()) {
  18.               //登录成功，修改登录错误次数为0
  19.               User user = (User) subject.getPrincipal();
  20.               userService.updateUserLoginNormal(user);
  21.               return new JsonResult(ResultCodeEnum.SUCCESS.getCode(), "登录成功！");
  22.           }
  23.       } catch (UnknownAccountException e) {
  24.           log.info("UnknownAccountException -- > 账号不存在：");
  25.           return new JsonResult(ResultCodeEnum.FAIL.getCode(), "用户不存在！"));
  26.       } catch (IncorrectCredentialsException e) {
  27.           //更新失败次数
  28.           User user;
  29.           if (Validator.isEmail(loginName)) {
  30.               user = userService.findByEmail(loginName);
  31.           } else {
  32.               user = userService.findByUserName(loginName);
  33.           }
  34.           if (user != null) {
  35.               Integer errorCount = userService.updateUserLoginError(user);
  36.               //超过五次禁用账户
  37.               if (errorCount >= CommonParamsEnum.FIVE.getValue()) {
  38.                   userService.updateUserLoginEnable(user, TrueFalseEnum.FALSE.getDesc());
  39.               }
  40.               int rest = 5 - errorCount;
  41.               return new JsonResult(ResultCodeEnum.FAIL.getCode(), "密码错误，你还可以尝试 [" + rest + "] 次！"));
  42.           }
  43.       } catch (LockedAccountException e) {
  44.           log.info("LockedAccountException -- > 账号被锁定");
  45.           return new JsonResult(ResultCodeEnum.FAIL.getCode(), "账号被锁定，十分钟后再来！");
  46.       } catch (Exception e) {
  47.           log.info(e.getMessage());
  48.       }
  49.       return new JsonResult(ResultCodeEnum.FAIL.getCode(), "发送错误！");
  50.   }


2.退出代码

  1. /**
  2.    * 退出登录
  3.    *
  4.    * @return 重定向到/admin/login
  5.    */
  6.   @GetMapping(value = "/logOut")
  7.   public String logOut() {
  8.       Subject subject = SecurityUtils.getSubject();
  9.       User user = (User) subject.getPrincipal();
  10.       subject.logout();
  11.       log.info("用户[{}]退出登录", user.getUserName());
  12.       return "redirect:/admin/login";
  13.   }


七、权限控制


主要是在方法上加 @RequiresPermissions 注解或者 @RequiresRoles 注解，也可以在类上加(那就是整个类里的所有方法必须要有这个权限)

下面以标签控制器为例
  1. package com.liuyanzhao.sens.web.controller.admin;
  3. import com.liuyanzhao.sens.entity.Tag;
  4. import com.liuyanzhao.sens.model.dto.JsonResult;
  5. import com.liuyanzhao.sens.model.enums.ResultCodeEnum;
  6. import com.liuyanzhao.sens.service.TagService;
  7. import com.liuyanzhao.sens.utils.LocaleMessageUtil;
  8. import lombok.extern.slf4j.Slf4j;
  9. import org.apache.shiro.authz.annotation.RequiresPermissions;
  10. import org.apache.shiro.authz.annotation.RequiresRoles;
  11. import org.springframework.beans.factory.annotation.Autowired;
  12. import org.springframework.stereotype.Controller;
  13. import org.springframework.ui.Model;
  14. import org.springframework.web.bind.annotation.*;
  16. import java.util.List;
  18. /**
  19.  * <pre>
  20.  *     后台标签管理控制器
  21.  * </pre>
  22.  *
  23.  * @author : saysky
  24.  * @date : 2017/12/10
  25.  */
  26. @Slf4j
  27. @Controller
  28. @RequestMapping(value = "/admin/tag")
  29. public class TagController {
  31.     @Autowired
  32.     private TagService tagService;
  34.     @Autowired
  35.     private LocaleMessageUtil localeMessageUtil;
  37.     /**
  38.      * 渲染标签管理页面
  39.      *
  40.      * @return 模板路径admin/admin_tag
  41.      */
  42.     @GetMapping
  43.     @RequiresPermissions("tag:list")
  44.     public String tags(Model model) {
  45.         List<Tag> tags = tagService.findAllTags();
  46.         model.addAttribute("tags", tags);
  47.         return "admin/admin_tag";
  48.     }
  50.     /**
  51.      * 新增/修改标签
  52.      *
  53.      * @param tag tag
  54.      * @return 重定向到/admin/tag
  55.      */
  56.     @PostMapping(value = "/save")
  57.     @RequiresPermissions("tag:save")
  58.     public String saveTag(@ModelAttribute Tag tag) {
  59.         try {
  60.             tagService.saveByTag(tag);
  61.         } catch (Exception e) {
  62.             log.error("新增/修改标签失败：{}", e.getMessage());
  63.         }
  64.         return "redirect:/admin/tag";
  65.     }
  67.     /**
  68.      * 验证是否存在该路径
  69.      *
  70.      * @param tagUrl 标签路径名
  71.      * @return true：不存在，false：已存在
  72.      */
  73.     @GetMapping(value = "/checkUrl")
  74.     @ResponseBody
  75.     @RequiresPermissions("tag:url:check")
  76.     public JsonResult checkTagUrlExists(@RequestParam("tagUrl") String tagUrl) {
  77.         Tag tag = tagService.findByTagUrl(tagUrl);
  78.         if (null != tag) {
  79.             return new JsonResult(ResultCodeEnum.FAIL.getCode(), localeMessageUtil.getMessage("code.admin.common.url-is-exists"));
  80.         }
  81.         return new JsonResult(ResultCodeEnum.SUCCESS.getCode(), "");
  82.     }
  84.     /**
  85.      * 删除标签
  86.      *
  87.      * @param tagId 标签Id
  88.      * @return JsonResult
  89.      */
  90.     @GetMapping(value = "/remove")
  91.     @ResponseBody
  92.     @RequiresPermissions("tag:delete")
  93.     public JsonResult checkDelete(@RequestParam("tagId") Long tagId) {
  94.         tagService.removeByTagId(tagId);
  95.         return new JsonResult(ResultCodeEnum.FAIL.getCode(), localeMessageUtil.getMessage("code.admin.common.delete-success"));
  97.     }
  99.     /**
  100.      * 跳转到修改标签页面
  101.      *
  102.      * @param model model
  103.      * @param tagId 标签编号
  104.      * @return 模板路径admin/admin_tag
  105.      */
  106.     @GetMapping(value = "/edit")
  107.     @RequiresPermissions("tag:edit")
  108.     public String toEditTag(Model model, @RequestParam("tagId") Long tagId) {
  109.         //当前修改的标签
  110.         Tag tag = tagService.findByTagId(tagId);
  111.         model.addAttribute("updateTag", tag);
  113.         //所有标签
  114.         List<Tag> tags = tagService.findAllTags();
  115.         model.addAttribute("tags", tags);
  116.         return "admin/admin_tag";
  117.     }
  118. }



本文介绍至此完毕，核心部分为 ShiroRealm.java 和 ShiroConfig.java 两个类





历史上的今天
一月
24日
  • 微信
  • 交流学习，资料分享
  • weinxin
  • 个人淘宝
  • 店铺名：言曌博客咨询部

  • (部分商品未及时上架淘宝)
avatar

发表评论

avatar 登录者：匿名
匿名评论，评论回复后会有邮件通知

  

已通过评论：2   待审核评论数：0
  1. avatar 访客
    回复 2019年04月07日 23:27:19

    文章不错支持一下吧

  2. avatar 刘慧斌
    回复 2019年02月25日 09:27:38

    springboot 整合shiro有Git链接吗，想学习下