shiro （五） spring结合 -- Realm

?

package com.miv.shiro.common;

?

import org.apache.shiro.authc.AuthenticationException;

import org.apache.shiro.authc.AuthenticationInfo;

import org.apache.shiro.authc.AuthenticationToken;

import org.apache.shiro.authc.LockedAccountException;

import org.apache.shiro.authc.SimpleAuthenticationInfo;

import org.apache.shiro.authc.UnknownAccountException;

import org.apache.shiro.authz.AuthorizationException;

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 com.miv.core.constant.DatabaseConstants;

import com.miv.entity.Role;

import com.miv.entity.User;

import com.miv.shiro.login.service.LoginService;

import com.miv.shiro.role.service.RolesService;

?

/**

?* shiro与工程接口类

?*?

?* @author 赵治宇

?* @version

?*/

public class WebRealm extends AuthorizingRealm {

?

? ? private LoginService loginService;

?

? ? private RolesService rolesService;

?

? ? public WebRealm() {

? ? ? ? setName("WebRealm");

? ? }

?

? ? public void setLoginService(LoginService loginService) {

? ? ? ? this.loginService = loginService;

? ? }

?

? ? public void setRolesService(RolesService rolesService) {

? ? ? ? this.rolesService = rolesService;

? ? }

?

? ? /**

? ? ?* 授权方法

? ? ?*?

? ? ?* @return

? ? ?*/

? ? @Override

? ? protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) throws AuthorizationException {

? ? ? ? String userName = (String) principals.fromRealm(getName()).iterator().next();

? ? ? ? User user = new User();

? ? ? ? user.setLoginName(userName);

? ? ? ? try {

? ? ? ? ? ? user = loginService.findUserByUsername(user);

? ? ? ? } catch (Exception e) {

? ? ? ? ? ? // TODO Auto-generated catch block

? ? ? ? ? ? e.printStackTrace();

? ? ? ? }

? ? ? ? SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();

? ? ? ? /*

? ? ? ? ?* List<Roles> roleList = rolesService.getCurrentRoleList(user); if (user != null) {

? ? ? ? ?*?

? ? ? ? ?* for (Roles role : roleList) { info.addRole(role.getRoleName()); } }

? ? ? ? ?*/

? ? ? ? Role role = rolesService.findRoleById(user);

? ? ? ? if (role == null) {

? ? ? ? ? ? throw new AuthorizationException();

? ? ? ? }

? ? ? ? info.addRole(role.getRoleCode());

? ? ? ? return info;

? ? }

?

? ? /**

? ? ?* 认证方法

? ? ?*?

? ? ?* @return

? ? ?*/

? ? @Override

? ? protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken authcToken) throws AuthenticationException {

?

? ? ? ? MIVshiroToken token = (MIVshiroToken) authcToken;

? ? ? ? User user = new User();

? ? ? ? user.setLoginName(token.getUsername());

? ? ? ? user.setPassword(String.valueOf(token.getPassword()));

? ? ? ? Role role = new Role();

? ? ? ? role.setId(token.getROLE_CODE() + 0L);

? ? ? ? user.setRole(role);

? ? ? ? try {

? ? ? ? ? ? user = loginService.findUserByUsernameAndPassword(user);

? ? ? ? } catch (Exception e) {

? ? ? ? ? ? throw new AuthenticationException();

? ? ? ? }

? ? ? ? if (user != null) {

? ? ? ? ? ? boolean flag = token.getROLE_CODE().equals(DatabaseConstants.ROLE_CODE_AGENCY)

? ? ? ? ? ? ? ? ? ? || token.getROLE_CODE().equals(DatabaseConstants.ROLE_CODE_CALL_CENTER)

? ? ? ? ? ? ? ? ? ? || token.getROLE_CODE().equals(DatabaseConstants.ROLE_CODE_ADMIN)

? ? ? ? ? ? ? ? ? ? || token.getROLE_CODE().equals(DatabaseConstants.ROLE_CODE_USER);

? ? ? ? ? ? if (user.getStatus() == DatabaseConstants.STATUS_2.intValue() && flag) {

? ? ? ? ? ? ? ? throw new LockedAccountException();

? ? ? ? ? ? } else {

? ? ? ? ? ? ? ? return new SimpleAuthenticationInfo(user.getLoginName(), user.getPassword(), getName());

? ? ? ? ? ? }

? ? ? ? } else {

? ? ? ? ? ? throw new UnknownAccountException();

? ? ? ? }

? ? }

}