devbox
我家小花儿
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

SpringBoot两级缓存实现_caffine 和redis 双缓存机制 spring boot 代码

作者：我家小花儿 | 2024-08-04 22:53:25

赞

踩

caffine 和redis 双缓存机制 spring boot 代码

 主要实现代码如下：

  1. /**
  2.  * 1. 支持解析 @CacheConfig.cacheNames ，独立为每个方法设置过期时间，单位秒
  3.  * 2. 返回增强后的Cache实例 ：new RedisPlusCache
  4.  *
  5.  * @see RedisPlusCache
  6.  **/
  7. public class RedisCachePlusManager extends RedisCacheManager {
  8.     private final CachePlusConfig config;
  9.  
  10.     public RedisCachePlusManager(RedisCacheWriter cacheWriter, RedisCacheConfiguration defaultCacheConfiguration, CachePlusConfig config) {
  11.         super(cacheWriter, defaultCacheConfiguration);
  12.         this.config = config.clone();
  13.     }
  14.  
  15.     @NonNull
  16.     @Override
  17.     protected RedisCache createRedisCache(@NonNull String name, @Nullable RedisCacheConfiguration cacheConfig) {
  18.         String[] array = name.split(this.config.getDelimiter());
  19.         name = array[0];
  20.         if (array.length > 1 && cacheConfig != null) {
  21.             long ttl = Long.parseLong(array[1]);
  22.             cacheConfig = cacheConfig.entryTtl(Duration.ofSeconds(ttl));
  23.         }
  24.         if (array.length > 2) {
  25.             this.config.setDuration(Long.parseLong(array[2]));
  26.         }
  27.         RedisCache redisCache = super.createRedisCache(name, cacheConfig);
  28.         return new RedisPlusCache(redisCache.getName(), redisCache.getNativeCache(), redisCache.getCacheConfiguration(), this.config);
  29.     }
  30. }
  1. /**
  2.  * 1. 基于CaffeineCache实现二级缓存
  3.  *
  4.  * @see CachePlusConfig
  5.  **/
  6. public class RedisPlusCache extends RedisCache {
  7.     private final Map<Long, Cache<Object, Object>> caffeineCaches;
  8.     private final CachePlusConfig config;
  9.  
  10.     protected RedisPlusCache(String name, RedisCacheWriter cacheWriter, RedisCacheConfiguration cacheConfig, CachePlusConfig config) {
  11.         super(name, cacheWriter, cacheConfig);
  12.         this.caffeineCaches = new ConcurrentHashMap<>();
  13.         this.config = config;
  14.     }
  15.  
  16.     @Override
  17.     protected Object lookup(@NonNull Object key) {
  18.         RedisCacheConfiguration cacheConfiguration = super.getCacheConfiguration();
  19.         Duration duration = cacheConfiguration.getTtl();
  20.         Cache<Object, Object> localCache = this.caffeineCaches.get(duration.getSeconds());
  21.         if (localCache == null) return this.lookupAndPutLocal(key);
  22.         Object v = localCache.get(key, oneKey -> null);
  23.         return v == null ? this.lookupAndPutLocal(key) : v;
  24.     }
  25.  
  26.     @Override
  27.     public void put(@NonNull Object key, @Nullable Object value) {
  28.         this.putLocal(key, value);
  29.         super.put(key, value);
  30.     }
  31.  
  32.     private Object lookupAndPutLocal(@NonNull Object key) {
  33.         Object v = super.lookup(key);
  34.         this.putLocal(key, v);
  35.         return v;
  36.     }
  37.  
  38.     private void putLocal(Object key, Object value) {
  39.         Object cacheValue = preProcessCacheValue(value);
  40.         if (cacheValue != null) {
  41.             Cache<Object, Object> localCache = this.createLocalCache();
  42.             localCache.put(key, cacheValue);
  43.         }
  44.     }
  45.  
  46.     private Cache<Object, Object> createLocalCache() {
  47.         long time = this.getCacheConfiguration().getTtl().getSeconds();
  48.         Cache<Object, Object> localCache = this.caffeineCaches.get(time);
  49.         if (localCache != null) return localCache;
  50.         synchronized (this) {
  51.             localCache = this.caffeineCaches.get(time);
  52.             if (localCache != null) return localCache;
  53.             localCache = Caffeine.newBuilder()
  54.                     .initialCapacity((int) (this.config.getDuration() >> 2))
  55.                     .expireAfterWrite(this.config.getDuration(), TimeUnit.SECONDS)
  56.                     .maximumSize(this.config.getMaximumSize()).build();
  57.             this.caffeineCaches.put(time, localCache);
  58.             return localCache;
  59.         }
  60.     }
  61. }
  1. /**
  2.  * 缓存配置类
  3.  **/
  4. @Data
  5. public class CachePlusConfig implements Serializable, Cloneable {
  6.     // 分隔符
  7.     private String delimiter;
  8.     // 过期时间
  9.     private long duration;
  10.     // 最大缓存容量
  11.     private long maximumSize;
  12.  
  13.     public CachePlusConfig() {
  14.         this.delimiter = "##";
  15.         this.duration = 600L;
  16.         this.maximumSize = 100_000L;
  17.     }
  18.  
  19.     @Override
  20.     public CachePlusConfig clone() {
  21.         try {
  22.             return (CachePlusConfig) super.clone();
  23.         } catch (CloneNotSupportedException e) {
  24.             throw new IllegalStateException("clone error", e);
  25.         }
  26.     }
  27. }

注册类：

  1. @Configuration
  2. @EnableCaching
  3. public class SpringCacheConfig {
  4.  
  5.     @Resource
  6.     @Qualifier("redisConnectionFactory")
  7.     private RedisConnectionFactory redisConnectionFactory;
  8.  
  9.     @Resource
  10.     private CachePlusConfig localCachePlusConfig;
  11.  
  12.     @ConfigurationProperties("cache-plus.local")
  13.     @Bean
  14.     public CachePlusConfig get() {
  15.         return new CachePlusConfig();
  16.     }
  17.  
  18.     @Bean("commonKeyGenerator")
  19.     public KeyGenerator getCommonKeyGenerator() {
  20.         return new CommonKeyGenerator();
  21.     }
  22.  
  23.     @Primary
  24.     @Bean
  25.     public CacheManager cacheManager() {
  26.         RedisCacheConfiguration redisCacheConfiguration = RedisCacheConfiguration.defaultCacheConfig()
  27.                 .entryTtl(Duration.ofSeconds(300));
  28.         StringRedisSerializer stringRedisSerializer = new StringRedisSerializer();
  29.         GenericFastJsonRedisSerializer genericFastJsonRedisSerializer = new GenericFastJsonRedisSerializer();
  30.         redisCacheConfiguration = redisCacheConfiguration.serializeKeysWith(RedisSerializationContext.SerializationPair.fromSerializer(stringRedisSerializer))
  31.                 .serializeValuesWith(RedisSerializationContext.SerializationPair.fromSerializer(genericFastJsonRedisSerializer));
  32.         return new RedisCachePlusManager(RedisCacheWriter.lockingRedisCacheWriter(redisConnectionFactory), redisCacheConfiguration, localCachePlusConfig);
  33.     }
  34.  
  35.     @Bean("localCacheManager")
  36.     public CacheManager localCacheManager() {
  37.         CaffeineCacheManager localCacheManager = new CaffeineCachePlusManager(this.localCachePlusConfig);
  38.         localCacheManager.setAllowNullValues(true);
  39.         return localCacheManager;
  40.     }
  41. }
  1. @Configuration
  2. public class RedisCommonConfig {
  3.  
  4.     @ConditionalOnMissingBean(name = "redisTemplate")
  5.     @Bean(name = "redisTemplate")
  6.     public RedisTemplate<String, Object> redisTemplate(@Qualifier("redisConnectionFactory") RedisConnectionFactory redisConnectionFactory) {
  7.         RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
  8.         redisTemplate.setConnectionFactory(redisConnectionFactory);
  9.         GenericFastJsonRedisSerializer genericFastJsonRedisSerializer = new GenericFastJsonRedisSerializer();
  10.         StringRedisSerializer stringRedisSerializer = new StringRedisSerializer();
  11.         // 设置value的序列化规则和 key的序列化规则
  12.         redisTemplate.setKeySerializer(stringRedisSerializer);
  13.         redisTemplate.setValueSerializer(genericFastJsonRedisSerializer);
  14.         redisTemplate.setHashKeySerializer(stringRedisSerializer);
  15.         redisTemplate.setHashValueSerializer(genericFastJsonRedisSerializer);
  16.         redisTemplate.setDefaultSerializer(genericFastJsonRedisSerializer);
  17.         redisTemplate.setEnableDefaultSerializer(true);
  18.         return redisTemplate;
  19.     }
  20.  
  21.     @ConditionalOnMissingBean(name = "redisConnectionFactory")
  22.     @Bean(name = "redisConnectionFactory")
  23.     public RedisConnectionFactory redisConnectionFactory() {
  24.         RedisStandaloneConfiguration configuration = new RedisStandaloneConfiguration("127.0.0.1", 6379);
  25.         return new LettuceConnectionFactory(configuration);
  26.     }
  27. }

扩展，单独注册只支持二级缓存(本地缓存)的CacheManager，基于CaffeineCache实现：

  1. /**
  2.  * 1. 支持解析 @CacheConfig.cacheNames ，独立为每个方法设置过期时间，单位秒
  3.  **/
  4. @Slf4j
  5. public class CaffeineCachePlusManager extends CaffeineCacheManager {
  6.  
  7.     private final CachePlusConfig config;
  8.  
  9.     public CaffeineCachePlusManager(CachePlusConfig config) {
  10.         this.config = config;
  11.     }
  12.  
  13.     @NonNull
  14.     @Override
  15.     protected com.github.benmanes.caffeine.cache.Cache<Object, Object> createNativeCaffeineCache(@NonNull String name) {
  16.         long expire = this.getExpire(name);
  17.         if (expire > 0) {
  18.             return Caffeine.newBuilder()
  19.                     .initialCapacity((int)(this.config.getMaximumSize() >> 4))
  20.                     // 3600秒后自动删除
  21.                     .expireAfterWrite(expire, TimeUnit.SECONDS)
  22.                     // 最大容量10w个，超过会自动清理空间
  23.                     .maximumSize(this.config.getMaximumSize())
  24.                     .removalListener(((key, value, cause) -> {
  25.                         //清理通知 key, value ==> 键值对   cause ==> 清理原因
  26.                         log.info("caffeine cache removed: {}, {}, {}", key, value, cause);
  27.                     })).build();
  28.         }
  29.         return super.createNativeCaffeineCache(name);
  30.     }
  31.  
  32.     private long getExpire(@NonNull String name) {
  33.         if (this.config.getDelimiter() == null) return this.config.getDuration();
  34.         String[] split = name.split(this.config.getDelimiter());
  35.         return split.length > 1 ? Long.parseLong(split[1]) : this.config.getDuration();
  36.     }
  37. }

扩展，键生成器

  1. /**
  2.  * 通用的缓存键生成器
  3.  **/
  4. public class CommonKeyGenerator implements KeyGenerator {
  5.  
  6.     @NonNull
  7.     @Override
  8.     public Object generate(Object target, Method method, @NonNull Object... params) {
  9.         return target.getClass().getName() + ":" + method.getName() + ":" + generateKey(params);
  10.     }
  11.  
  12.     /**
  13.      * Generate a key based on the specified parameters.
  14.      */
  15.     public static Object generateKey(Object... params) {
  16.         if (params.length == 0) {
  17.             return SimpleKey.EMPTY;
  18.         }
  19.         if (params.length == 1) {
  20.             Object param = params[0];
  21.             if (param != null && !param.getClass().isArray()) {
  22.                 return param;
  23.             }
  24.         }
  25.         return StringUtils.arrayToDelimitedString(params, "-");
  26.     }
  27. }
  1. /**
  2.  * 缓存键生成器-从方法开始拼接
  3.  **/
  4. public class MethodKeyGenerator implements KeyGenerator {
  5.  
  6.     @NonNull
  7.     @Override
  8.     public Object generate(@NonNull Object target, Method method, @NonNull Object... params) {
  9.         return method.getName() + ":" + CommonKeyGenerator.generateKey(params);
  10.     }
  11. }

业务代码示例：

  1. @CacheConfig(cacheNames = "DICT##120", keyGenerator = "commonKeyGenerator")
  2. @Service
  3. public class RegionService {
  4.  
  5.     // 过期时间、键生成，继承@CacheConfig
  6.     @Cacheable
  7.     public String getCityName(String code) {
  8.         if ("010".equals(code)) return "beijing";
  9.         else if (("020".equals(code))) return "shanghai";
  10.         else return "shenzhen";
  11.     }
  12.  
  13.     @Cacheable
  14.     public String getCityName(String code, String lang) {
  15.         if ("010".equals(code)) return "beijing";
  16.         else if (("020".equals(code))) return "shanghai";
  17.         else return "shenzhen";
  18.     }
  19.  
  20.     // 该方法独立设置过期时间240秒
  21.     @Cacheable(cacheNames = "DICT##240")
  22.     public String getCityName240(String code, String lang) {
  23.         if ("010".equals(code)) return "beijing";
  24.         else if (("020".equals(code))) return "shanghai";
  25.         else return "shenzhen";
  26.     }
  27.  
  28.     // 该方法只支持二级缓存
  29.     @Cacheable(cacheNames = "DICT##10", cacheManager = "localCacheManager")
  30.     public String getCityNameByLocal(String code) {
  31.         if ("010".equals(code)) return "beijing";
  32.         else if (("020".equals(code))) return "shanghai";
  33.         else return "shenzhen";
  34.     }
  35.  
  36. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/929810
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号