|
1
2
|
package com.huaheng.common.utils.security;
|
|
3
4
|
import com.huaheng.common.utils.StringUtils;
import com.huaheng.common.utils.bean.BeanUtils;
|
|
5
6
7
8
9
10
11
|
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;
import com.huaheng.framework.shiro.realm.UserRealm;
|
|
12
|
import com.huaheng.pc.system.user.domain.User;
|
|
13
|
|
|
14
15
|
import java.util.List;
|
|
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
/**
* shiro 工具类
*
* @author huaheng
*/
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()
{
|
|
41
42
43
44
45
46
47
48
49
|
Object obj = getSubjct().getPrincipal();
if (StringUtils.isNotNull(obj)) {
User user = new User();
BeanUtils.copyBeanProp(user, obj);
return user;
}
else {
return null;
}
|
|
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
}
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();
}
|
|
69
|
public static Integer getUserId()
|
|
70
|
{
|
|
71
|
return getUser().getId();
|
|
72
73
74
75
76
77
78
|
}
public static String getLoginName()
{
return getUser().getLoginName();
}
|
|
79
80
81
82
83
84
85
86
87
88
|
public static Integer getWarehouseId()
{
return getUser().getWarehouseId();
}
public static String getWarehouseCode()
{
return getUser().getWarehouseCode();
}
|
|
89
|
public static List<Integer> getCompanyIdList()
|
|
90
|
{
|
|
91
|
return getUser().getCompanyIdList();
|
|
92
93
|
}
|
|
94
|
public static List<String> getCompanyCodeList()
|
|
95
|
{
|
|
96
|
return getUser().getCompanyCodeList();
|
|
97
98
|
}
|
|
99
100
101
102
103
104
105
106
107
108
|
public static String getIp()
{
return getSubjct().getSession().getHost();
}
public static String getSessionId()
{
return String.valueOf(getSubjct().getSession().getId());
}
}
|