【WEEK14】【DAY1】Shiro第六部分【中文版】

ZzzZzzzZzzzz-

已于 2024-05-29 08:03:05 修改

阅读量689

点赞数 23

分类专栏： Spring Boot学习文章标签： spring boot 后端 java

于 2024-05-28 14:00:00 首次发布

本文链接：https://blog.csdn.net/2401_83329143/article/details/139053546

版权

Spring Boot学习专栏收录该内容

74 篇文章 0 订阅

订阅专栏

2024.5.27 Monday

15.7.Shiro请求授权的实现

接上文【WEEK13】【DAY5】Shiro第五部分【中文版】

15.7.6.修改数据库相关的参数，使用真实数据库的数据

15.7.6.1.修改数据库

因为在现实中，用户是否有访问某一功能（页面）的权限应该储存在数据库中，程序在读取后加以判断相应的页面是否对指定用户开放。前文在UserRealm中使用addStringPermission方法添加授权的方法并不合理，又因为当前的数据库中缺少用户权限栏，所以在此需要添加新的列。
在这里插入图片描述

给用户分别授予权限：

lisi和wangwu没有任何权限，也不能直接空置perms列，需要填null或user:null，否则在尝试访问add或update时，出现的报错页面仍然不是修改过的提示语句，还是默认的401页面。

15.7.6.2.修改User.java

添加一行参数

package com.P40.pojo;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class User {
    private int id;
    private String name;
    private String pwd;
    private String perms;
}

15.7.7.修改UserRealm.java

package com.P40.config;

import com.P40.pojo.User;
import com.P40.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;

//UserRealm是一个bean
//自定义的UserRealm,必须要继承AuthorizingRealm方法，然后implement methods(alt+insert)
public class UserRealm extends AuthorizingRealm {

    @Autowired
    UserService userService;

    //授权authorization
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
        System.out.println("do doGetAuthorizationInfo 授权");

        //authorization
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();   //可以先写new SimpleAuthorizationInfo();然后使用Alt+Enter快捷创建前面的框架
//        info.addStringPermission("user:add");   //给所有登录的用户增加访问User：add的权限，此时登录任一用户都能访问add页面

        //拿到当前登录的对象
        Subject subject = SecurityUtils.getSubject();   //代表当前登录的用户
        User currentUser = (User) subject.getPrincipal();//从SimpleAuthenticationInfo(user,user.getPwd(),"");中第一个参数（user）取出，得到user对象
        //设置当前用户的权限
        info.addStringPermission(currentUser.getPerms());

        //不能return null
        return info;
    }

    //认证authentication
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authenticationToken) throws AuthenticationException {
        System.out.println("do doGetAuthenticationInfo 认证");

        UsernamePasswordToken userToken = (UsernamePasswordToken) authenticationToken;

        //改为连接真实数据库
        User user = userService.queryUserByName(userToken.getUsername());
        if(user == null){   //此用户不存在
            return null;    //UnknownAccountException
        }

        //密码认证,shiro运行
        return new SimpleAuthenticationInfo(user,user.getPwd(),"");
    }
}

15.7.8.修改ShiroConfig.java

添加授权

package com.P40.config;

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();
        //要配置什么：点击进入ShiroFilterFactoryBean源码查看
        //设置安全管理器
        bean.setSecurityManager(defaultWebSecurityManager);

        //添加shiro的内置过滤器
        /*
        anon： 无需认证就可以访问
        authc： 必须认证了才能访问
        user： 必须拥有记住我功能才能用
        perms： 拥有对某个资源的权限才能访问
        role： 拥有某个角色权限
         */
        //登录拦截
        Map<String,String> filterMap = new LinkedHashMap<>();

        //授权。正常情况下，应该跳转到未授权页面，但是此时由于只添加了以下的验证，导致直接跳转到401页面
        filterMap.put("/user/add","perms[user:add]");
        filterMap.put("/user/update","perms[user:update]");

//        filterMap.put("/user/add","authc");
//        filterMap.put("/user/update","authc");
        //仅在此处修改add和update页面的访问权限后，重启项目，点击add或update都将被拦截，显示404错误，希望跳转到登录页面
        filterMap.put("/user/*","authc");   //也可以使用通配符实现（替代上两行的/user/add和/user/update）
        bean.setFilterChainDefinitionMap(filterMap);

        //如果没有权限，则需要跳转到登录页
        bean.setLoginUrl("/toLogin");   //设置登录的请求

        //未授权页面
        bean.setUnauthorizedUrl("/noauth");

        return bean;
    }

    //DefaultWebSecurityManager
    @Bean(name = "securityManager") //给这个类起别名，便于ShiroFilterFactoryBean调用
    public DefaultWebSecurityManager getDefaultWebSecurityManager(@Qualifier("userRealm") UserRealm userRealm){    //获取UserRealm,但是这里好像不需要使用注解，可以直接调用
        DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
        //默认的DefaultWebSecurityManager类的类名是defaultWebSecurityManager，只是在这里修改成securityManager

        //关联UserRealm
        securityManager.setRealm(userRealm);
        return securityManager;
    }

    //创建realm对象，需要自定义类
    @Bean
    public UserRealm userRealm(){
        return new UserRealm();
    }

    //创建时顺序是相反的（从real->DefaultWebSecurityManager->ShiroFilterFactoryBean）
}

15.7.9.新增退出登录功能

15.7.9.1.修改index.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<h1>首页</h1>
<p th:text="${msg}"></p>

<hr>    <!--在 HTML 页面中创建一条水平线-->
<a th:href="@{/user/add}">add</a>  |   <a th:href="@{/user/update}">update</a>
<!--a标签定义超链接，用于从一个页面链接到另一个页面-->
<hr>
<a class="nav-link" th:href="@{/logout}" onclick="alert('已退出')">Sign out退出登录</a><!--单词別拼错了href不是herf-->

</body>
</html>

15.7.9.2.修改MyController.java

package com.P40.controller;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
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;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpSession;

@Controller
public class MyController {

    @RequestMapping({"/", "/index"})
    public String toIndex(Model model){
        model.addAttribute("msg","hello, Shiro");
        return "index";
    }


    @RequestMapping("/user/add")
    public String add(){
        return "user/add";
    }

    @RequestMapping("/user/update")
    public String update(){
        return "user/update";
    }

    @RequestMapping("/toLogin")
    public String toLogin(){
        return "login";
    }


    @RequestMapping("/login")
    public String login(String username, String password, Model model){
        //获取当前的用户
        Subject subject = SecurityUtils.getSubject();

        //封装用户的登录数据
        UsernamePasswordToken token = new UsernamePasswordToken(username,password);

        //执行登录的方法，没有异常则成功了，选中代码后Ctrl+Alt+T添加try catch
        try {
            subject.login(token);
            return "index";
        } catch (UnknownAccountException e) {   //用户名不存在
            model.addAttribute("msg","用户名错误");
            return "login";
        } catch (IncorrectCredentialsException e) {   //密码错误
            model.addAttribute("msg","密码错误");
            return "login";
        }
    }

    @RequestMapping("/noauth")
    @ResponseBody
    public String unauthorized(){
        return "未授权，禁止访问此页面";
    }

    //退出登录
    @RequestMapping("/logout")
    public String logout(){
        Subject currentUser = SecurityUtils.getSubject();
        currentUser.logout();
        return "/login";
    }

}