ShiroUtils.java
3.72 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package com.huaheng.common.utils.security;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;
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.constant.QuantityConstant;
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() == null ? QuantityConstant.WAREHOUSECODE : 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());
}
/**
* 获取list重复数据
* @param stream
* @param <T>
* @return
*/
public static <T> List<T> getDuplicateElements(Stream<T> stream) {
return stream.collect(Collectors.toMap(e -> e, e -> 1, (a, b) -> a + b))
// 获得元素出现频率的 Map，键为元素，值为元素出现的次数
.entrySet().stream()
// Set<Entry>转换为Stream<Entry>
.filter(entry -> entry.getValue() > 1)
// 过滤出元素出现次数大于 1 的 entry
.map(entry -> entry.getKey())
// 获得 entry 的键（重复元素）对应的 Stream
.collect(Collectors.toList());
// 转化为 List
}
}