org.swiftboot.service.SwiftbootServiceConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swiftboot-service Show documentation
The newest version!
package org.swiftboot.service;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.swiftboot.service.config.SwiftbootServiceConfigBean;
import org.swiftboot.service.service.CaptchaService;
import org.swiftboot.service.service.RedisService;
import org.swiftboot.service.service.impl.CaptchaServiceImpl;
import org.swiftboot.service.service.impl.RedisClusterServiceImpl;
import org.swiftboot.service.service.impl.RedisServiceImpl;
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.JedisCluster;

import javax.annotation.Resource;
import java.util.HashSet;
import java.util.Set;

/**
 * @author swiftech 2019-05-24
 **/
@Configuration
@EnableConfigurationProperties
public class SwiftbootServiceConfig {

    private static final Logger log = LoggerFactory.getLogger(SwiftbootServiceConfig.class);

    @Resource
    SwiftbootServiceConfigBean swiftbootServiceConfigBean;

    @Bean
    @ConditionalOnProperty("swiftboot.service.redis.host")
    public RedisService redisService() {
        log.info("default redis service implementation");
        return new RedisServiceImpl();
    }

    @Bean
    @ConditionalOnProperty("swiftboot.service.redis.cluster")
    public JedisCluster redisCluster() {
        log.info("clustered redis service implementation");
        SwiftbootServiceConfigBean config = swiftbootServiceConfigBean;
        String[] serverArray = config.getRedis().getCluster().split(",");
        Set nodes = new HashSet<>();
        for (String ipPort : serverArray) {
            String[] ipPortPair = ipPort.split(":");
            nodes.add(new HostAndPort(ipPortPair[0].trim(), Integer.parseInt(ipPortPair[1].trim())));
        }
        return new JedisCluster(nodes);
    }

    /**
     * 依赖于配置了 JedisCluster
     *
     * @return
     */
    @Bean
    @ConditionalOnBean(JedisCluster.class)
    public RedisClusterServiceImpl redisClusterService() {
        return new RedisClusterServiceImpl();
    }

    /**
     * Captcha 服务，依赖于 RedisService
     *
     * @return
     */
    @Bean
    @ConditionalOnProperty("swiftboot.captcha")
    @ConditionalOnBean(RedisService.class)
    public CaptchaService captchaService() {
        return new CaptchaServiceImpl();
    }

}