com.tencent.cloud.polaris.loadbalancer.config.PolarisRibbonClientConfiguration Maven / Gradle / Ivy

Go to download
/*
 * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * Unless required by applicable law or agreed to in writing, software distributed
 * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 *
 */

package com.tencent.cloud.polaris.loadbalancer.config;

import com.netflix.client.config.IClientConfig;
import com.netflix.loadbalancer.ILoadBalancer;
import com.netflix.loadbalancer.IPing;
import com.netflix.loadbalancer.IRule;
import com.netflix.loadbalancer.RandomRule;
import com.netflix.loadbalancer.RoundRobinRule;
import com.netflix.loadbalancer.Server;
import com.netflix.loadbalancer.ServerList;
import com.tencent.cloud.polaris.context.PolarisSDKContextManager;
import com.tencent.cloud.polaris.loadbalancer.PolarisLoadBalancer;
import com.tencent.cloud.polaris.loadbalancer.PolarisRingHashRule;
import com.tencent.cloud.polaris.loadbalancer.PolarisWeightedRandomRule;
import com.tencent.cloud.polaris.loadbalancer.PolarisWeightedRoundRobinRule;
import com.tencent.cloud.polaris.loadbalancer.transformer.InstanceTransformer;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;

/**
 * Configuration of ribbon client of Polaris.
 *
 * @author Haotian Zhang
 */
public class PolarisRibbonClientConfiguration {

	@Bean
	public ILoadBalancer polarisLoadBalancer(IClientConfig iClientConfig, IRule iRule,
			IPing iPing, ServerList serverList,
			PolarisSDKContextManager polarisSDKContextManager, PolarisLoadBalancerProperties polarisLoadBalancerProperties,
			@Autowired(required = false) InstanceTransformer instanceTransformer) {
		return new PolarisLoadBalancer(iClientConfig, iRule, iPing, serverList,
				polarisSDKContextManager.getConsumerAPI(), polarisLoadBalancerProperties, instanceTransformer);
	}

	@Bean
	@ConditionalOnMissingBean
	@ConditionalOnProperty(value = "spring.cloud.polaris.loadbalancer.strategy", havingValue = "roundRobin", matchIfMissing = true)
	public IRule roundRobinRule() {
		return new RoundRobinRule();
	}

	@Bean
	@ConditionalOnMissingBean
	@ConditionalOnProperty(value = "spring.cloud.polaris.loadbalancer.strategy", havingValue = "random")
	public IRule randomRule() {
		return new RandomRule();
	}

	@Bean
	@ConditionalOnMissingBean
	@ConditionalOnProperty(value = "spring.cloud.polaris.loadbalancer.strategy", havingValue = "polarisWeightedRandom")
	public IRule polarisWeightedRandomRule(PolarisSDKContextManager polarisSDKContextManager) {
		return new PolarisWeightedRandomRule(polarisSDKContextManager.getRouterAPI());
	}

	@Bean
	@ConditionalOnMissingBean
	@ConditionalOnProperty(value = "spring.cloud.polaris.loadbalancer.strategy", havingValue = "polarisWeightedRoundRobin")
	public IRule polarisWeightedRoundRobinRule(PolarisSDKContextManager polarisSDKContextManager) {
		return new PolarisWeightedRoundRobinRule(polarisSDKContextManager.getRouterAPI());
	}

	@Bean
	@ConditionalOnMissingBean
	@ConditionalOnProperty(value = "spring.cloud.polaris.loadbalancer.strategy", havingValue = "polarisRingHash")
	public IRule polarisRingHashRule(PolarisSDKContextManager polarisSDKContextManager) {
		return new PolarisRingHashRule(polarisSDKContextManager.getRouterAPI());
	}
}