lywsvip/hsweb-framework
1
0
Fork 0
mirror of https://github.com/hs-web/hsweb-framework.git synced 2026-06-02 02:43:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化权限

Browse Source
This commit is contained in:
zhou-hao
2017-11-24 00:07:20 +08:00
parent 2d4064440f
commit b010cf7a24
14 changed files with 143 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
hsweb-authorization/hsweb-authorization-basic/src/test/java/org/hswebframework/web/authorization/RedisUserTokenManagerTests.java
View File

@@ -1,9 +1,6 @@
package org.hswebframework.web.authorization;
import org.hswebframework.web.authorization.token.DefaultUserTokenManager;
import org.hswebframework.web.authorization.token.SimpleUserToken;
import org.hswebframework.web.authorization.token.UserToken;
import org.hswebframework.web.authorization.token.UserTokenManager;
import org.hswebframework.web.authorization.token.*;
import org.hswebframework.web.id.IDGenerator;
import org.junit.Assert;
import org.redisson.Redisson;
@@ -15,12 +12,14 @@ import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ConcurrentMap;
public class RedisUserTokenManagerTests {
static UserTokenManager userTokenManager;
static DefaultUserTokenManager userTokenManager;
static String token = IDGenerator.MD5.generate();
@@ -31,26 +30,28 @@ public class RedisUserTokenManagerTests {
try {
ConcurrentMap<String, SimpleUserToken> repo = client.getMap("hsweb.user-token", new SerializationCodec());
ConcurrentMap<String, List<String>> userRepo = client.getMap("hsweb.user-token-u", new SerializationCodec());
ConcurrentMap<String, Set<String>> userRepo = client.getMap("hsweb.user-token-u", new SerializationCodec());
userTokenManager = new DefaultUserTokenManager(repo, userRepo) {
@Override
protected List<String> getUserToken(String userId) {
userRepo.computeIfAbsent(userId,u->new ArrayList<>());
protected Set<String> getUserToken(String userId) {
userRepo.computeIfAbsent(userId,u->new HashSet<>());
return client.getList("hsweb.user-token-"+userId, new SerializationCodec());
return client.getSet("hsweb.user-token-"+userId, new SerializationCodec());
}
};
userTokenManager.setAllopatricLoginMode(AllopatricLoginMode.deny);
// userTokenManager=new DefaultUserTokenManager();
userRepo.clear();
repo.clear();
for (int i = 0; i < 1000; i++) {
userTokenManager.signIn(IDGenerator.MD5.generate(), "sessionId", "admin", 60*3600*1000);
}
userTokenManager.signIn(IDGenerator.MD5.generate(), "sessionId", "admin2", 60*3600*1000);
// userRepo.clear();
// repo.clear();
// for (int i = 0; i < 1000; i++) {
// userTokenManager.signIn(IDGenerator.MD5.generate(), "sessionId", "admin", 60*3600*1000);
// }
// userTokenManager.signIn(IDGenerator.MD5.generate(), "sessionId", "admin2", 60*3600*1000);
testGet();
testGetAll();