第七讲 自定义Realm实现授权

1、仅仅通过配置文件来指定权限不够灵活，并且不方便，在实际的应用中大多数情况下都是将用户信息，角色信息，权限信息保存到了数据库中。所以需要从数据库中去获取相关的数据信息。可以使用shiro提供的JdbcRealm来实现，也可以自定义Realm来实现。使用JdbcRealm往往也不够灵活。所以在实际应用中大多数情况下都是自定义realm来实现。

2、自定义Realm需要继承AuthorizingRealm，代码如下：

 package com.sun123.realm;

 import java.util.ArrayList;
 import java.util.List;

 import org.apache.shiro.authc.AuthenticationException;
 import org.apache.shiro.authc.AuthenticationInfo;
 import org.apache.shiro.authc.AuthenticationToken;
 import org.apache.shiro.authc.SimpleAuthenticationInfo;
 import org.apache.shiro.authz.AuthorizationInfo;
 import org.apache.shiro.authz.SimpleAuthorizationInfo;
 import org.apache.shiro.realm.AuthorizingRealm;
 import org.apache.shiro.subject.PrincipalCollection;

 public class UserRealm extends AuthorizingRealm {

     /**
      * 自定义realm的实现    该realm类提供了两个方法
      * doGetAuthorizationInfo    获取认证信息
      * doGetAuthenticationInfo    获取权限信息
      */
     @Override
     public String getName() {
         // 自定义
         return "userRealm";
     }

     // 完成身份认证，并且返回认证信息
     // 如果身份认证失败，返回null
     @Override
     protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
         // 获取用户输入的用户名
         String username = (String) token.getPrincipal();// 获取身份信息
         System.out.println("username:" + username);
         // 根据用户名到数据库查询密码信息——模拟
         // 假定从数据库获取的密码为1111
         String pwd = "1111";
         // 将从数据库中查询的信息封装到SimpleAuthenticationInfo中
         SimpleAuthenticationInfo info = new SimpleAuthenticationInfo(username, pwd, getName());
         return info;
     }

     // 授权的信息
     @Override
     protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principal) {
         String username = principal.getPrimaryPrincipal().toString();
         System.out.println("授权————————————————————————");
         System.out.println("username==========="+username);
         //根据用户名到数据库查询该用户对应的信息——————模拟
         List<String> permission = new ArrayList<String>();
         permission.add("user:add");
         permission.add("user:delete");
         permission.add("user:update");
         permission.add("user:find");

         SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
         for (String perms : permission) {
             info.addStringPermission(perms);
         }
         return info;
     }

 }

shiro.ini

 [main]
 userRealm=com.sun123.realm.UserRealm
 securityManager.realm=$userRealm
 #在realm中给定了用户信息，该用户信息可以不用配置
 [users]
 zhangsan=1111

测试代码：

 package com.sun123.shiro;

 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.config.IniSecurityManagerFactory;
 import org.apache.shiro.util.Factory;
 //java.lang中有SecurityManager的包，需要改成Apache的
 import org.apache.shiro.mgt.SecurityManager;
 import org.apache.shiro.subject.Subject;

 public class UserRealmDemo {

     public static void main(String[] args) {
         // 1、创建SecurityManager工厂，读取相应的配置文件
         Factory<SecurityManager> factory = new IniSecurityManagerFactory("classpath:shiro.ini");
         // 2、通过SecurityManager工厂获取SecurityManager的实例
         SecurityManager securityManager = factory.getInstance();
         // 3、将securityManager对象设置到运行环境中
         SecurityUtils.setSecurityManager(securityManager);
         // 4、通过SecurityUtils获取主体Subject
         Subject subject = SecurityUtils.getSubject();
         // 5、加入登录的用户名zhangsan和1111，这个地方的zhangsan和1111表示用户登录时输入的信息
         // 而shiro.ini文件中的信息相当于数据库中存放的信息
         // UsernamePasswordToken token = new UsernamePasswordToken("zhangsan", "111");
         UsernamePasswordToken token = new UsernamePasswordToken("zhangsan", "1111");
         try {
             // 6、进行用户身份验证
             subject.login(token);
             // 7、通过subject来判断用户是否通过验证
             if (subject.isAuthenticated()) {
                 System.out.println("验证通过");
             }
         } catch (UnknownAccountException e) {// AuthenticationException为父异常 Ctrl+T看子异常
             System.out.println("用户名/密码不正确");
         } catch (IncorrectCredentialsException e) {// AuthenticationException为父异常 Ctrl+T看子异常
             System.out.println("用户名或密码不正确");
         }

         System.out.println(subject.isPermittedAll("user:add","user:delete"));
     }
 }