111 lines
4.6 KiB
Java
111 lines
4.6 KiB
Java
package com.senmesh.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;
|
||
}
|
||
|
||
}
|