提供redis scan()

pull/64/head
郭向欣 5 years ago
parent 3d0f8d681e
commit 9c2b84d761

@ -1,24 +1,18 @@
package com.ruoyi.common.redis.service; package com.ruoyi.common.redis.service;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.BoundSetOperations; import org.springframework.data.redis.core.*;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.TimeUnit;
/** /**
* spring redis * spring redis
* *
* @author ruoyi * @author ruoyi
**/ **/
@SuppressWarnings(value = { "unchecked", "rawtypes" })
@Component @Component
public class RedisService public class RedisService
{ {
@ -31,7 +25,7 @@ public class RedisService
* @param key * @param key
* @param value * @param value
*/ */
public <T> void setCacheObject(final String key, final T value) public <T> void setCacheObject(String key, T value)
{ {
redisTemplate.opsForValue().set(key, value); redisTemplate.opsForValue().set(key, value);
} }
@ -44,7 +38,7 @@ public class RedisService
* @param timeout * @param timeout
* @param timeUnit * @param timeUnit
*/ */
public <T> void setCacheObject(final String key, final T value, final Long timeout, final TimeUnit timeUnit) public <T> void setCacheObject(String key, T value, Long timeout, TimeUnit timeUnit)
{ {
redisTemplate.opsForValue().set(key, value, timeout, timeUnit); redisTemplate.opsForValue().set(key, value, timeout, timeUnit);
} }
@ -56,7 +50,7 @@ public class RedisService
* @param timeout * @param timeout
* @return true=false= * @return true=false=
*/ */
public boolean expire(final String key, final long timeout) public boolean expire(String key, long timeout)
{ {
return expire(key, timeout, TimeUnit.SECONDS); return expire(key, timeout, TimeUnit.SECONDS);
} }
@ -69,7 +63,7 @@ public class RedisService
* @param unit * @param unit
* @return true=false= * @return true=false=
*/ */
public boolean expire(final String key, final long timeout, final TimeUnit unit) public boolean expire(String key, long timeout, TimeUnit unit)
{ {
return redisTemplate.expire(key, timeout, unit); return redisTemplate.expire(key, timeout, unit);
} }
@ -80,7 +74,7 @@ public class RedisService
* @param key * @param key
* @return * @return
*/ */
public <T> T getCacheObject(final String key) public <T> T getCacheObject(String key)
{ {
ValueOperations<String, T> operation = redisTemplate.opsForValue(); ValueOperations<String, T> operation = redisTemplate.opsForValue();
return operation.get(key); return operation.get(key);
@ -91,7 +85,7 @@ public class RedisService
* *
* @param key * @param key
*/ */
public boolean deleteObject(final String key) public boolean deleteObject(String key)
{ {
return redisTemplate.delete(key); return redisTemplate.delete(key);
} }
@ -102,7 +96,7 @@ public class RedisService
* @param collection * @param collection
* @return * @return
*/ */
public long deleteObject(final Collection collection) public long deleteObject(Collection collection)
{ {
return redisTemplate.delete(collection); return redisTemplate.delete(collection);
} }
@ -114,7 +108,7 @@ public class RedisService
* @param dataList List * @param dataList List
* @return * @return
*/ */
public <T> long setCacheList(final String key, final List<T> dataList) public <T> long setCacheList(String key, List<T> dataList)
{ {
Long count = redisTemplate.opsForList().rightPushAll(key, dataList); Long count = redisTemplate.opsForList().rightPushAll(key, dataList);
return count == null ? 0 : count; return count == null ? 0 : count;
@ -126,7 +120,7 @@ public class RedisService
* @param key * @param key
* @return * @return
*/ */
public <T> List<T> getCacheList(final String key) public <T> List<T> getCacheList(String key)
{ {
return redisTemplate.opsForList().range(key, 0, -1); return redisTemplate.opsForList().range(key, 0, -1);
} }
@ -138,7 +132,7 @@ public class RedisService
* @param dataSet * @param dataSet
* @return * @return
*/ */
public <T> BoundSetOperations<String, T> setCacheSet(final String key, final Set<T> dataSet) public <T> BoundSetOperations<String, T> setCacheSet(String key, Set<T> dataSet)
{ {
BoundSetOperations<String, T> setOperation = redisTemplate.boundSetOps(key); BoundSetOperations<String, T> setOperation = redisTemplate.boundSetOps(key);
Iterator<T> it = dataSet.iterator(); Iterator<T> it = dataSet.iterator();
@ -155,7 +149,7 @@ public class RedisService
* @param key * @param key
* @return * @return
*/ */
public <T> Set<T> getCacheSet(final String key) public <T> Set<T> getCacheSet(String key)
{ {
return redisTemplate.opsForSet().members(key); return redisTemplate.opsForSet().members(key);
} }
@ -166,7 +160,7 @@ public class RedisService
* @param key * @param key
* @param dataMap * @param dataMap
*/ */
public <T> void setCacheMap(final String key, final Map<String, T> dataMap) public <T> void setCacheMap(String key, Map<String, T> dataMap)
{ {
if (dataMap != null) { if (dataMap != null) {
redisTemplate.opsForHash().putAll(key, dataMap); redisTemplate.opsForHash().putAll(key, dataMap);
@ -179,7 +173,7 @@ public class RedisService
* @param key * @param key
* @return * @return
*/ */
public <T> Map<String, T> getCacheMap(final String key) public <T> Map<String, T> getCacheMap(String key)
{ {
return redisTemplate.opsForHash().entries(key); return redisTemplate.opsForHash().entries(key);
} }
@ -191,7 +185,7 @@ public class RedisService
* @param hKey Hash * @param hKey Hash
* @param value * @param value
*/ */
public <T> void setCacheMapValue(final String key, final String hKey, final T value) public <T> void setCacheMapValue(String key, String hKey, T value)
{ {
redisTemplate.opsForHash().put(key, hKey, value); redisTemplate.opsForHash().put(key, hKey, value);
} }
@ -203,7 +197,7 @@ public class RedisService
* @param hKey Hash * @param hKey Hash
* @return Hash * @return Hash
*/ */
public <T> T getCacheMapValue(final String key, final String hKey) public <T> T getCacheMapValue(String key, String hKey)
{ {
HashOperations<String, String, T> opsForHash = redisTemplate.opsForHash(); HashOperations<String, String, T> opsForHash = redisTemplate.opsForHash();
return opsForHash.get(key, hKey); return opsForHash.get(key, hKey);
@ -216,7 +210,7 @@ public class RedisService
* @param hKeys Hash * @param hKeys Hash
* @return Hash * @return Hash
*/ */
public <T> List<T> getMultiCacheMapValue(final String key, final Collection<Object> hKeys) public <T> List<T> getMultiCacheMapValue(String key, Collection<Object> hKeys)
{ {
return redisTemplate.opsForHash().multiGet(key, hKeys); return redisTemplate.opsForHash().multiGet(key, hKeys);
} }
@ -227,8 +221,31 @@ public class RedisService
* @param pattern * @param pattern
* @return * @return
*/ */
public Collection<String> keys(final String pattern) @Deprecated
public Collection<String> keys(String pattern)
{ {
return redisTemplate.keys(pattern); return redisTemplate.keys(pattern);
} }
/**
* keys
*
* @param matchKey
* @return
*/
public Set<String> scan(String matchKey) {
Set<String> keys = (Set<String>) redisTemplate.execute((RedisCallback<Set<String>>) connection -> {
Set<String> keysTmp = new HashSet<>();
Cursor<byte[]> cursor = connection.scan(new ScanOptions.ScanOptionsBuilder().match(matchKey).count(1000).build());
while (cursor.hasNext()) {
keysTmp.add(new String(cursor.next()));
}
try {
cursor.close();
} catch (IOException e) {
}
return keysTmp;
});
return keys;
}
} }

Loading…
Cancel
Save