init

src/main/java/com/newyetai/frame/redis/config/RedisCacheConfig.java

@@ -0,0 +1,110 @@
+package com.newyetai.frame.redis.config;
+
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import org.springframework.beans.factory.ObjectProvider;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.AutoConfigureAfter;
+import org.springframework.boot.autoconfigure.cache.CacheManagerCustomizer;
+import org.springframework.boot.autoconfigure.cache.CacheManagerCustomizers;
+import org.springframework.boot.autoconfigure.cache.CacheProperties;
+import org.springframework.boot.autoconfigure.cache.CacheProperties.Redis;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+import org.springframework.core.io.ResourceLoader;
+import org.springframework.data.redis.cache.RedisCacheManager;
+import org.springframework.data.redis.cache.RedisCacheManager.RedisCacheManagerBuilder;
+import org.springframework.data.redis.connection.RedisConnectionFactory;
+import org.springframework.data.redis.serializer.JdkSerializationRedisSerializer;
+import org.springframework.data.redis.serializer.RedisSerializationContext.SerializationPair;
+
+/**
+ * 参考 org.springframework.boot.autoconfigure.cache.RedisCacheConfiguration
+ * org.springframework.boot.autoconfigure.cache.CacheAutoConfiguration
+ * 
+ * @author sr on 2020-07-16
+ */
+@Configuration
+@AutoConfigureAfter({RedisAutoConfiguration.class})
+@ConditionalOnClass(RedisConnectionFactory.class)
+//@ConditionalOnBean(RedisConnectionFactory.class)
+@ConditionalOnMissingBean(RedisCacheManager.class)
+@EnableConfigurationProperties({CacheProperties.class})
+public class RedisCacheConfig {
+
+  private final CacheProperties cacheProperties;
+
+  /** CacheAutoConfiguration中定义，自动注入 **/
+  @Autowired
+  private CacheManagerCustomizers customizerInvoker;
+
+  private final org.springframework.data.redis.cache.RedisCacheConfiguration redisCacheConfiguration;
+
+  RedisCacheConfig(CacheProperties cacheProperties,
+      ObjectProvider<org.springframework.data.redis.cache.RedisCacheConfiguration> redisCacheConfiguration) {
+    this.cacheProperties = cacheProperties;
+    this.redisCacheConfiguration = redisCacheConfiguration.getIfAvailable();
+  }
+
+  @Bean
+  @ConditionalOnMissingBean
+  public CacheManagerCustomizers cacheManagerCustomizers(
+      ObjectProvider<CacheManagerCustomizer<?>> customizers) {
+    return new CacheManagerCustomizers(customizers.orderedStream().collect(Collectors.toList()));
+  }
+
+  /**
+   * redis缓存，默认缓存
+   * 
+   * @param redisConnectionFactory
+   * @param resourceLoader
+   * @return
+   * @author sr on 2020-07-16
+   */
+  @Bean
+  @Primary
+  public RedisCacheManager redisCacheManager(RedisConnectionFactory redisConnectionFactory,
+      ResourceLoader resourceLoader) {
+    RedisCacheManagerBuilder builder = RedisCacheManager.builder(redisConnectionFactory)
+        .cacheDefaults(determineConfiguration(resourceLoader.getClassLoader()));
+    List<String> cacheNames = this.cacheProperties.getCacheNames();
+    if (!cacheNames.isEmpty()) {
+      builder.initialCacheNames(new LinkedHashSet<>(cacheNames));
+    }
+    return this.customizerInvoker.customize(builder.build());
+  }
+
+  private org.springframework.data.redis.cache.RedisCacheConfiguration determineConfiguration(
+      ClassLoader classLoader) {
+    if (this.redisCacheConfiguration != null) {
+      return this.redisCacheConfiguration;
+    }
+    Redis redisProperties = this.cacheProperties.getRedis();
+    org.springframework.data.redis.cache.RedisCacheConfiguration config =
+        org.springframework.data.redis.cache.RedisCacheConfiguration
+            .defaultCacheConfig();
+    config = config.serializeValuesWith(
+        SerializationPair.fromSerializer(new JdkSerializationRedisSerializer(classLoader)));
+    if (redisProperties.getTimeToLive() != null) {
+      config = config.entryTtl(redisProperties.getTimeToLive());
+    }
+    if (redisProperties.getKeyPrefix() != null) {
+      config = config.prefixKeysWith(redisProperties.getKeyPrefix());
+    }
+    if (!redisProperties.isCacheNullValues()) {
+      config = config.disableCachingNullValues();
+    }
+    if (!redisProperties.isUseKeyPrefix()) {
+      config = config.disableKeyPrefix();
+    }
+    return config;
+  }
+
+}