要实现的目的:根据登录用户、查询出当前用户具有的所有权限、然后登录系统后、根据查询到的权限信息进行不同的操作。
以下的代码是在搭好的框架之下进行的编码。
文章目录
- 核心实现部分。
- 第一种是将用户表和角色表进行一次查询、角色表和权限表进行一次查询。
- 第二种实现权限的查询(只需要一条sql语句)耦合性太大了
请参考这个框架的搭建以及实现shiro用户登录的验证: https://blog.csdn.net/weixin_43304253/article/details/121111530
我直接贴代码、在关键的部分我会叙述一下。代码更具有说服力
编写的数据库表
用户表、用户角色关联表、角色表、角色权限关联表、权限表。(这个我就不贴代码了、就是简单的建表)
- 一个用户可以对应一个角色、也可以对应多个角色(看自己的需求设计了)
- 一个角色往往有不同的权限。
实体类
角色表
package com.zheng.pojo;
public class Role {
private String Rid;
private String Rname;
public String getRid() {
return Rid;
}
public void setRid(String rid) {
Rid = rid;
}
public String getRname() {
return Rname;
}
public void setRname(String rname) {
Rname = rname;
}
}
权限表
package com.zheng.pojo;
public class Perms {
private String Pid;
private String Pname;
public String getPid() {
return Pid;
}
public void setPid(String pid) {
Pid = pid;
}
public String getPname() {
return Pname;
}
public void setPname(String pname) {
Pname = pname;
}
}
controller层
package com.zheng.controller;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.IncorrectCredentialsException;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
@Controller
public class UserController {
@RequestMapping("/toLogin")
public String index(){
return "login";
}
//跳转到未授权界面
@RequestMapping("/ungrant")
public String ungrant(){
return "ungrant";
}
//跳转到add
@RequestMapping("/admin/add")
public String addpPage(){
return "add";
}
//跳转到show
@RequestMapping("/admin/show")
public String showPage(){
return "show";
}
@RequestMapping("/admin/update")
public String updatePage(){
return "update";
}
@RequestMapping("/index")
public String toIndex(){
return "index";
}
@RequestMapping("/login")
public String login(String username, String password, Model model){
//获取当前的用户
Subject subject= SecurityUtils.getSubject();
//封装用户的登录数据
UsernamePasswordToken token= new UsernamePasswordToken(username,password);
try {
subject.login(token);//执行登陆的方法,如果没有异常则是正确的
return "index";//进入个人信息界面
} catch (UnknownAccountException e) {
model.addAttribute("msg","用户名错误");
return "login";
}catch (IncorrectCredentialsException e){
model.addAttribute("msg","密码不存在");
return "login";
}
}
}
srvice层
角色
package com.zheng.service;
import com.zheng.pojo.Role;
import java.util.Set;
public interface RoleService {
//查询当前用户具有的角色
Set<Role> queryRole(String name);
}
角色实现
package com.zheng.service.impl;
import com.zheng.mapper.RoleMapper;
import com.zheng.pojo.Role;
import com.zheng.service.RoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.Set;
@Service
public class RoleServiceImpl implements RoleService {
@Autowired
RoleMapper roleMapper;
//查询当前用户具有的角色
@Override
public Set<Role> queryRole(String name) {
Set<Role> Roleset = roleMapper.queryRole(name);
int a=3;
return Roleset ;
}
}
权限
package com.zheng.service;
import com.zheng.pojo.Role;
import java.util.Set;
public interface RoleService {
//查询当前用户具有的角色
Set<Role> queryRole(String name);
}
权限实现
package com.zheng.service.impl;
import com.zheng.mapper.PermsMapper;
import com.zheng.pojo.Perms;
import com.zheng.service.PermService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.Set;
@Service
public class PermServiceImpl implements PermService {
@Autowired
PermsMapper permsMapper;
//查询当前角色拥有的权限
@Override
public Set<Perms> queryPerms(String Rid) {
return permsMapper.queryPerms(Rid);
}
}
dao层
角色
package com.zheng.mapper;
import com.zheng.pojo.Role;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Repository;
import java.util.Set;
@Mapper //这个注解表示这个是mybatis的mapeper
@Repository
public interface RoleMapper {
//查询当前用户具有的角色
Set<Role> queryRole(String name);
}
权限
package com.zheng.mapper;
import com.zheng.pojo.Perms;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Repository;
import java.util.Set;
@Mapper //这个注解表示这个是mybatis的mapeper
@Repository
public interface PermsMapper {
//查询当前角色拥有的权限
Set<Perms> queryPerms(String Rid);
}
编写的mapper
角色
<?xml version="1.0" encoding="UTF8"?>
<!DOCTYPE mapper
PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.zheng.mapper.RoleMapper">
<!--查询某一个用户的信息-->
<select id="queryRole" resultType="com.zheng.pojo.Role" >
select role.Rid
from user,role,us_ro
where us_ro.name=user.name and us_ro.Rid=role.Rid and user.name=#{name}
</select>
</mapper>
权限
<?xml version="1.0" encoding="UTF8"?>
<!DOCTYPE mapper
PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.zheng.mapper.PermsMapper">
<!--查询某一个用户的信息-->
<select id="queryPerms" resultType="com.zheng.pojo.Perms" >
select perms.Pname
from perms,rop,role
where rop.Pid=perms.Pid and rop.Rid=role.Rid and role.Rid=#{Rid}
</select>
</mapper>
看一下实现的效果:
1、第一种情况、还未授权就访问首页里的内容
2、第二种情况、验证登录后的情况。根据当前用户的权限进行跟踪操作。
这里的admin:add和admin:show是来自两个不同角色的
1、add权限来自角色 1 2、show权限来自角色2。这里就实现了一个用户对应多个角色、然后不同的角色对应不同的权限。
来看效果
有兴趣的小伙伴可以看一下debug模式下的权限
当点击add选项时的debug
点击show链接的时候
核心实现部分。
第一种是将用户表和角色表进行一次查询、角色表和权限表进行一次查询。
这边是过滤权限
package com.zheng.config;
import at.pollux.thymeleaf.shiro.dialect.ShiroDialect;
import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.util.LinkedHashMap;
import java.util.Map;
@Configuration
public class ShiroConfig {
//ShiroFilterFactoryBean
@Bean
public ShiroFilterFactoryBean getShiroFilterFactoryBean(@Qualifier("securitymanager") DefaultWebSecurityManager defaultWebSecurityManager) {
ShiroFilterFactoryBean bean = new ShiroFilterFactoryBean();
//设置安全管理器
bean.setSecurityManager(defaultWebSecurityManager);
//添加shiro的内置过滤器
/**
* anon:无需认证就可访问 * authc:必须认证才能访问* User:必须拥有 记住我 功能才能用
* perms:拥有对某个资源的权限才能访问* role:拥有某个角色权限才能访问
*/
Map<String, String> filterMap = new LinkedHashMap<>();
//授权,正常的情况下,没有授权会跳转到未授权页面
// filterMap.put("/admin/*","authc");//admin请求下的都需要认证
filterMap.put("/admin/add", "perms[admin:add]");
filterMap.put("/admin/show","perms[admin:show]");
filterMap.put("/admin/update", "perms[user:update]");
bean.setFilterChainDefinitionMap(filterMap);
//如果没有认证、设置登录的请求
bean.setLoginUrl("/toLogin");
//如果没有授权,跳转到未收取页面
bean.setUnauthorizedUrl("/ungrant");
return bean;
}
//DefaultWebSecurityManager
@Bean(name = "securitymanager")
public DefaultWebSecurityManager getDefaultWebSecurityManager(@Qualifier("userRealm") UserRealm userRealm) {
DefaultWebSecurityManager securitymanager = new DefaultWebSecurityManager();
//关联UserRealm
securitymanager.setRealm(userRealm);
return securitymanager;
}
//创建realm对象,需要自定义安装
@Bean
public UserRealm userRealm() {
return new UserRealm();
}
//整合ShiroDialect:用来整合shiro thymeleaf
@Bean
public ShiroDialect getShiroDialect() {
return new ShiroDialect();
}
}
这边是认证和授权
package com.zheng.config;
import com.zheng.pojo.Perms;
import com.zheng.pojo.Role;
import com.zheng.pojo.User;
import com.zheng.service.PermService;
import com.zheng.service.RoleService;
import com.zheng.service.UserService;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.*;
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.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
//自定义的UserRealm
public class UserRealm extends AuthorizingRealm {
@Autowired
UserService userService;
@Autowired
RoleService roleServic;
@Autowired
PermService permService;
//授权
@Override
protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
System.out.println("执行了授权");
SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
//拿到当前的登录对象
Subject subject = SecurityUtils.getSubject();
User user = (User)subject.getPrincipal();
//设置当前用户的权限
Collection<String> perms = new HashSet<String>();//设置集合
//查询当前用户的角色
Set<Role> roleSet =roleServic.queryRole(user.getName());
//迭代器迭代集合元素 遍历
Iterator<Role> iterator = roleSet.iterator();
while(iterator.hasNext()){
Role role1 =iterator.next();
Set<Perms> permsSet=permService.queryPerms(role1.getRid());//查询当前角色具有的权限
for(Perms perm : permsSet){
perms.add(perm.getPname());//将每一个当前用户的权限加入
}
}
info.addStringPermissions(perms);
return info;
}
//认证
@Override
protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
// 第一种方式
// 获取用户输入的账号和密码(一般只需要获取账号就可以)
System.out.println("执行了认证");
UsernamePasswordToken userToken = (UsernamePasswordToken) token;
//连接真实的数据库
User user = userService.login(userToken.getUsername());
//1、思路:在用户表中新增加一个权限字段、然后再次查询数据库获得该用户所具有的权限(该权限为要给集合??????
if(user == null){
//没有这个人
return null;
}
//密码认证
return new SimpleAuthenticationInfo(user,user.getPassword(),"");
}
}
两次分开查询
第二种实现权限的查询(只需要一条sql语句)耦合性太大了
这个是我之前写的用一条语句写的、我不想在重复写一次了、看的快吐了。直接贴以前的代码了、操作步骤和上边的一模一样。
这里是sql、看的头晕、写的也头晕
<select id="perms" resultType="com.zheng.pojo.Perms">
select perms.Pname from
user,us_ro,role,rop,perms
where us_ro.name=user.name and us_ro.Rid=role.Rid and rop.Rid=role.Rid and rop.Pid=perms.Pid and user.name=#{name}
</select>