Apache-Shiro分布式环境配置(与redis集成)

前段时间项目要用到权限控制的相关模块,经过讨论决定采用Apache下面的Shiro开源框架进行身份校验与权限控制,因项目需部署在集群环境下,所以需要分布式的支持,故配置了Redis作为权限数据的存储,这里简单的记录下相关的配置

 

applicationContext-shiro.xml

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
<? xml  version = "1.0"  encoding = "UTF-8" ?>
< beans  xmlns = "http://www.springframework.org/schema/beans"  xmlns:xsi = "http://www.w3.org/2001/XMLSchema-instance" 
xmlns:p = "http://www.springframework.org/schema/p"  xmlns:c = "http://www.springframework.org/schema/c"
     xmlns:util = "http://www.springframework.org/schema/util"  xmlns:aop = "http://www.springframework.org/schema/aop" 
     xmlns:tx = "http://www.springframework.org/schema/tx"  xmlns:context = "http://www.springframework.org/schema/context"
     xsi:schemaLocation="
         http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.1.xsd
         http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util-3.1.xsd
         http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-3.1.xsd
         http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.1.xsd
         http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.1.xsd">
 
     < bean  id = "lifecycleBeanPostProcessor"  class = "org.apache.shiro.spring.LifecycleBeanPostProcessor"  />
 
 
     < bean  id = "securityManager"  class = "org.apache.shiro.web.mgt.DefaultWebSecurityManager" >
         < property  name = "authenticator"  ref = "authenticator"  />
         < property  name = "sessionManager"  ref = "sessionManager"  />
         <!-- <property name="sessionMode" value="http"/> -->
         < property  name = "cacheManager"  ref = "redisCacheManager"  />
         < property  name = "realms" >
             < list >
                 < ref  local = "customRealm"  />
             </ list >
         </ property >
     </ bean >
     < bean  class = "com.bbk.security.realm.CustomRealm"  id = "customRealm" >
         < property  name = "credentialsMatcher"  ref = "hashedCredentialsMatcher"  />
     </ bean >
     < bean  id = "redisCacheManager"  class = "com.bbk.security.cache.RedisCacheManager" > <!-- 自定义cacheManager -->
         < property  name = "redisManager"  ref = "redisManager"  />
     </ bean >
     < bean  id = "redisCache"  class = "com.bbk.security.cache.RedisCache" > <!-- 自定义cacheManager -->
         < constructor-arg  ref = "redisManager" ></ constructor-arg >
     </ bean >
     < bean  id = "roleOR"  class = "com.bbk.filter.CustomAuthorizationFilter"  />
     < bean  id = "shiroFilter"  class = "org.apache.shiro.spring.web.ShiroFilterFactoryBean" >
         <!-- 调用我们配置的权限管理器 -->
         < property  name = "securityManager"  ref = "securityManager"  />
     </ bean >
     < bean  id = "authenticator"  class = "org.apache.shiro.authc.pam.ModularRealmAuthenticator"  />
 
     < bean  id = "sessionManager"  class = "org.apache.shiro.web.session.mgt.DefaultWebSessionManager" >
         < property  name = "sessionDAO"  ref = "redisShiroSessionDAO"  />
         < property  name = "globalSessionTimeout"  value = "${shiro.session.timeout}"  /> <!-- 会话过期时间,在配置文件里面配置 -->
         < property  name = "sessionValidationInterval"  value = "3000000"  />
         < property  name = "sessionValidationSchedulerEnabled"  value = "true"  />
     </ bean >
 
     < bean  id = "redisShiroSessionDAO"  class = "com.bbk.security.cache.RedisSessionDAO" >
         < property  name = "redisManager"  ref = "redisManager"  />
     </ bean >
     < bean  id = "redisManager"  class = "com.bbk.security.cache.RedisManager" ></ bean >
     < bean  id = "hashedCredentialsMatcher"  class = "org.apache.shiro.authc.credential.HashedCredentialsMatcher" > <!-- 密钥的算法 -->
         < property  name = "hashAlgorithmName"  value = "MD5"  />
         < property  name = "storedCredentialsHexEncoded"  value = "true"  />
         < property  name = "hashIterations"  value = "1"  />
     </ bean >
 
     <!-- copy to 'spring-mvc.xml' -->
     <!-- <bean class="org.springframework.aop.framework.autoproxy.DefaultAdvisorAutoProxyCreator" depends-on="lifecycleBeanPostProcessor" />
      <bean class="org.apache.shiro.spring.security.interceptor.AuthorizationAttributeSourceAdvisor"> 
         <property name="securityManager" ref="securityManager" /> </bean> -->
 
</ beans >

RedisCache

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
package  com.bbk.security.cache;
 
import  java.util.ArrayList;
import  java.util.Collection;
import  java.util.Collections;
import  java.util.HashSet;
import  java.util.List;
import  java.util.Set;
 
import  org.apache.shiro.cache.Cache;
import  org.apache.shiro.cache.CacheException;
import  org.apache.shiro.subject.PrincipalCollection;
import  org.apache.shiro.util.CollectionUtils;
import  org.slf4j.Logger;
import  org.slf4j.LoggerFactory;
 
import  com.bbk.common.SerializeUtils;
 
public  class  RedisCache<K, V>  implements  Cache<K, V> {
 
     private  Logger logger = LoggerFactory.getLogger( this .getClass());
 
     /**
      * The wrapped Jedis instance.
      */
     private  RedisManager cache;
 
     /**
      * The Redis key prefix for the sessions 
      */
     private  String keyPrefix =  "shiro_session:" ;
 
     /**
      * Returns the Redis session keys
      * prefix.
      * @return The prefix
      */
     public  String getKeyPrefix() {
         return  keyPrefix;
     }
 
     /**
      * Sets the Redis sessions key 
      * prefix.
      * @param keyPrefix The prefix
      */
     public  void  setKeyPrefix(String keyPrefix) {
         this .keyPrefix = keyPrefix;
     }
 
     /**
      * 通过一个JedisManager实例构造RedisCache
      */
     public  RedisCache(RedisManager cache) {
         if  (cache ==  null ) {
             throw  new  IllegalArgumentException( "Cache argument cannot be null." );
         }
         this .cache = cache;
     }
 
     /**
      * Constructs a cache instance with the specified
      * Redis manager and using a custom key prefix.
      * @param cache The cache manager instance
      * @param prefix The Redis key prefix
      */
     public  RedisCache(RedisManager cache, String prefix) {
 
         this (cache);
 
         // set the prefix
         this .keyPrefix = prefix;
     }
 
     /**
      * 获得byte[]型的key
      * @param key
      * @return
      */
     private  byte [] getByteKey(K key) {
         if  (key  instanceof  String) {
             String preKey =  this .keyPrefix + key;
             return  preKey.getBytes();
         else  if (key  instanceof  PrincipalCollection){
             String preKey =  this .keyPrefix + key.toString();
             return  preKey.getBytes();
         } else {
             return  SerializeUtils.serialize(key);
         }
     }
 
     @Override
     public  V get(K key)  throws  CacheException {
         logger.debug( "根据key从Redis中获取对象 key ["  + key +  "]" );
         try  {
             if  (key ==  null ) {
                 return  null ;
             else  {
                 byte [] rawValue = cache.get(getByteKey(key));
                 @SuppressWarnings ( "unchecked" ) V value = (V) SerializeUtils.deserialize(rawValue);
                 return  value;
             }
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
 
     }
     public  String getStr(String key)  throws  CacheException {
         logger.debug( "根据key从Redis中获取对象 key ["  + key +  "]" );
         try  {
             if  (key ==  null ) {
                 return  null ;
             else  {
                 return  cache.get(key);
             }
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
         
     }
 
     @Override
     public  V put(K key, V value)  throws  CacheException {
         logger.debug( "根据key从存储 key ["  + key +  "]" );
         try  {
             cache.set(getByteKey(key), SerializeUtils.serialize(value));
             return  value;
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     public  String putStr(String key, String value)  throws  CacheException {
         logger.debug( "根据key从存储 key ["  + key +  "]" );
         try  {
             cache.set(key, value);
             return  value;
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     public  String put(String key,String value,  int  expire)  throws  CacheException {
         logger.debug( "根据key从存储 key ["  + key +  "]" );
         try  {
             cache.set(key, value, expire);
             return  value;
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     public  String removeString(String key)  throws  CacheException {
         logger.debug( "从redis中删除 key ["  + key +  "]" );
         try  {
             String previous = cache.get(key);
             cache.del(key);
             return  previous;
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     @Override
     public  V remove(K key)  throws  CacheException {
         logger.debug( "从redis中删除 key ["  + key +  "]" );
         try  {
             V previous = get(key);
             cache.del(getByteKey(key));
             return  previous;
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     @Override
     public  void  clear()  throws  CacheException {
         logger.debug( "从redis中删除所有元素" );
         try  {
             cache.flushDB();
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     @Override
     public  int  size() {
         try  {
             Long longSize =  new  Long(cache.dbSize());
             return  longSize.intValue();
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     @SuppressWarnings ( "unchecked" )
     @Override
     public  Set<K> keys() {
         try  {
             Set< byte []> keys = cache.keys( this .keyPrefix +  "*" );
             if  (CollectionUtils.isEmpty(keys)) {
                 return  Collections.emptySet();
             else  {
                 Set<K> newKeys =  new  HashSet<K>();
                 for  ( byte [] key : keys) {
                     newKeys.add((K) key);
                 }
                 return  newKeys;
             }
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
     @Override
     public  Collection<V> values() {
         try  {
             Set< byte []> keys = cache.keys( this .keyPrefix +  "*" );
             if  (!CollectionUtils.isEmpty(keys)) {
                 List<V> values =  new  ArrayList<V>(keys.size());
                 for  ( byte [] key : keys) {
                     @SuppressWarnings ( "unchecked" ) V value = get((K) key);
                     if  (value !=  null ) {
                         values.add(value);
                     }
                 }
                 return  Collections.unmodifiableList(values);
             else  {
                 return  Collections.emptyList();
             }
         catch  (Throwable t) {
             throw  new  CacheException(t);
         }
     }
 
}

 

 Realm

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
package  com.bbk.security.realm;
 
import  java.util.Collection;
import  java.util.List;
import  java.util.Set;
 
import  org.apache.log4j.Logger;
import  org.apache.shiro.SecurityUtils;
import  org.apache.shiro.authc.AuthenticationException;
import  org.apache.shiro.authc.AuthenticationInfo;
import  org.apache.shiro.authc.AuthenticationToken;
import  org.apache.shiro.authc.SimpleAuthenticationInfo;
import  org.apache.shiro.authc.UsernamePasswordToken;
import  org.apache.shiro.authz.AuthorizationInfo;
import  org.apache.shiro.authz.SimpleAuthorizationInfo;
import  org.apache.shiro.cache.Cache;
import  org.apache.shiro.cache.CacheManager;
import  org.apache.shiro.cache.CacheManagerAware;
import  org.apache.shiro.realm.AuthorizingRealm;
import  org.apache.shiro.realm.Realm;
import  org.apache.shiro.subject.PrincipalCollection;
import  org.apache.shiro.subject.SimplePrincipalCollection;
import  org.apache.shiro.subject.Subject;
import  org.springframework.beans.factory.InitializingBean;
import  org.springframework.beans.factory.annotation.Autowired;
import  org.springframework.cache.annotation.CacheEvict;
 
import  com.bbk.bean.Module;
import  com.bbk.bean.Role;
import  com.bbk.bean.User;
import  com.bbk.bean.enums.EnumStatus;
import  com.bbk.common.Constants;
import  com.bbk.common.ObjectUtils;
import  com.bbk.common.exception.UserInputException;
import  com.bbk.security.cache.RedisCache;
import  com.bbk.service.IBaseService;
import  com.bbk.service.IModuleService;
import  com.bbk.service.IRoleService;
import  com.bbk.service.IUserService;
 
import  fmath.a.a.c;
 
public  class  CustomRealm  extends  AuthorizingRealm  implements  Realm, InitializingBean {
 
     @Autowired
     private  IRoleService roleService;
     @Autowired
     private  IModuleService moduleService;
     @Autowired
     private  IUserService userService;
 
     @Autowired
     private  RedisCache<String, Object> redisCache;
 
     @Autowired
     private  IBaseService baseService;
 
     private  Logger logger = Logger.getLogger( this .getClass());
 
     public  CustomRealm() {
     }
 
     @Override
     protected  AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
         //logger.info("授权查询回调函数, 进行鉴权但缓存中无用户的授权信息时调用");
         SimpleAuthorizationInfo info =  new  SimpleAuthorizationInfo();
         String username = (String)  super .getAvailablePrincipal(principals);
         User user = userService.findUserByUsername(username);
         List<Role> roles = roleService.queryRolesByUserId(user);
 
         StringBuilder sb =  new  StringBuilder();
         for  (Role role : roles) {
             info.addRole(role.getCode());
             sb.append(role.getId());
             sb.append( "," );
         }
         if  (sb.length() >  0 ) {
             List<Module> perms = moduleService.findPermNamesByUser(sb.substring( 0 , sb.length() -  1 ));
             for  (Module module : perms) {
                 if  (ObjectUtils.isNotEmpty(module.getCode())) {
                     info.addStringPermission(module.getCode());
                 }
             }
             redisCache.put(Constants.getUserModuleCacheKey(user.getUserName()), perms);
         }
         //cacheManager.getCache("").clear();
         redisCache.put(Constants.getUserRolesCacheKey(user.getUserName()), roles);
         return  info;
     }
 
     @Override
     protected  AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token)  throws  AuthenticationException {
 
         SimpleAuthenticationInfo info =  null ;
 
         UsernamePasswordToken upt = (UsernamePasswordToken) token;
         User user = userService.findUserByUsername(upt.getUsername());
         if  (user ==  null ) {
             throw  new  AuthenticationException( "用户名不存在!" );
         }
         if  (user.getStatus() < EnumStatus.NORMAL.getStatus()) {
             throw  new  UserInputException( "您的帐号号已被限制登录,如需帮助请联系管理员!" );
         }
         redisCache.put(Constants.getUserSessionKey(user.getUserName()), user);
         info =  new  SimpleAuthenticationInfo(user.getUserName(), user.getUserPsw(), getName());
         return  info;
     }
 
     /**
      * 更新用户授权信息缓存.
      */
     public  void  clearCachedAuthorizationInfo(String principal) {
         SimplePrincipalCollection principals =  new  SimplePrincipalCollection(principal, getName());
         super .clearCachedAuthorizationInfo(principals);
         super .clearCache(principals);
         super .clearCachedAuthenticationInfo(principals);
         redisCache.remove(Constants.getUserModuleCacheKey(principal));
         redisCache.remove(Constants.getUserRolesCacheKey(principal));
     }
 
     /**
      * 清除所有用户授权信息缓存.
      */
     public  void  clearAllCachedAuthorizationInfo() {
         redisCache.clear();
         /*Cache<Object, AuthenticationInfo> cache = getAuthenticationCache();
         if (null != cache) {
             for (Object key : cache.keys()) {
                 cache.remove(key);
             }
         }*/
     }
 
     @Override
     public  void  afterPropertiesSet()  throws  Exception {
     }
 
}

 

RedisCacheManager

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
package  com.bbk.security.cache;
 
import  java.util.concurrent.ConcurrentHashMap;
import  java.util.concurrent.ConcurrentMap;
 
import  org.apache.shiro.cache.Cache;
import  org.apache.shiro.cache.CacheException;
import  org.apache.shiro.cache.CacheManager;
import  org.slf4j.Logger;
import  org.slf4j.LoggerFactory;
 
 
public  class  RedisCacheManager  implements  CacheManager {
 
     private  static  final  Logger logger = LoggerFactory.getLogger(RedisCacheManager. class );
 
     // fast lookup by name map
     private  final  ConcurrentMap<String, Cache> caches =  new  ConcurrentHashMap<String, Cache>();
 
     private  RedisManager redisManager;
 
     /**
      * The Redis key prefix for caches 
      */
     private  String keyPrefix =  "shiro_redis_cache:" ;
 
     /**
      * Returns the Redis session keys
      * prefix.
      * @return The prefix
      */
     public  String getKeyPrefix() {
         return  keyPrefix;
     }
 
     /**
      * Sets the Redis sessions key 
      * prefix.
      * @param keyPrefix The prefix
      */
     public  void  setKeyPrefix(String keyPrefix) {
         this .keyPrefix = keyPrefix;
     }
 
     @Override
     public  <K, V> Cache<K, V> getCache(String name)  throws  CacheException {
         logger.debug( "获取名称为: "  + name +  " 的RedisCache实例" );
 
         Cache c = caches.get(name);
 
         if  (c ==  null ) {
 
             // initialize the Redis manager instance
             redisManager.init();
 
             // create a new cache instance
             c =  new  RedisCache<K, V>(redisManager, keyPrefix);
 
             // add it to the cache collection
             caches.put(name, c);
         }
         return  c;
     }
 
     public  RedisManager getRedisManager() {
         return  redisManager;
     }
 
     public  void  setRedisManager(RedisManager redisManager) {
         this .redisManager = redisManager;
     }
 
}

 

 

 

 

 

RedisSessionDAO

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package  com.bbk.security.cache;
 
import  java.io.Serializable;
import  java.util.Collection;
import  java.util.HashSet;
import  java.util.Set;
 
import  org.apache.shiro.session.Session;
import  org.apache.shiro.session.UnknownSessionException;
import  org.apache.shiro.session.mgt.eis.AbstractSessionDAO;
import  org.slf4j.Logger;
import  org.slf4j.LoggerFactory;
 
import  com.bbk.common.SerializeUtils;
 
/**
  * @author Jon Chiang
  * @project video
  * @create_date 2014-5-6 下午5:35:07
  */
public  class  RedisSessionDAO  extends  AbstractSessionDAO {
 
     private  static  Logger logger = LoggerFactory.getLogger(RedisSessionDAO. class );
     /**
      * shiro-redis的session对象前缀
      */
     private  RedisManager redisManager;
     
     /**
      * The Redis key prefix for the sessions 
      */
     private  String keyPrefix =  "shiro_session:" ;
     
     @Override
     public  void  update(Session session)  throws  UnknownSessionException {
         this .saveSession(session);
     }
     
     /**
      * save session
      * @param session
      * @throws UnknownSessionException
      */
     private  void  saveSession(Session session)  throws  UnknownSessionException{
         if (session ==  null  || session.getId() ==  null ){
             logger.error( "session or session id is null" );
             return ;
         }
         
         byte [] key = getByteKey(session.getId());
         byte [] value = SerializeUtils.serialize(session);
         session.setTimeout(redisManager.getExpire()* 1000 );     
         this .redisManager.set(key, value, redisManager.getExpire());
     }
 
     @Override
     public  void  delete(Session session) {
         if (session ==  null  || session.getId() ==  null ){
             logger.error( "session or session id is null" );
             return ;
         }
         redisManager.del( this .getByteKey(session.getId()));
 
     }
 
     @Override
     public  Collection<Session> getActiveSessions() {
         Set<Session> sessions =  new  HashSet<Session>();
         
         Set< byte []> keys = redisManager.keys( this .keyPrefix +  "*" );
         if (keys !=  null  && keys.size()> 0 ){
             for ( byte [] key:keys){
                 Session s = (Session)SerializeUtils.deserialize(redisManager.get(key));
                 sessions.add(s);
             }
         }
         
         return  sessions;
     }
 
     @Override
     protected  Serializable doCreate(Session session) {
         Serializable sessionId =  this .generateSessionId(session);  
         this .assignSessionId(session, sessionId);
         this .saveSession(session);
         return  sessionId;
     }
 
     @Override
     protected  Session doReadSession(Serializable sessionId) {
         if (sessionId ==  null ){
             logger.error( "session id is null" );
             return  null ;
         }
         
         Session s = (Session)SerializeUtils.deserialize(redisManager.get( this .getByteKey(sessionId)));
         return  s;
     }
     
     /**
      * 获得byte[]型的key
      * @param key
      * @return
      */
     private  byte [] getByteKey(Serializable sessionId){
         String preKey =  this .keyPrefix + sessionId;
         return  preKey.getBytes();
     }
 
     public  RedisManager getRedisManager() {
         return  redisManager;
     }
 
     public  void  setRedisManager(RedisManager redisManager) {
         this .redisManager = redisManager;
         
         /**
          * 初始化redisManager
          */
         this .redisManager.init();
     }
 
     /**
      * Returns the Redis session keys
      * prefix.
      * @return The prefix
      */
     public  String getKeyPrefix() {
         return  keyPrefix;
     }
 
     /**
      * Sets the Redis sessions key 
      * prefix.
      * @param keyPrefix The prefix
      */
     public  void  setKeyPrefix(String keyPrefix) {
         this .keyPrefix = keyPrefix;
     }
     
     
}

 

RedisManager

 

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
package  com.bbk.security.cache;
 
import  java.util.Set;
 
import  org.slf4j.Logger;
import  org.slf4j.LoggerFactory;
import  org.springframework.beans.factory.annotation.Value;
 
import  com.bbk.common.JedisUtil;
 
import  redis.clients.jedis.Jedis;
import  redis.clients.jedis.JedisPool;
import  redis.clients.jedis.JedisPoolConfig;
 
public  class  RedisManager {
     
     private  static  Logger logger = LoggerFactory.getLogger(RedisSessionDAO. class );
     
     @Value ( "${redis.ip}" )
     private  String host;
 
     @Value ( "${redis.port}" )
     private  int  port;
 
     // 0 - never expire
     private  int  expire =  0 ;
 
     private  static  JedisPool jedisPool =  null ;
 
     public  RedisManager() {
     }
 
     /**
      * 初始化方法
      */
     public  void  init(){
         if ( null  == host ||  0  == port){
             logger.error( "请初始化redis配置文件!" );
             throw  new  NullPointerException( "找不到redis配置" );
         }
         if (jedisPool ==  null ){
             //jedisPool = JedisUtil.getJedisPool();
             jedisPool =  new  JedisPool( new  JedisPoolConfig(), host, port);
         }
     }
 
     /**
      * get value from redis
      * @param key
      * @return
      */
     public  byte [] get( byte [] key) {
         byte [] value =  null ;
         Jedis jedis = jedisPool.getResource();
         try  {
             value = jedis.get(key);
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
     /**
      * get value from redis
      * @param key
      * @return
      */
     public  String get(String key) {
         String value= null ;
         Jedis jedis = jedisPool.getResource();
         try  {
             value = jedis.get(key);
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
 
     /**
      * set 
      * @param key
      * @param value
      * @return
      */
     public  byte [] set( byte [] key,  byte [] value) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.set(key, value);
             if  ( this .expire !=  0 ) {
                 jedis.expire(key,  this .expire);
             }
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
     /**
      * set 
      * @param key
      * @param value
      * @return
      */
     public  String set(String key,String value) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.set(key, value);
             if  ( this .expire !=  0 ) {
                 jedis.expire(key,  this .expire);
             }
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
 
     /**
      * set 
      * @param key
      * @param value
      * @param expire
      * @return
      */
     public  byte [] set( byte [] key,  byte [] value,  int  expire) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.set(key, value);
             if  (expire !=  0 ) {
                 jedis.expire(key, expire);
             }
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
     /**
      * set 
      * @param key
      * @param value
      * @param expire
      * @return
      */
     public  String set(String key,String value,  int  expire) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.set(key, value);
             if  (expire !=  0 ) {
                 jedis.expire(key, expire);
             }
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  value;
     }
 
     /**
      * del
      * @param key
      */
     public  void  del( byte [] key) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.del(key);
         finally  {
             jedisPool.returnResource(jedis);
         }
     }
     /**
      * del
      * @param key
      */
     public  void  del(String key) {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.del(key);
         finally  {
             jedisPool.returnResource(jedis);
         }
     }
 
     /**
      * flush
      */
     public  void  flushDB() {
         Jedis jedis = jedisPool.getResource();
         try  {
             jedis.flushDB();
         finally  {
             jedisPool.returnResource(jedis);
         }
     }
 
     /**
      * size
      */
     public  Long dbSize() {
         Long dbSize = 0L;
         Jedis jedis = jedisPool.getResource();
         try  {
             dbSize = jedis.dbSize();
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  dbSize;
     }
 
     /**
      * keys
      * @param regex
      * @return
      */
     public  Set< byte []> keys(String pattern) {
         Set< byte []> keys =  null ;
         Jedis jedis = jedisPool.getResource();
         try  {
             keys = jedis.keys(pattern.getBytes());
         finally  {
             jedisPool.returnResource(jedis);
         }
         return  keys;
     }
 
     public  String getHost() {
         return  host;
     }
 
     public  void  setHost(String host) {
         this .host = host;
     }
 
     public  int  getPort() {
         return  port;
     }
 
     public  void  setPort( int  port) {
         this .port = port;
     }
 
     public  int  getExpire() {
         return  expire;
     }
 
     public  void  setExpire( int  expire) {
         this .expire = expire;
     }
 
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值