2018-07-09 08:44:52 +08:00
|
|
|
package com.ruoyi.common.utils.security;
|
|
|
|
|
|
|
|
import org.apache.shiro.SecurityUtils;
|
|
|
|
import org.apache.shiro.mgt.RealmSecurityManager;
|
|
|
|
import org.apache.shiro.session.Session;
|
|
|
|
import org.apache.shiro.subject.PrincipalCollection;
|
|
|
|
import org.apache.shiro.subject.SimplePrincipalCollection;
|
|
|
|
import org.apache.shiro.subject.Subject;
|
2018-08-03 10:32:00 +08:00
|
|
|
import com.ruoyi.common.utils.StringUtils;
|
2018-07-12 18:07:04 +08:00
|
|
|
import com.ruoyi.common.utils.bean.BeanUtils;
|
2018-07-09 08:44:52 +08:00
|
|
|
import com.ruoyi.framework.shiro.realm.UserRealm;
|
|
|
|
import com.ruoyi.project.system.user.domain.User;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* shiro 工具类
|
|
|
|
*
|
|
|
|
* @author ruoyi
|
|
|
|
*/
|
|
|
|
public class ShiroUtils
|
|
|
|
{
|
|
|
|
public static Subject getSubjct()
|
|
|
|
{
|
|
|
|
return SecurityUtils.getSubject();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static Session getSession()
|
|
|
|
{
|
|
|
|
return SecurityUtils.getSubject().getSession();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static void logout()
|
|
|
|
{
|
|
|
|
getSubjct().logout();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static User getUser()
|
|
|
|
{
|
2018-08-03 10:32:00 +08:00
|
|
|
User user = null;
|
|
|
|
Object obj = getSubjct().getPrincipal();
|
|
|
|
if (StringUtils.isNotNull(obj))
|
|
|
|
{
|
|
|
|
user = new User();
|
|
|
|
BeanUtils.copyBeanProp(user, obj);
|
|
|
|
}
|
2018-07-12 18:07:04 +08:00
|
|
|
return user;
|
2018-07-09 08:44:52 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
public static void setUser(User user)
|
|
|
|
{
|
|
|
|
Subject subject = getSubjct();
|
|
|
|
PrincipalCollection principalCollection = subject.getPrincipals();
|
|
|
|
String realmName = principalCollection.getRealmNames().iterator().next();
|
|
|
|
PrincipalCollection newPrincipalCollection = new SimplePrincipalCollection(user, realmName);
|
|
|
|
// 重新加载Principal
|
|
|
|
subject.runAs(newPrincipalCollection);
|
|
|
|
}
|
|
|
|
|
|
|
|
public static void clearCachedAuthorizationInfo()
|
|
|
|
{
|
|
|
|
RealmSecurityManager rsm = (RealmSecurityManager) SecurityUtils.getSecurityManager();
|
|
|
|
UserRealm realm = (UserRealm) rsm.getRealms().iterator().next();
|
|
|
|
realm.clearCachedAuthorizationInfo();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static Long getUserId()
|
|
|
|
{
|
|
|
|
return getUser().getUserId().longValue();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static String getLoginName()
|
|
|
|
{
|
|
|
|
return getUser().getLoginName();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static String getIp()
|
|
|
|
{
|
|
|
|
return getSubjct().getSession().getHost();
|
|
|
|
}
|
|
|
|
|
|
|
|
public static String getSessionId()
|
|
|
|
{
|
|
|
|
return String.valueOf(getSubjct().getSession().getId());
|
|
|
|
}
|
|
|
|
}
|