SpringBoot20-整合-Shiro-框架实现用户权限管理

共计 7203 个字符，预计需要花费 19 分钟才能阅读完成。

GitHub 源码地址：知了一笑
https://github.com/cicadasmile/middle-ware-parent

Apache Shiro 是一个强大且易用的 Java 安全框架, 执行身份验证、授权、密码和会话管理。作为一款安全框架 Shiro 的设计相当巧妙。Shiro 的应用不依赖任何容器，它不仅可以在 JavaEE 下使用，还可以应用在 JavaSE 环境中。

1）Subject：认证主体

代表当前系统的使用者，就是用户，在 Shiro 的认证中，认证主体通常就是 userName 和 passWord，或者其他用户相关的唯一标识。

2）SecurityManager：安全管理器

Shiro 架构中最核心的组件，通过它可以协调其他组件完成用户认证和授权。实际上，SecurityManager 就是 Shiro 框架的控制器。

3）Realm：域对象

定义了访问数据的方式，用来连接不同的数据源，如：关系数据库，配置文件等等。

Shiro 自己不维护用户和权限，通过 Subject 用户主体和 Realm 域对象的注入，完成用户的认证和授权。

<dependency>
    <groupId>org.apache.shiro</groupId>
    <artifactId>shiro-core</artifactId>
    <version>1.4.0</version>
</dependency>
<dependency>
    <groupId>org.apache.shiro</groupId>
    <artifactId>shiro-spring</artifactId>
    <version>1.4.0</version>
</dependency>

@Configuration
public class ShiroConfig {
    /**
     * Session Manager：会话管理
     * 即用户登录后就是一次会话，在没有退出之前，它的所有信息都在会话中；* 会话可以是普通 JavaSE 环境的，也可以是如 Web 环境的；*/
    @Bean("sessionManager")
    public SessionManager sessionManager(){DefaultWebSessionManager sessionManager = new DefaultWebSessionManager();
        // 设置 session 过期时间
        sessionManager.setGlobalSessionTimeout(60 * 60 * 1000);
        sessionManager.setSessionValidationSchedulerEnabled(true);
        // 去掉 shiro 登录时 url 里的 JSESSIONID
        sessionManager.setSessionIdUrlRewritingEnabled(false);
        return sessionManager;
    }

    /**
     * SecurityManager：安全管理器
     */
    @Bean("securityManager")
    public SecurityManager securityManager(UserRealm userRealm, SessionManager sessionManager) {DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
        securityManager.setSessionManager(sessionManager);
        securityManager.setRealm(userRealm);
        return securityManager;
    }
    /**
     * ShiroFilter 是整个 Shiro 的入口点，用于拦截需要安全控制的请求进行处理
     */
    @Bean("shiroFilter")
    public ShiroFilterFactoryBean shiroFilter(SecurityManager securityManager) {ShiroFilterFactoryBean shiroFilter = new ShiroFilterFactoryBean();
        shiroFilter.setSecurityManager(securityManager);
        shiroFilter.setLoginUrl("/userLogin");
        shiroFilter.setUnauthorizedUrl("/");
        Map<String, String> filterMap = new LinkedHashMap<>();
        filterMap.put("/userLogin", "anon");
        shiroFilter.setFilterChainDefinitionMap(filterMap);
        return shiroFilter;
    }
    /**
     * 管理 Shiro 中一些 bean 的生命周期
     */
    @Bean("lifecycleBeanPostProcessor")
    public LifecycleBeanPostProcessor lifecycleBeanPostProcessor() {return new LifecycleBeanPostProcessor();
    }
    /**
     * 扫描上下文，寻找所有的 Advistor( 通知器）* 将这些 Advisor 应用到所有符合切入点的 Bean 中。*/
    @Bean
    public DefaultAdvisorAutoProxyCreator defaultAdvisorAutoProxyCreator() {DefaultAdvisorAutoProxyCreator proxyCreator = new DefaultAdvisorAutoProxyCreator();
        proxyCreator.setProxyTargetClass(true);
        return proxyCreator;
    }
    /**
     * 匹配所有加了 Shiro 认证注解的方法
     */
    @Bean
    public AuthorizationAttributeSourceAdvisor authorizationAttributeSourceAdvisor(SecurityManager securityManager) {AuthorizationAttributeSourceAdvisor advisor = new AuthorizationAttributeSourceAdvisor();
        advisor.setSecurityManager(securityManager);
        return advisor;
    }
}

@Component
public class UserRealm extends AuthorizingRealm {
    @Resource
    private SysUserMapper sysUserMapper ;
    @Resource
    private SysMenuMapper sysMenuMapper ;
    /**
     * 授权 (验证权限时调用)
     * 获取用户权限集合
     */
    @Override
    public AuthorizationInfo doGetAuthorizationInfo
    (PrincipalCollection principals) {SysUserEntity user = (SysUserEntity)principals.getPrimaryPrincipal();
        if(user == null) {throw new UnknownAccountException("账号不存在");
        }
        List<String> permsList;
        // 默认用户拥有最高权限
        List<SysMenuEntity> menuList = sysMenuMapper.selectList();
        permsList = new ArrayList<>(menuList.size());
        for(SysMenuEntity menu : menuList){permsList.add(menu.getPerms());
        }
        // 用户权限列表
        Set<String> permsSet = new HashSet<>();
        for(String perms : permsList){if(StringUtils.isEmpty(perms)){continue;}
            permsSet.addAll(Arrays.asList(perms.trim().split(",")));
        }
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
        info.setStringPermissions(permsSet);
        return info;
    }
    /**
     * 认证 (登录时调用)
     * 验证用户登录
     */
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authToken) throws AuthenticationException {UsernamePasswordToken token = (UsernamePasswordToken)authToken;
        // 查询用户信息
        SysUserEntity user = sysUserMapper.selectOne(token.getUsername());
        // 账号不存在
        if(user == null) {throw new UnknownAccountException("账号或密码不正确");
        }
        // 账号锁定
        if(user.getStatus() == 0){throw new LockedAccountException("账号已被锁定, 请联系管理员");
        }
        SimpleAuthenticationInfo info = new SimpleAuthenticationInfo
                (user, user.getPassword(),
                        ByteSource.Util.bytes(user.getSalt()),
                        getName());
        return info;
    }
    @Override
    public void setCredentialsMatcher(CredentialsMatcher credentialsMatcher) {HashedCredentialsMatcher shaCredentialsMatcher = new HashedCredentialsMatcher();
        shaCredentialsMatcher.setHashAlgorithmName(ShiroUtils.hashAlgorithmName);
        shaCredentialsMatcher.setHashIterations(ShiroUtils.hashIterations);
        super.setCredentialsMatcher(shaCredentialsMatcher);
    }
}

public class ShiroUtils {
    /**  加密算法 */
    public final static String hashAlgorithmName = "SHA-256";
    /**  循环次数 */
    public final static int hashIterations = 16;
    public static String sha256(String password, String salt) {return new SimpleHash(hashAlgorithmName, password, salt, hashIterations).toString();}
    // 获取一个测试账号 admin
    public static void main(String[] args) {
        // 3743a4c09a17e6f2829febd09ca54e627810001cf255ddcae9dabd288a949c4a
        System.out.println(sha256("admin","123")) ;
    }
    /**
     * 获取会话
     */
    public static Session getSession() {return SecurityUtils.getSubject().getSession();}
    /**
     * Subject：主体，代表了当前“用户”*/
    public static Subject getSubject() {return SecurityUtils.getSubject();
    }
    public static SysUserEntity getUserEntity() {return (SysUserEntity)SecurityUtils.getSubject().getPrincipal();
    }
    public static Long getUserId() {return getUserEntity().getUserId();}
    public static void setSessionAttribute(Object key, Object value) {getSession().setAttribute(key, value);
    }
    public static Object getSessionAttribute(Object key) {return getSession().getAttribute(key);
    }
    public static boolean isLogin() {return SecurityUtils.getSubject().getPrincipal() != null;}
    public static void logout() {SecurityUtils.getSubject().logout();}
}

@RestControllerAdvice
public class ShiroException {@ExceptionHandler(AuthorizationException.class)
    public String authorizationException (){return "抱歉您没有权限访问该内容!";}
    @ExceptionHandler(Exception.class)
    public String handleException(Exception e){return "系统异常!";}
}

@RestController
public class ShiroController {private static Logger LOGGER = LoggerFactory.getLogger(ShiroController.class) ;
    @Resource
    private SysMenuMapper sysMenuMapper ;
    /**
     * 登录测试
     * http://localhost:7011/userLogin?userName=admin&passWord=admin
     */
    @RequestMapping("/userLogin")
    public void userLogin (@RequestParam(value = "userName") String userName,
            @RequestParam(value = "passWord") String passWord){
        try{Subject subject = ShiroUtils.getSubject();
            UsernamePasswordToken token = new UsernamePasswordToken(userName, passWord);
            subject.login(token);
            LOGGER.info("登录成功");
        }catch (Exception e) {e.printStackTrace();
        }
    }
    /**
     * 服务器每次重启请求该接口之前必须先请求上面登录接口
     * http://localhost:7011/menu/list 获取所有菜单列表
     * 权限要求：sys:user:shiro
     */
    @RequestMapping("/menu/list")
    @RequiresPermissions("sys:user:shiro")
    public List list(){return sysMenuMapper.selectList() ;
    }
    /**
     * 用户没有该权限，无法访问
     * 权限要求：ccc:ddd:bbb
     */
    @RequestMapping("/menu/list2")
    @RequiresPermissions("ccc:ddd:bbb")
    public List list2(){return sysMenuMapper.selectList() ;
    }
    /**
     * 退出测试，退出后没有任何权限
     */
    @RequestMapping("/userLogOut")
    public String logout (){ShiroUtils.logout();
        return "success" ;
    }
}

1)、登录后取得权限
http://localhost:7011/userLogin?userName=admin&passWord=admin
2)、访问有权限接口
http://localhost:7011/menu/list
3)、访问无权限接口
http://localhost:7011/menu/list2
4)、退出登录
http://localhost:7011/userLogOut

GitHub 地址：知了一笑
https://github.com/cicadasmile
码云地址：知了一笑
https://gitee.com/cicadasmile

一、Shiro 简介

1、基础概念

2、核心角色

3、核心理念

二、整合 SpringBoot2 框架

1、核心依赖

2、Shiro 核心配置

3、域对象配置

4、核心工具类

5、自定义权限异常提示

三、案例演示代码

1、测试接口

2、测试流程

四、源代码地址

Just My Socks（注册教程内含优惠码）