XiangJuWu/src/main/java/com/izouma/awesomeadmin/shiro/BaseRealm.java

package com.izouma.awesomeadmin.shiro;

import com.izouma.awesomeadmin.model.SysRole;
import com.izouma.awesomeadmin.model.UserInfo;
import com.izouma.awesomeadmin.service.SysRoleService;
import com.izouma.awesomeadmin.service.UserInfoService;
import org.apache.log4j.Logger;
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.springframework.beans.factory.annotation.Autowired;

public abstract class BaseRealm extends AuthorizingRealm {
    protected static Logger logger = Logger.getLogger(PhoneCodeRealm.class);

    @Autowired
    protected UserInfoService userInfoService;

    @Autowired
    protected SysRoleService sysRoleService;

    //授权
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
        try {
            UserInfo userInfo = (UserInfo) principalCollection.getPrimaryPrincipal();
            //获取角色数据权限授权
            SysRole sysRole = sysRoleService.getSysRoleById(userInfo.getRoleId());
            String[] permission = sysRole.getPermission().split(",");
            for (String perm : permission) {
                info.addStringPermission(perm);
            }
        } catch (Exception e) {
            logger.error("doGetAuthorizationInfo", e);
        }
        return info;
    }
}