RedisTemplateConfig sentinel(哨兵模式)/ cluster(集群模式) 常用配置 以及如何 一键配置切换

package cn.goswan.orient.gateway.config;
import io.lettuce.core.ReadFrom;
import io.lettuce.core.cluster.ClusterClientOptions;
import io.lettuce.core.cluster.ClusterTopologyRefreshOptions;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.AutoConfigureBefore;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
import org.springframework.boot.autoconfigure.data.redis.RedisProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisClusterConfiguration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.connection.RedisNode;
import org.springframework.data.redis.connection.RedisSentinelConfiguration;
import org.springframework.data.redis.connection.lettuce.LettuceClientConfiguration;
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
import org.springframework.data.redis.connection.lettuce.LettucePoolingClientConfiguration;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.JdkSerializationRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
/**
 * RedisTemplate  config配置
 *
 * @author alan.wang
 */
// @EnableCaching TODO
@Configuration
@AllArgsConstructor
@AutoConfigureBefore(RedisAutoConfiguration.class)
@Slf4j
public class RedisTemplateConfig {
  @Bean
  public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
    RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
    redisTemplate.setKeySerializer(new StringRedisSerializer());
    redisTemplate.setHashKeySerializer(new StringRedisSerializer());
    redisTemplate.setValueSerializer(new JdkSerializationRedisSerializer());
    redisTemplate.setHashValueSerializer(new JdkSerializationRedisSerializer());
    redisTemplate.setConnectionFactory(redisConnectionFactory);
    return redisTemplate;
  }
  @Bean
  public StringRedisTemplate stringRedisTemplate(RedisConnectionFactory redisConnectionFactory) {
    StringRedisTemplate stringRedisTemplate = new StringRedisTemplate();
    stringRedisTemplate.setKeySerializer(new StringRedisSerializer());
    stringRedisTemplate.setConnectionFactory(redisConnectionFactory);
    return stringRedisTemplate;
  }
  @Autowired
  RedisProperties redisProperties;
  @Bean
  public GenericObjectPoolConfig poolConfig() {
    GenericObjectPoolConfig config = new GenericObjectPoolConfig();
    config.setMinIdle(redisProperties.getLettuce().getPool().getMinIdle());
    config.setMaxIdle(redisProperties.getLettuce().getPool().getMaxIdle());
    config.setMaxTotal(redisProperties.getLettuce().getPool().getMaxActive());
    config.setMaxWaitMillis(redisProperties.getLettuce().getPool().getMaxWait().toMillis());
    return config;
  }
  @Bean
  @ConditionalOnProperty(value = "spring.redis.mode",havingValue = "sentinel")
  public RedisSentinelConfiguration redisSentinelConfiguration() {
    RedisSentinelConfiguration redisConfig = new RedisSentinelConfiguration();
    redisConfig.setMaster(redisProperties.getSentinel().getMaster());
    if(redisProperties.getSentinel().getNodes()!=null) {
      List<RedisNode> sentinelNode=new ArrayList<RedisNode>();
      for(String sen : redisProperties.getSentinel().getNodes()) {
        String[] arr = sen.split(":");
        sentinelNode.add(new RedisNode(arr[0],Integer.parseInt(arr[1])));
      }
      redisConfig.setDatabase(redisProperties.getDatabase());
      redisConfig.setPassword(redisProperties.getPassword());
      redisConfig.setSentinelPassword(redisConfig.getPassword());
      redisConfig.setSentinels(sentinelNode);
    }
    return redisConfig;
  }
  @Bean
  @ConditionalOnProperty(value = "spring.redis.mode",havingValue = "sentinel")
  public RedisConnectionFactory redisConnectionFactory(@Qualifier("poolConfig") GenericObjectPoolConfig config,
                             RedisSentinelConfiguration redisConfig) {//注意传入的对象名和类型RedisSentinelConfiguration
    LettuceClientConfiguration clientConfiguration = LettucePoolingClientConfiguration.builder().poolConfig(config).build();
    return new LettuceConnectionFactory(redisConfig, clientConfiguration);
  }
  @Bean
  @ConditionalOnProperty(value = "spring.redis.mode",havingValue = "cluster")
  public RedisClusterConfiguration redisClusterConfiguration() {
    RedisClusterConfiguration redisClusterConfiguration = new RedisClusterConfiguration(redisProperties.getCluster().getNodes());
    redisClusterConfiguration.setPassword(redisProperties.getPassword());
    return redisClusterConfiguration;
  }
  @Bean
  @ConditionalOnProperty(value = "spring.redis.mode",havingValue = "cluster")
  public RedisConnectionFactory redisConnectionFactory(RedisClusterConfiguration redisClusterConfiguration) {
    ClusterTopologyRefreshOptions clusterTopologyRefreshOptions = ClusterTopologyRefreshOptions.builder()
        .enablePeriodicRefresh()
        .enableAllAdaptiveRefreshTriggers()
        .refreshPeriod(Duration.ofSeconds(5))
        .build();
    ClusterClientOptions clusterClientOptions = ClusterClientOptions.builder()
        .topologyRefreshOptions(clusterTopologyRefreshOptions).build();
    LettuceClientConfiguration lettuceClientConfiguration = LettuceClientConfiguration.builder()
        .readFrom(ReadFrom.REPLICA_PREFERRED)
        .clientOptions(clusterClientOptions).build();
    return new LettuceConnectionFactory(redisClusterConfiguration, lettuceClientConfiguration);
  }
}