com.github.dts.sdk.client.DiscoveryService Maven / Gradle / Ivy

Go to download
package com.github.dts.sdk.client;

import com.github.dts.sdk.conf.DtsSdkConfig;
import com.github.dts.sdk.util.PlatformDependentUtil;
import com.github.dts.sdk.util.ReferenceCounted;
import com.github.dts.sdk.util.SpringUtil;
import com.github.dts.sdk.util.Util;
import org.springframework.beans.factory.ListableBeanFactory;
import org.springframework.core.env.Environment;

import java.util.List;
import java.util.Objects;

public interface DiscoveryService {

    static DiscoveryService newInstance(DtsSdkConfig.ClusterConfig config,
                                        ListableBeanFactory beanFactory) {
        DtsSdkConfig.DiscoveryEnum discoveryEnum = config.getDiscovery();
        if (discoveryEnum == DtsSdkConfig.DiscoveryEnum.YAML) {
            return new YamlDiscoveryService(config);
        }
        if (discoveryEnum == DtsSdkConfig.DiscoveryEnum.AUTO) {
            if (!Objects.toString(config.getNacos().getServerAddr(), "").isEmpty()) {
                discoveryEnum = DtsSdkConfig.DiscoveryEnum.NACOS;
            } else if (PlatformDependentUtil.isSupportSpringframeworkRedis() && beanFactory.getBeanNamesForType(PlatformDependentUtil.REDIS_CONNECTION_FACTORY_CLASS).length > 0) {
                discoveryEnum = DtsSdkConfig.DiscoveryEnum.REDIS;
            }
        }

        Environment env = beanFactory.getBean(Environment.class);
        String ip = Util.getIPAddress();
        Integer port = env.getProperty("server.port", Integer.class, 8080);
        switch (discoveryEnum) {
            case REDIS: {
                DtsSdkConfig.ClusterConfig.Redis redis = config.getRedis();
                String redisKeyRootPrefix = redis.getRedisKeyRootPrefix();
                if (redisKeyRootPrefix != null) {
                    redisKeyRootPrefix = env.resolvePlaceholders(redisKeyRootPrefix);
                }
                Object redisConnectionFactory = SpringUtil.getBean(beanFactory, redis.getRedisConnectionFactoryBeanName(), PlatformDependentUtil.REDIS_CONNECTION_FACTORY_CLASS);
                return new RedisDiscoveryService(
                        redisConnectionFactory,
                        redisKeyRootPrefix,
                        redis.getRedisInstanceExpireSec(),
                        config, ip, port);
            }
            case NACOS:
            default: {
                throw new IllegalArgumentException("ServiceDiscoveryService newInstance fail! remote discovery config is empty!");
            }
        }
    }

    void registerSdkInstance();

     ReferenceCounted> getServerListRef();

    void addServerListener(ServerListener serverListener);

    interface ServerListener {
         void onChange(ServerChangeEvent event);
    }

    class ServerChangeEvent {
        public final List insertList;
        public final List deleteList;
        // 首次=0，从0开始计数
        public int updateInstanceCount;

        public ServerChangeEvent(int updateInstanceCount,
                                 List insertList, List deleteList) {
            this.updateInstanceCount = updateInstanceCount;
            this.insertList = insertList;
            this.deleteList = deleteList;
        }

    }

}