ShiroUtils.java
2.85 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package com.huaheng.common.utils.security;
import java.util.List;
import javax.annotation.Resource;
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.common.utils.StringUtils;
import com.huaheng.common.utils.bean.BeanUtils;
import com.huaheng.framework.shiro.realm.UserRealm;
import com.huaheng.pc.config.warehouse.service.WarehouseService;
import com.huaheng.pc.system.user.domain.User;
/**
* shiro 工具类
*
* @author huaheng
*/
public class ShiroUtils {
@Resource
private WarehouseService warehouseService;
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() {
try {
Object obj = getSubjct().getPrincipal();
if (StringUtils.isNotNull(obj)) {
User user = new User();
BeanUtils.copyBeanProp(user, obj);
return user;
} else {
return null;
}
} catch (Exception e) {
return null;
}
}
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 Integer getUserId() {
return getUser().getId();
}
public static String getLoginName() {
if (getUser() == null) {
return null;
}
return getUser().getLoginName();
}
public static String getWarehouseCode() {
return getUser().getWarehouseCode();
}
public static List<Integer> getCompanyIdList() {
return getUser().getCompanyIdList();
}
public static List<String> getCompanyCodeList() {
return getUser().getCompanyCodeList();
}
public static String getIp() {
return getSubjct().getSession().getHost();
}
public static String getSessionId() {
return String.valueOf(getSubjct().getSession().getId());
}
}