com.netease.nim.camellia.spring.redis.zk.discovery.SpringRedisZkDiscoveryConfiguration Maven / Gradle / Ivy

package com.netease.nim.camellia.spring.redis.zk.discovery;

import com.netease.nim.camellia.redis.proxy.discovery.jedis.RedisProxyJedisPool;
import com.netease.nim.camellia.redis.proxy.discovery.common.RegionResolver;
import com.netease.nim.camellia.redis.proxy.discovery.zk.ZkProxyDiscovery;
import com.netease.nim.camellia.spring.redis.base.RedisProxyRedisConnectionFactory;
import com.netease.nim.camellia.zk.ZkClientFactory;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import redis.clients.jedis.JedisPoolConfig;

/**
 *
 * Created by caojiajun on 2020/12/2
 */
@Configuration
@EnableConfigurationProperties({SpringRedisZkDiscoveryProperties.class})
@ConditionalOnExpression("${camellia-spring-redis-zk-discovery.enable:true}")
public class SpringRedisZkDiscoveryConfiguration {
    private static final Logger logger = LoggerFactory.getLogger(SpringRedisZkDiscoveryConfiguration.class);

    @Bean
    @ConditionalOnMissingBean(ZkClientFactory.class)
    public ZkClientFactory zkClientFactory(SpringRedisZkDiscoveryProperties properties) {
        SpringRedisZkDiscoveryProperties.ZkConf zk = properties.getZkConf();
        return new ZkClientFactory(zk.getSessionTimeoutMs(),
                zk.getConnectionTimeoutMs(), zk.getBaseSleepTimeMs(), zk.getMaxRetries());
    }

    @Bean
    @ConditionalOnMissingBean(RedisProxyJedisPool.class)
    public RedisProxyJedisPool redisProxyJedisPool(SpringRedisZkDiscoveryProperties properties) {
        SpringRedisZkDiscoveryProperties.ZkConf zkConf = properties.getZkConf();
        ZkProxyDiscovery zkProxyDiscovery = new ZkProxyDiscovery(zkClientFactory(properties),
                zkConf.getZkUrl(), zkConf.getBasePath(), properties.getApplicationName(), zkConf.getReloadIntervalSeconds());

        boolean sideCarFirst = zkConf.isSideCarFirst();
        String regionResolveConf = zkConf.getRegionResolveConf();
        String defaultRegion = zkConf.getDefaultRegion();
        RegionResolver regionResolver = new RegionResolver.IpSegmentRegionResolver(regionResolveConf, defaultRegion);

        SpringRedisZkDiscoveryProperties.RedisConf redisConf = properties.getRedisConf();
        GenericObjectPoolConfig poolConfig = new JedisPoolConfig();
        poolConfig.setMaxIdle(redisConf.getMaxIdle());
        poolConfig.setMinIdle(redisConf.getMinIdle());
        poolConfig.setMaxTotal(redisConf.getMaxActive());
        poolConfig.setMaxWaitMillis(redisConf.getMaxWaitMillis());
        Long bid = properties.getBid();
        String bgroup = properties.getBgroup();
        if (bid == null || bid < 0 || bgroup == null) {
            return new RedisProxyJedisPool.Builder()
                    .proxyDiscovery(zkProxyDiscovery)
                    .poolConfig(poolConfig)
                    .timeout(redisConf.getTimeout())
                    .password(properties.getPassword())
                    .sideCarFirst(sideCarFirst)
                    .regionResolver(regionResolver)
                    .jedisPoolLazyInit(zkConf.isJedisPoolLazyInit())
                    .jedisPoolInitialSize(zkConf.getJedisPoolInitialSize())
                    .build();
        } else {
            return new RedisProxyJedisPool.Builder()
                    .bid(bid)
                    .bgroup(bgroup)
                    .proxyDiscovery(zkProxyDiscovery)
                    .poolConfig(poolConfig)
                    .timeout(redisConf.getTimeout())
                    .password(properties.getPassword())
                    .sideCarFirst(sideCarFirst)
                    .regionResolver(regionResolver)
                    .jedisPoolLazyInit(zkConf.isJedisPoolLazyInit())
                    .jedisPoolInitialSize(zkConf.getJedisPoolInitialSize())
                    .build();
        }
    }

    @Bean
    public RedisProxyRedisConnectionFactory redisProxyRedisConnectionFactory(SpringRedisZkDiscoveryProperties properties) {
        RedisProxyRedisConnectionFactory redisProxyRedisConnectionFactory = new RedisProxyRedisConnectionFactory(redisProxyJedisPool(properties));
        logger.info("RedisProxyRedisConnectionFactory init success");
        return redisProxyRedisConnectionFactory;
    }
}