package com.vincent.rsf.common.utils;
|
|
import lombok.extern.slf4j.Slf4j;
|
import redis.clients.jedis.Jedis;
|
import redis.clients.jedis.JedisPool;
|
import redis.clients.jedis.JedisPoolConfig;
|
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.Set;
|
|
/**
|
* redis tools
|
* Created by vincent on 2023-03-13
|
*/
|
@Slf4j
|
public class RedisSupport {
|
|
private static final Map<String, RedisSupport> REDIS_CACHE = new HashMap<>();
|
|
public static final String DEFAULT_FLAG = "redis";
|
|
private static final String LINK = ".";
|
|
public static RedisSupport defaultRedisSupport = RedisSupport.getRedis(DEFAULT_FLAG);
|
|
protected JedisPool pool;
|
|
protected JedisPoolConfig config;
|
|
Integer index = 0;
|
|
public Boolean initialize = false;
|
|
public synchronized static RedisSupport getRedis(String name){
|
RedisSupport redisSupport = REDIS_CACHE.get(name);
|
if (null == redisSupport){
|
redisSupport = new RedisSupport(name);
|
REDIS_CACHE.put(name, redisSupport);
|
}
|
return redisSupport;
|
}
|
|
private RedisSupport(String name) {
|
try {
|
boolean bAll = true;
|
String[] configKeys={"host","port","max","min","timeout","index"};
|
for(String key : configKeys){
|
if(!ConfigHelper.contains(name + "." + key)){
|
bAll = false;
|
}
|
}
|
if (bAll) {
|
config = new JedisPoolConfig();
|
config.setMaxIdle(ConfigHelper.getInteger(name+".min"));
|
config.setMaxWaitMillis(ConfigHelper.getLong(name+".timeout"));
|
// 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
|
config.setTestOnBorrow(false);
|
// 当前jedis连接的库号
|
index = ConfigHelper.getInteger(name+".index");
|
// 实例化jedis池
|
String host = ConfigHelper.getString(name+".host");
|
int port = ConfigHelper.getInteger(name+".port");
|
int timeout = ConfigHelper.getInteger(name+".timeout");
|
String password = ConfigHelper.getString(name+".password");
|
|
pool = new JedisPool(config,host , port,timeout,password);
|
Jedis jedis = this.getJedis();
|
this.pool.returnResource(jedis);
|
initialize = true;
|
} else {
|
log.error("ERROR - 初始化Reids时出错，在配置文件中未找到{}.***属性(host,port,max,min,timeout,index)", name);
|
}
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
public Jedis getJedis(){
|
try{
|
Jedis jedis = pool.getResource();
|
|
if(this.index != jedis.getDB().intValue()) {
|
jedis.select(this.index);
|
}
|
return jedis;
|
} catch (Exception e){
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
|
return null;
|
}
|
|
// key - value ----------------------------------------------------------------------------------------------------------
|
|
public String setValue(String flag, String key, String value) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.set(flag + LINK + key, value);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public String setValue(String flag, String key, String value,Integer seconds) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.setex(flag + LINK + key, seconds , value);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public String getValue(String flag, String key) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.get(flag + LINK + key);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public Long deleteValue(String flag, String... key) {
|
if(!this.initialize) {
|
return null;
|
}
|
|
Jedis jedis = this.getJedis();
|
try{
|
String[] keys = new String[key.length];
|
for(int i=0;i<key.length;i++){
|
keys[i] = flag + LINK + key[i];
|
}
|
return jedis.del(keys);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public Long clearValue(String flag) {
|
if(!this.initialize) {
|
return null;
|
}
|
|
this.setValue(flag, "CLEARING", "true");
|
Jedis jedis = this.getJedis();
|
|
try{
|
Object returnValue = jedis.eval("return redis.call('del', unpack(redis.call('keys', ARGV[1])))",0,flag + LINK + "*");
|
return Long.parseLong(String.valueOf(returnValue));
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public void setValueExpire(String flag, String key,int seconds){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expire((flag + LINK + key).getBytes(), seconds);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
public void setValueExpireTime(String flag, String key,Date atTime){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expireAt((flag + LINK + key).getBytes(), atTime.getTime()/1000);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
|
// hash ----------------------------------------------------------------------------------------------------------
|
|
public Long setMap(String name, String key, Object value) {
|
if(!this.initialize) {
|
return null;
|
}
|
if(value == null){
|
deleteMap(name,key);
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try {
|
return jedis.hset(name.getBytes(), key.getBytes(), Serialize.serialize(value));
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public <T> T getMap(String name, String key) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
byte[] bytes = jedis.hget(name.getBytes(), key.getBytes());
|
if (bytes == null || bytes.length == 0) {
|
return null;
|
}
|
return (T) Serialize.unSerialize(bytes);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public Set<String> getMapKeys(String name) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.hkeys(name);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public Long deleteMap(String name, String... key) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
String[] keys = new String[key.length];
|
System.arraycopy(key, 0, keys, 0, key.length);
|
return jedis.hdel(name, keys);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public Long clearMap(String name) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.del(name);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public void setMapExpire(String name,int seconds){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expire(name.getBytes(), seconds);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
public void setMapExpireTime(String name,Date atTime){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expireAt(name.getBytes(), atTime.getTime()/1000);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
|
// mq ----------------------------------------------------------------------------------------------------------
|
|
// 列表末尾添加元素
|
public Long push(String name, Object value) {
|
if(!this.initialize) {
|
return null;
|
}
|
if(value == null){
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.rpush(name.getBytes(), Serialize.serialize(value));
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
// 获取列表头部元素 && 删除
|
public <T> T pop(String name) {
|
if(!this.initialize){
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
byte[] bytes = jedis.lpop(name.getBytes());
|
if(bytes == null || bytes.length == 0) {
|
return null;
|
}
|
return (T) Serialize.unSerialize(bytes);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
// 删除
|
public Long deleteList(String name) {
|
if(!this.initialize) {
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.del(name);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
return null;
|
}
|
|
public void setListExpire(String name, int seconds){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expire(name.getBytes(), seconds);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
public void setListExpireTime(String name, Date atTime){
|
if(!this.initialize) {
|
return;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expireAt(name.getBytes(), atTime.getTime()/1000);
|
} catch (Exception e) {
|
log.error(this.getClass().getSimpleName(), e);
|
}
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public Object setObject(String flag,String key, Object value) {
|
if(!this.initialize)return null;
|
|
if(value==null){
|
this.deleteObject(flag, key);
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
String result = jedis.set(("OBJ."+flag+"."+key).getBytes(), Serialize.serialize(value));
|
return value;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return value;
|
}
|
public Object setObject(String flag,String key, Object value,Integer seconds) {
|
if(!this.initialize)return null;
|
|
if(value==null){
|
this.deleteObject(flag, key);
|
return null;
|
}
|
Jedis jedis = this.getJedis();
|
try{
|
String result = jedis.setex(("OBJ."+flag+"."+key).getBytes(),seconds,Serialize.serialize(value));
|
return value;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return value;
|
}
|
public <T> T getObject(String flag,String key) {
|
if(!this.initialize)return null;
|
Jedis jedis = this.getJedis();
|
try{
|
byte[] bytes = jedis.get(("OBJ."+flag+"."+key).getBytes());
|
if(bytes==null||bytes.length==0)
|
return null;
|
T returnValue = (T) Serialize.unSerialize(bytes);
|
return (T) returnValue;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
public <T> Long deleteObject(String flag, String key) {
|
if(!this.initialize)return null;
|
|
Jedis jedis = this.getJedis();
|
try{
|
Long returnValue = jedis.del(("OBJ."+flag+"."+key).getBytes());
|
return returnValue;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
|
public <T> Long clearObject(String flag) {
|
if(!this.initialize)return null;
|
|
this.setObject(flag, "CLEARING", "true");
|
Jedis jedis = this.getJedis();
|
try{
|
//EVAL "return redis.call('del', unpack(redis.call('keys', 'OBJ.*')))" 0
|
//EVAL "return redis.call('del', unpack(redis.call('keys', ARGV[1])))" 0 OBJ.*
|
//Object returnValue = jedis.eval("return redis.call('del', unpack(redis.call('keys', ARGV[1])))",0,"OBJ."+flag+"."+"*");
|
Object returnValue = jedis.eval("local keys = redis.call('keys', ARGV[1]) for i=1,#keys,1000 do redis.call('del', unpack(keys, i, math.min(i+4999, #keys))) end return #keys",0,"OBJ."+flag+"."+"*");
|
return Long.parseLong(String.valueOf(returnValue));
|
}catch(Exception ex){
|
////Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
|
public void setObjectExpire(String flag, String key,int seconds){
|
if(!this.initialize)return;
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expire(("OBJ."+flag+"."+key).getBytes(), seconds);
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
}
|
|
|
public void setObjectExpireTime(String flag, String key,Date toTime){
|
if(!this.initialize)return;
|
Jedis jedis = this.getJedis();
|
try{
|
jedis.expireAt(("OBJ."+flag+"."+key).getBytes(), toTime.getTime()/1000);
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
}
|
|
|
|
|
|
|
public String[] getKeys(String pattern) {
|
if(!this.initialize)return null;
|
|
Jedis jedis = this.getJedis();
|
try{
|
return jedis.keys(pattern).toArray(new String[]{});
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
|
|
public Long incr(String key) {
|
if(!this.initialize)return null;
|
Jedis jedis = this.getJedis();
|
try{
|
Long returnValue = jedis.incr("DI."+key);
|
return returnValue;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
public Long decr(String key) {
|
if(!this.initialize)return null;
|
Jedis jedis = this.getJedis();
|
try{
|
Long returnValue = jedis.decr("DI."+key);
|
return returnValue;
|
}catch(Exception ex){
|
//Utils.trace(ex.getMessage());
|
}finally{
|
if(jedis!=null)
|
this.pool.returnResource(jedis);
|
}
|
return null;
|
}
|
|
|
|
public static void main(String...strings){
|
Date start = new Date();
|
RedisSupport r = RedisSupport.getRedis("lazy.redis.0");
|
r.clearObject("aaa");
|
}
|
|
}
|
