feat: ✨ redis cache 设置

bubble-spring-boot-starters/bubble-spring-boot-starter-data-redis/src/main/java/cn/fxbin/bubble/data/redis/autoconfigure/BubbleRedisCacheAutoConfiguration.java

@@ -0,0 +1,109 @@
+package cn.fxbin.bubble.data.redis.autoconfigure;
+
+import cn.fxbin.bubble.core.constant.CharPool;
+import cn.fxbin.bubble.core.constant.StringPool;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.ObjectProvider;
+import org.springframework.boot.autoconfigure.AutoConfigureBefore;
+import org.springframework.boot.autoconfigure.cache.CacheAutoConfiguration;
+import org.springframework.boot.autoconfigure.cache.CacheManagerCustomizers;
+import org.springframework.boot.autoconfigure.cache.CacheProperties;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.cache.CacheManager;
+import org.springframework.cache.annotation.EnableCaching;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+import org.springframework.data.redis.cache.RedisCacheConfiguration;
+import org.springframework.data.redis.cache.RedisCacheManager;
+import org.springframework.data.redis.cache.RedisCacheWriter;
+import org.springframework.data.redis.connection.RedisConnectionFactory;
+import org.springframework.data.redis.serializer.RedisSerializationContext;
+import org.springframework.data.redis.serializer.RedisSerializer;
+import org.springframework.lang.Nullable;
+
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+/**
+ * RedisCacheAutoConfiguration
+ *
+ * @author fxbin
+ * @version v1.0
+ * @since 2023/9/14 20:27
+ */
+@EnableCaching
+@Configuration(
+ proxyBeanMethods = false
+)
+@RequiredArgsConstructor
+@AutoConfigureBefore(CacheAutoConfiguration.class)
+@EnableConfigurationProperties(CacheProperties.class)
+public class BubbleRedisCacheAutoConfiguration {
+
+
+ /**
+ * 序列化方式
+ */
+ private final RedisSerializer<Object> redisSerializer;
+ private final CacheProperties cacheProperties;
+ private final CacheManagerCustomizers customizerInvoker;
+ @Nullable
+ private final RedisCacheConfiguration redisCacheConfiguration;
+
+ @Primary
+ @Bean("cacheResolver")
+ public CacheManager redisCacheManager(ObjectProvider<RedisConnectionFactory> connectionFactoryObjectProvider) {
+ RedisConnectionFactory connectionFactory = connectionFactoryObjectProvider.getIfAvailable();
+ Objects.requireNonNull(connectionFactory, "Bean RedisConnectionFactory is null.");
+ RedisCacheWriter redisCacheWriter = RedisCacheWriter.nonLockingRedisCacheWriter(connectionFactory);
+ RedisCacheConfiguration cacheConfiguration = this.determineConfiguration();
+ List<String> cacheNames = this.cacheProperties.getCacheNames();
+ Map<String, RedisCacheConfiguration> initialCaches = new LinkedHashMap<>();
+ if (!cacheNames.isEmpty()) {
+ Map<String, RedisCacheConfiguration> cacheConfigMap = new LinkedHashMap<>(cacheNames.size());
+ cacheNames.forEach(it -> cacheConfigMap.put(it, cacheConfiguration));
+ initialCaches.putAll(cacheConfigMap);
+ }
+ boolean allowInFlightCacheCreation = true;
+ boolean enableTransactions = false;
+ RedisCacheManager cacheManager = new RedisCacheManager(
+ redisCacheWriter, cacheConfiguration, allowInFlightCacheCreation, initialCaches
+ );
+ cacheManager.setTransactionAware(enableTransactions);
+ return this.customizerInvoker.customize(cacheManager);
+ }
+
+ /**
+ * 选择配置
+ *
+ * @return {@link RedisCacheConfiguration}
+ */
+ private RedisCacheConfiguration determineConfiguration() {
+ if (this.redisCacheConfiguration != null) {
+ return this.redisCacheConfiguration;
+ } else {
+ CacheProperties.Redis redisProperties = this.cacheProperties.getRedis();
+ RedisCacheConfiguration config = RedisCacheConfiguration.defaultCacheConfig();
+ // 设置默认缓存名分割符号为 “:”，如果已经带 “:” 则不设置。
+ config = config.computePrefixWith(name -> name.endsWith(StringPool.COLON) ? name : name + CharPool.COLON);
+ config = config.serializeValuesWith(RedisSerializationContext.SerializationPair.fromSerializer(redisSerializer));
+ if (redisProperties.getTimeToLive() != null) {
+ config = config.entryTtl(redisProperties.getTimeToLive());
+ }
+ if (redisProperties.getKeyPrefix() != null) {
+ config = config.prefixCacheNameWith(redisProperties.getKeyPrefix());
+ }
+ if (!redisProperties.isCacheNullValues()) {
+ config = config.disableCachingNullValues();
+ }
+ if (!redisProperties.isUseKeyPrefix()) {
+ config = config.disableKeyPrefix();
+ }
+ return config;
+ }
+ }
+
+}