update 重写 spring-cache 实现 更人性化的操作 支持注解指定ttl等一些参数
疯狂的狮子li
3 years ago
parent
87f5fdee6d
commit
373a975c3b
@ -0,0 +1,48 @@
|
|||||||
|
package com.ruoyi.common.core.constant;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 缓存组名称常量
|
||||||
|
* <p>
|
||||||
|
* key 格式为 cacheNames#ttl#maxIdleTime#maxSize
|
||||||
|
* <p>
|
||||||
|
* ttl 过期时间 如果设置为0则不过期 默认为0
|
||||||
|
* maxIdleTime 最大空闲时间 根据LRU算法清理空闲数据 如果设置为0则不检测 默认为0
|
||||||
|
* maxSize 组最大长度 根据LRU算法清理溢出数据 如果设置为0则无限长 默认为0
|
||||||
|
* <p>
|
||||||
|
* 例子: test#60s、test#0#60s、test#0#1m#1000、test#1h#0#500
|
||||||
|
*
|
||||||
|
* @author Lion Li
|
||||||
|
*/
|
||||||
|
public interface CacheNames {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 演示案例
|
||||||
|
*/
|
||||||
|
String DEMO_CACHE = "demo:cache#60s#10m#20";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 系统配置
|
||||||
|
*/
|
||||||
|
String SYS_CONFIG = "sys_config";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 数据字典
|
||||||
|
*/
|
||||||
|
String SYS_DICT = "sys_dict";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* OSS内容
|
||||||
|
*/
|
||||||
|
String SYS_OSS = "sys_oss#30d";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* OSS配置
|
||||||
|
*/
|
||||||
|
String SYS_OSS_CONFIG = "sys_oss_config";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 在线用户
|
||||||
|
*/
|
||||||
|
String ONLINE_TOKEN = "online_tokens";
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,191 @@
|
|||||||
|
/**
|
||||||
|
* Copyright (c) 2013-2021 Nikita Koksharov
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package com.ruoyi.common.redis.manager;
|
||||||
|
|
||||||
|
import com.ruoyi.common.redis.utils.RedisUtils;
|
||||||
|
import org.redisson.api.RMap;
|
||||||
|
import org.redisson.api.RMapCache;
|
||||||
|
import org.redisson.spring.cache.CacheConfig;
|
||||||
|
import org.redisson.spring.cache.RedissonCache;
|
||||||
|
import org.springframework.boot.convert.DurationStyle;
|
||||||
|
import org.springframework.cache.Cache;
|
||||||
|
import org.springframework.cache.CacheManager;
|
||||||
|
import org.springframework.cache.transaction.TransactionAwareCacheDecorator;
|
||||||
|
import org.springframework.util.StringUtils;
|
||||||
|
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.ConcurrentMap;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A {@link CacheManager} implementation
|
||||||
|
* backed by Redisson instance.
|
||||||
|
* <p>
|
||||||
|
* 修改 RedissonSpringCacheManager 源码
|
||||||
|
* 重写 cacheName 处理方法 支持多参数
|
||||||
|
*
|
||||||
|
* @author Nikita Koksharov
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
public class PlusSpringCacheManager implements CacheManager {
|
||||||
|
|
||||||
|
private boolean dynamic = true;
|
||||||
|
|
||||||
|
private boolean allowNullValues = true;
|
||||||
|
|
||||||
|
private boolean transactionAware = true;
|
||||||
|
|
||||||
|
Map<String, CacheConfig> configMap = new ConcurrentHashMap<>();
|
||||||
|
ConcurrentMap<String, Cache> instanceMap = new ConcurrentHashMap<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates CacheManager supplied by Redisson instance
|
||||||
|
*/
|
||||||
|
public PlusSpringCacheManager() {
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Defines possibility of storing {@code null} values.
|
||||||
|
* <p>
|
||||||
|
* Default is <code>true</code>
|
||||||
|
*
|
||||||
|
* @param allowNullValues stores if <code>true</code>
|
||||||
|
*/
|
||||||
|
public void setAllowNullValues(boolean allowNullValues) {
|
||||||
|
this.allowNullValues = allowNullValues;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Defines if cache aware of Spring-managed transactions.
|
||||||
|
* If {@code true} put/evict operations are executed only for successful transaction in after-commit phase.
|
||||||
|
* <p>
|
||||||
|
* Default is <code>false</code>
|
||||||
|
*
|
||||||
|
* @param transactionAware cache is transaction aware if <code>true</code>
|
||||||
|
*/
|
||||||
|
public void setTransactionAware(boolean transactionAware) {
|
||||||
|
this.transactionAware = transactionAware;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Defines 'fixed' cache names.
|
||||||
|
* A new cache instance will not be created in dynamic for non-defined names.
|
||||||
|
* <p>
|
||||||
|
* `null` parameter setups dynamic mode
|
||||||
|
*
|
||||||
|
* @param names of caches
|
||||||
|
*/
|
||||||
|
public void setCacheNames(Collection<String> names) {
|
||||||
|
if (names != null) {
|
||||||
|
for (String name : names) {
|
||||||
|
getCache(name);
|
||||||
|
}
|
||||||
|
dynamic = false;
|
||||||
|
} else {
|
||||||
|
dynamic = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Set cache config mapped by cache name
|
||||||
|
*
|
||||||
|
* @param config object
|
||||||
|
*/
|
||||||
|
public void setConfig(Map<String, ? extends CacheConfig> config) {
|
||||||
|
this.configMap = (Map<String, CacheConfig>) config;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected CacheConfig createDefaultConfig() {
|
||||||
|
return new CacheConfig();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Cache getCache(String name) {
|
||||||
|
Cache cache = instanceMap.get(name);
|
||||||
|
if (cache != null) {
|
||||||
|
return cache;
|
||||||
|
}
|
||||||
|
if (!dynamic) {
|
||||||
|
return cache;
|
||||||
|
}
|
||||||
|
|
||||||
|
CacheConfig config = configMap.get(name);
|
||||||
|
if (config == null) {
|
||||||
|
config = createDefaultConfig();
|
||||||
|
configMap.put(name, config);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 重写 cacheName 支持多参数
|
||||||
|
String[] array = StringUtils.delimitedListToStringArray(name, "#");
|
||||||
|
name = array[0];
|
||||||
|
if (array.length > 1) {
|
||||||
|
config.setTTL(DurationStyle.detectAndParse(array[1]).toMillis());
|
||||||
|
}
|
||||||
|
if (array.length > 2) {
|
||||||
|
config.setMaxIdleTime(DurationStyle.detectAndParse(array[2]).toMillis());
|
||||||
|
}
|
||||||
|
if (array.length > 3) {
|
||||||
|
config.setMaxSize(Integer.parseInt(array[3]));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (config.getMaxIdleTime() == 0 && config.getTTL() == 0 && config.getMaxSize() == 0) {
|
||||||
|
return createMap(name, config);
|
||||||
|
}
|
||||||
|
|
||||||
|
return createMapCache(name, config);
|
||||||
|
}
|
||||||
|
|
||||||
|
private Cache createMap(String name, CacheConfig config) {
|
||||||
|
RMap<Object, Object> map = RedisUtils.getClient().getMap(name);
|
||||||
|
|
||||||
|
Cache cache = new RedissonCache(map, allowNullValues);
|
||||||
|
if (transactionAware) {
|
||||||
|
cache = new TransactionAwareCacheDecorator(cache);
|
||||||
|
}
|
||||||
|
Cache oldCache = instanceMap.putIfAbsent(name, cache);
|
||||||
|
if (oldCache != null) {
|
||||||
|
cache = oldCache;
|
||||||
|
}
|
||||||
|
return cache;
|
||||||
|
}
|
||||||
|
|
||||||
|
private Cache createMapCache(String name, CacheConfig config) {
|
||||||
|
RMapCache<Object, Object> map = RedisUtils.getClient().getMapCache(name);
|
||||||
|
|
||||||
|
Cache cache = new RedissonCache(map, config, allowNullValues);
|
||||||
|
if (transactionAware) {
|
||||||
|
cache = new TransactionAwareCacheDecorator(cache);
|
||||||
|
}
|
||||||
|
Cache oldCache = instanceMap.putIfAbsent(name, cache);
|
||||||
|
if (oldCache != null) {
|
||||||
|
cache = oldCache;
|
||||||
|
} else {
|
||||||
|
map.setMaxSize(config.getMaxSize());
|
||||||
|
}
|
||||||
|
return cache;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Collection<String> getCacheNames() {
|
||||||
|
return Collections.unmodifiableSet(configMap.keySet());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue