package com.gkhy.exam.common.utils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.data.redis.connection.RedisStringCommands;
|
import org.springframework.data.redis.connection.ReturnType;
|
import org.springframework.data.redis.core.Cursor;
|
import org.springframework.data.redis.core.RedisCallback;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.data.redis.core.ScanOptions;
|
import org.springframework.data.redis.core.types.Expiration;
|
import org.springframework.data.redis.serializer.RedisSerializer;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.nio.charset.Charset;
|
import java.nio.charset.StandardCharsets;
|
import java.util.*;
|
import java.util.concurrent.TimeUnit;
|
|
@Component
|
public class RedisUtils<T> {
|
/**
|
* 注入redisTemplate bean
|
*/
|
@Autowired
|
private RedisTemplate redisTemplate;
|
|
@Value("${spring.application.name}")
|
public String prefixRedisName;
|
|
@Value("${spring.profiles.active}")
|
public String activeName;
|
|
private static final Long RELEASE_SUCCESS = 1L;
|
private static final String LOCK_SUCCESS = "OK";
|
private static final String SET_IF_NOT_EXIST = "NX";
|
// 当前设置 过期时间单位, EX = seconds; PX = milliseconds
|
private static final String SET_WITH_EXPIRE_TIME = "EX";
|
// if get(key) == value return del(key)
|
private static final String RELEASE_LOCK_SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
|
|
/**
|
* 删除指定前缀的key
|
* @param pattern
|
*/
|
public void delByScan(String pattern) {
|
RedisCallback<Set<String>> redisCallback = connection -> {
|
Set<String> keySet = new HashSet<>();
|
ScanOptions.ScanOptionsBuilder scanOptionsBuilder = ScanOptions.scanOptions();
|
scanOptionsBuilder.match(generateKey(pattern) + "*");
|
//scanOptionsBuilder.count(100);
|
ScanOptions scanOptions = scanOptionsBuilder.build();
|
Cursor<byte[]> cursor = connection.scan(scanOptions);
|
int count = 0;
|
while (cursor.hasNext()) {
|
keySet.add(RedisSerializer.string().deserialize(cursor.next()));
|
count++;
|
}
|
return keySet;
|
};
|
Set<String> keySet = (Set<String>) redisTemplate.execute(redisCallback);
|
Iterator<String> it = keySet.iterator();
|
while (it.hasNext()) {
|
redisTemplate.delete(it.next());
|
}
|
}
|
|
/**
|
* 指定缓存失效时间
|
*
|
* @param key 键
|
* @param time 时间(秒)
|
* @return
|
*/
|
public boolean expire(String key, long time) {
|
try {
|
if (time > 0) {
|
redisTemplate.expire(key, time, TimeUnit.SECONDS);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
/**
|
* 指定缓存失效时间
|
*
|
* @param key 键
|
* @param time 时间(秒)
|
* @return
|
*/
|
public boolean expire(String key, long time,TimeUnit timeUnit) {
|
try {
|
if (time > 0) {
|
redisTemplate.expire(key, time, timeUnit);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 根据key获取过期时间
|
*
|
* @param key 键 不能为null
|
* @return 时间(秒) 返回0代表为永久有效
|
*/
|
public long getExpire(String key) {
|
return redisTemplate.getExpire(key, TimeUnit.SECONDS);
|
}
|
|
/**
|
* 判断key是否存在
|
*
|
* @param key 键
|
* @return true 存在 false不存在
|
*/
|
public boolean hasKey(String key) {
|
try {
|
return redisTemplate.hasKey(key);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 删除缓存
|
*
|
* @param key 可以传一个值 或多个
|
*/
|
@SuppressWarnings("unchecked")
|
public void del(String... key) {
|
if (key != null && key.length > 0) {
|
if (key.length == 1) {
|
redisTemplate.delete(key[0]);
|
} else {
|
redisTemplate.delete((Collection<String>) CollectionUtils.arrayToList(key));
|
}
|
}
|
}
|
|
/**
|
* 删除集合对象
|
*
|
* @param collection 多个对象
|
* @return
|
*/
|
public boolean del(final Collection collection)
|
{
|
return redisTemplate.delete(collection) > 0;
|
}
|
|
|
|
// ============================String(字符串)=============================
|
|
/**
|
* 普通缓存获取
|
*
|
* @param key 键
|
* @return 值
|
*/
|
public Object get(String key) {
|
return redisTemplate.opsForValue().get(key);
|
}
|
|
/**
|
* 普通缓存放入
|
*
|
* @param key 键
|
* @param value 值
|
* @return true成功 false失败
|
*/
|
public boolean set(String key, Object value) {
|
try {
|
redisTemplate.opsForValue().set(key, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 普通缓存放入并设置时间
|
*
|
* @param key 键
|
* @param value 值
|
* @param time 时间(秒) time要大于0 如果time小于等于0 将设置无限期
|
* @return true成功 false 失败
|
*/
|
public boolean set(String key, Object value, long time) {
|
try {
|
if (time > 0) {
|
redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
|
} else {
|
set(key, value);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 普通缓存放入并设置时间,指定时间单位
|
*
|
* @param key 键
|
* @param value 值
|
* @param time 时间(秒) time要大于0 如果time小于等于0 将设置无限期
|
* @return true成功 false 失败
|
*/
|
public boolean set(String key, Object value, long time, TimeUnit timeUnit) {
|
try {
|
if (time > 0) {
|
redisTemplate.opsForValue().set(key, value, time, timeUnit);
|
} else {
|
set(key, value);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 普通缓存放入并设置时间
|
*
|
* @param key 键
|
* @param value 值
|
* @param time 设置失效时间 s
|
* @return true成功 false 失败
|
*/
|
public boolean setNotChangeExpire(String key, Object value,long time) {
|
try {
|
Long expire=redisTemplate.getExpire(key,TimeUnit.SECONDS);
|
if(expire==null ||expire<=0){
|
expire=time;
|
}
|
redisTemplate.opsForValue().set(key, value, expire, TimeUnit.SECONDS);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 递增
|
*
|
* @param key 键
|
* @param delta 要增加几(大于0)
|
* @return
|
*/
|
public long incr(String key, long delta) {
|
if (delta < 0) {
|
throw new RuntimeException("递增因子必须大于0");
|
}
|
return redisTemplate.opsForValue().increment(key, delta);
|
}
|
|
/**
|
* 递减
|
*
|
* @param key 键
|
* @param delta 要减少几(小于0)
|
* @return
|
*/
|
public long decr(String key, long delta) {
|
if (delta < 0) {
|
throw new RuntimeException("递减因子必须大于0");
|
}
|
return redisTemplate.opsForValue().increment(key, -delta);
|
}
|
// ================================Hash(哈希)=================================
|
|
/**
|
* HashGet
|
*
|
* @param key 键 不能为null
|
* @param item 项 不能为null
|
* @return 值
|
*/
|
public Object hget(String key, String item) {
|
return redisTemplate.opsForHash().get(key, item);
|
}
|
|
/**
|
* 获取hashKey对应的所有键值
|
*
|
* @param key 键
|
* @return 对应的多个键值
|
*/
|
public Map<Object, Object> hmget(String key) {
|
return redisTemplate.opsForHash().entries(key);
|
}
|
|
/**
|
* HashSet
|
*
|
* @param key 键
|
* @param map 对应多个键值
|
* @return true 成功 false 失败
|
*/
|
public boolean hmset(String key, Map<Object, Object> map) {
|
try {
|
redisTemplate.opsForHash().putAll(key, map);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* HashSet 并设置时间
|
*
|
* @param key 键
|
* @param map 对应多个键值
|
* @param time 时间(秒)
|
* @return true成功 false失败
|
*/
|
public boolean hmset(String key, Map<Object, Object> map, long time) {
|
try {
|
redisTemplate.opsForHash().putAll(key, map);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 向一张hash表中放入数据,如果不存在将创建
|
*
|
* @param key 键
|
* @param item 项
|
* @param value 值
|
* @return true 成功 false失败
|
*/
|
public boolean hset(String key, String item, Object value) {
|
try {
|
redisTemplate.opsForHash().put(key, item, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 向一张hash表中放入数据,如果不存在将创建
|
*
|
* @param key 键
|
* @param item 项
|
* @param value 值
|
* @param time 时间(秒) 注意:如果已存在的hash表有时间,这里将会替换原有的时间
|
* @return true 成功 false失败
|
*/
|
public boolean hset(String key, String item, Object value, long time) {
|
try {
|
redisTemplate.opsForHash().put(key, item, value);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 删除hash表中的值
|
*
|
* @param key 键 不能为null
|
* @param item 项 可以使多个 不能为null
|
*/
|
public void hdel(String key, Object... item) {
|
redisTemplate.opsForHash().delete(key, item);
|
}
|
|
/**
|
* 判断hash表中是否有该项的值
|
*
|
* @param key 键 不能为null
|
* @param item 项 不能为null
|
* @return true 存在 false不存在
|
*/
|
public boolean hHasKey(String key, String item) {
|
return redisTemplate.opsForHash().hasKey(key, item);
|
}
|
|
/**
|
* hash递增 如果不存在,就会创建一个 并把新增后的值返回
|
*
|
* @param key 键
|
* @param item 项
|
* @param by 要增加几(大于0)
|
* @return
|
*/
|
public double hincr(String key, String item, double by) {
|
return redisTemplate.opsForHash().increment(key, item, by);
|
}
|
|
/**
|
* hash递减
|
*
|
* @param key 键
|
* @param item 项
|
* @param by 要减少记(小于0)
|
* @return
|
*/
|
public double hdecr(String key, String item, double by) {
|
return redisTemplate.opsForHash().increment(key, item, -by);
|
}
|
// ============================Set(集合)=============================
|
|
/**
|
* 根据key获取Set中的所有值
|
*
|
* @param key 键
|
* @return
|
*/
|
public Set<Object> sGet(String key) {
|
try {
|
return redisTemplate.opsForSet().members(key);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
/**
|
* 根据value从一个set中查询,是否存在
|
*
|
* @param key 键
|
* @param value 值
|
* @return true 存在 false不存在
|
*/
|
public boolean sHasKey(String key, Object value) {
|
try {
|
return redisTemplate.opsForSet().isMember(key, value);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 将数据放入set缓存
|
*
|
* @param key 键
|
* @param values 值 可以是多个
|
* @return 成功个数
|
*/
|
public long sSet(String key, Object... values) {
|
try {
|
return redisTemplate.opsForSet().add(key, values);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
|
/**
|
* 将set数据放入缓存
|
*
|
* @param key 键
|
* @param time 时间(秒)
|
* @param values 值 可以是多个
|
* @return 成功个数
|
*/
|
public long sSetAndTime(String key, long time, Object... values) {
|
try {
|
Long count = redisTemplate.opsForSet().add(key, values);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return count;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
|
/**
|
* 获取set缓存的长度
|
*
|
* @param key 键
|
* @return
|
*/
|
public long sGetSetSize(String key) {
|
try {
|
return redisTemplate.opsForSet().size(key);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
|
/**
|
* 移除值为value的
|
*
|
* @param key 键
|
* @param values 值 可以是多个
|
* @return 移除的个数
|
*/
|
public long setRemove(String key, Object... values) {
|
try {
|
Long count = redisTemplate.opsForSet().remove(key, values);
|
return count;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
// ===============================List(列表)=================================
|
|
/**
|
* 获取list缓存的内容
|
*
|
* @param key 键
|
* @param start 开始
|
* @param end 结束 0 到 -1代表所有值
|
* @return
|
*/
|
public List<Object> lGet(String key, long start, long end) {
|
try {
|
return redisTemplate.opsForList().range(key, start, end);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
/**
|
* 获取list缓存的长度
|
*
|
* @param key 键
|
* @return
|
*/
|
public long lGetListSize(String key) {
|
try {
|
return redisTemplate.opsForList().size(key);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
|
/**
|
* 通过索引 获取list中的值
|
*
|
* @param key 键
|
* @param index 索引 index>=0时, 0 表头,1 第二个元素,依次类推;index<0时,-1,表尾,-2倒数第二个元素,依次类推
|
* @return
|
*/
|
public Object lGetIndex(String key, long index) {
|
try {
|
return redisTemplate.opsForList().index(key, index);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
/**
|
* 将list放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @return
|
*/
|
public boolean lSet(String key, Object value) {
|
try {
|
redisTemplate.opsForList().rightPush(key, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 将list放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @param time 时间(秒)
|
* @return
|
*/
|
public boolean lSet(String key, Object value, long time) {
|
try {
|
redisTemplate.opsForList().rightPush(key, value);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 将list放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @return
|
*/
|
public boolean lSet(String key, List<Object> value) {
|
try {
|
redisTemplate.opsForList().rightPushAll(key, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
|
/**
|
* 将list从左边放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @return
|
*/
|
public boolean lLSet(String key, Object value, long time) {
|
try {
|
redisTemplate.opsForList().leftPush(key, value);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 将list放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @return
|
*/
|
public boolean lLSet(String key, Object value) {
|
try {
|
redisTemplate.opsForList().leftPush(key, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
|
/**
|
* 将list放入缓存
|
*
|
* @param key 键
|
* @param value 值
|
* @param time 时间(秒)
|
* @return
|
*/
|
public boolean lSet(String key, List<Object> value, long time) {
|
try {
|
redisTemplate.opsForList().rightPushAll(key, value);
|
if (time > 0) {
|
expire(key, time);
|
}
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 根据索引修改list中的某条数据
|
*
|
* @param key 键
|
* @param index 索引
|
* @param value 值
|
* @return
|
*/
|
public boolean lUpdateIndex(String key, long index, Object value) {
|
try {
|
redisTemplate.opsForList().set(key, index, value);
|
return true;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* 移除N个值为value
|
*
|
* @param key 键
|
* @param count 移除多少个
|
* @param value 值
|
* @return 移除的个数
|
*/
|
public long lRemove(String key, long count, Object value) {
|
try {
|
Long remove = redisTemplate.opsForList().remove(key, count, value);
|
return remove;
|
} catch (Exception e) {
|
e.printStackTrace();
|
return 0;
|
}
|
}
|
|
|
/**
|
* 生成统一的前缀key
|
* @param subKey
|
* @return
|
*/
|
public String generateKey(String subKey){
|
if ("pro".equals(activeName)) {
|
return generateNoPrefixKey(subKey);
|
}
|
return activeName + ":" + prefixRedisName+":"+subKey;
|
}
|
|
/**
|
* 生成小程序token的前缀key
|
* @param subKey
|
* @return
|
*/
|
public String generateNoPrefixKey(String subKey){
|
return prefixRedisName+":"+subKey;
|
}
|
|
|
/**
|
* 加锁直接使用set命令同时设置唯一id和过期时间
|
* @param lockKey
|
* @param clientId
|
* @param seconds
|
* @return
|
*/
|
// public Boolean tryLock(String lockKey,String clientId,int seconds){
|
// try{
|
// RedisCallback<Boolean> callback =(redisConnection) -> {
|
// RedisSerializer valueSerializer=redisTemplate.getValueSerializer();
|
// RedisSerializer keySerializer=redisTemplate.getKeySerializer();
|
// Object result= redisConnection.set(keySerializer.serialize(lockKey),
|
// keySerializer.serialize(lockKey),
|
// Expiration.seconds(seconds),
|
// RedisStringCommands.SetOption.SET_IF_ABSENT
|
// );
|
// System.out.println(result+"-"+lockKey+"*********************");
|
// return (Boolean) result;
|
// };
|
// Boolean result = (Boolean) redisTemplate.execute(callback);
|
// return result;
|
// }catch (Exception e) {
|
// logger.error("set redis occured an exception", e);
|
// }
|
// return false;
|
// }
|
public Boolean tryLock(String lockKey,String clientId,int seconds) {
|
RedisCallback<Boolean> callback = connection -> connection.set(lockKey.getBytes(StandardCharsets.UTF_8),
|
clientId.getBytes(StandardCharsets.UTF_8), Expiration.seconds(seconds),
|
RedisStringCommands.SetOption.SET_IF_ABSENT);
|
Object obj=redisTemplate.execute(callback);
|
return (Boolean) obj;
|
}
|
|
/**
|
*释放锁
|
* 利用redis的lua脚本保证两个命令的原子性执行
|
* @param lockKey
|
* @param clientId
|
* @return
|
*/
|
public Boolean releaseLock(String lockKey,String clientId){
|
RedisCallback<Long> callback =(redisConnection) -> {
|
byte[][] keysAndArgs = new byte[2][];
|
keysAndArgs[0] = lockKey.getBytes(Charset.forName("UTF-8"));
|
keysAndArgs[1] = clientId.getBytes(Charset.forName("UTF-8"));
|
Long result = (Long)redisConnection.scriptingCommands().eval(RELEASE_LOCK_SCRIPT.getBytes(StandardCharsets.UTF_8), ReturnType.INTEGER,1,keysAndArgs);
|
return result;
|
};
|
Long result= (Long) redisTemplate.execute(callback);
|
if(result!=null&&result==1){
|
return true;
|
}
|
return false;
|
}
|
|
/**
|
* 获得缓存的基本对象列表
|
*
|
* @param pattern 字符串前缀
|
* @return 对象列表
|
*/
|
public Set<String> keys(final String pattern)
|
{
|
return redisTemplate.keys(pattern);
|
}
|
|
|
}
|