1. 程式人生 > 程式設計 >Springboot shiro認證授權實現原理及例項

Springboot shiro認證授權實現原理及例項

關於認證授權,需要的資料表有:使用者表,角色表,使用者角色關聯表,許可權表,角色許可權關聯表,一次如下

Springboot shiro認證授權實現原理及例項Springboot shiro認證授權實現原理及例項Springboot shiro認證授權實現原理及例項

Springboot shiro認證授權實現原理及例項Springboot shiro認證授權實現原理及例項

之前寫過了shiro的登入認證,在自定義的realm中,我們實現AuthorizingRealm介面中的方法:package com.zs.springboot.realm;

import com.zs.springboot.model.User;
import com.zs.springboot.service.UserService;
import com.zs.springboot.util.ResultUtil;
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.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.util.ByteSource;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;
import java.util.Map;

/**
 * @author zs
 * 登入realm
 */
public class LoginRealm extends AuthorizingRealm {

  @Autowired
  private UserService userService;
  /**
   * 授權方法,只有頁面出現關於許可權標籤的時候或者專案中關於授權註解的時候,才可以被使用
   * @param principal
   * @return
   */
  @Override
  protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principal) { 
    User user = (User) principal.getPrimaryPrincipal();
    Map<String,Object> resultMap1 = userService.getRoleByUid(user.getUid());
    //獲取角色,通過多表查詢獲取使用者對應的角色
    List<String> roleList = (List<String>) ResultUtil.getResult(resultMap1);
    SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
    info.addRoles(roleList);
    //獲取許可權,通過關聯中間表,多表查詢出該使用者對應的角色所擁有的許可權
    Map<String,Object> resultMap2 = userService.getPromessionByUid(user.getUid());
    List<String> permissionList = (List<String>) ResultUtil.getResult(resultMap2);
    info.addStringPermissions(permissionList);
    return info;
  }

  /**
   * 認證方法
   * @param token
   * @return
   * @throws AuthenticationException
   */
  @Override
  protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
    //獲取當前登入的使用者名稱
    String username = (String) token.getPrincipal();
    //根據使用者到資料庫搜尋使用者資訊
    Map<String,Object> login = userService.login(username);
    //如果使用者不存在則丟擲異常
    if ((Integer) login.get("code") == 404) {
      throw new UnknownAccountException("使用者不存在");
    }
    //如果使用者存在,獲取使用者資訊
    User user = (User) login.get("result");
    //進行認證
    SimpleAuthenticationInfo info = new SimpleAuthenticationInfo(user,user.getPassword(),ByteSource.Util.bytes(user.getSalt()),this.getName());
    //將使用者資訊放入session中,密碼制空
    Session session = SecurityUtils.getSubject().getSession();
    user.setPassword(null);
    session.setAttribute("user",user);
    return info;
  }
}

注意,在授權方法中,principal.getPrimaryPrincipal()方法;如果在認證方法中,simpleAuthenticationInfo物件中傳遞的第一個引數如果為obj物件,則返回值為物件(可以強轉為User物件),如果傳遞的時username字串,則返回的是username字串,字串是不能強轉為User物件的。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。