文章目录
Pre需求工程Code继承AbstractLoadBalancerRule实现自定义Rule随机权重策略配置验证源码Pre
Spring Cloud Alibaba - 05 Nacos 领域模型_NameSpac/Group/Cluster
中提到了同一个Namespace下 + 同一个Group下， 不同Cluster 内的服务，可以互相访问.
如果我们要想实现一个 同集群优先权重负载均衡算法， 怎么办呢？
比如实现如下调用
需求
举个例子： 有两个微服务artisan-order-center, artisan-product-center。我们在北京机房部署一套artisan-order-center,artisan-product-center。为了容灾处理，我们在广东同样部署一套artisan-order-center,artisan-product-center。
但是 北京的artisan-order-center访问广东的artisan-product-center毕竟不如调用本地的artisan-product-center快。 如果本地的artisan-product-center挂掉了，那么再访问广东的artisan-product-center。
工程
接着 Spring Cloud Alibaba - 10 Ribbon 自定义负载均衡策略（权重算法） 中的工程，我们继续改造。
Code
继承AbstractLoadBalancerRule实现自定义Rule
package com.artisan.customrules;import com.alibaba.cloud.nacos.NacosDiscoveryProperties;import com.alibaba.cloud.nacos.ribbon.NacosServer;import com.alibaba.nacos.api.exception.NacosException;import com.alibaba.nacos.api.naming.NamingService;import com.alibaba.nacos.api.naming.pojo.Instance;import flix.client.config.IClientConfig;import flix.loadbalancer.AbstractLoadBalancerRule;import flix.loadbalancer.BaseLoadBalancer;import flix.loadbalancer.Server;import lombok.extern.slf4j.Slf4j;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.util.StringUtils;import java.util.ArrayList;import java.util.List;/*** @author 小工匠* @version 1.0* @description: 同一个集群优先调用策略* @date /2/3 0:47* @mark: show me the code , change the world*/@Slf4jpublic class SameClusterPriorityRule extends AbstractLoadBalancerRule {@Autowiredprivate NacosDiscoveryProperties discoveryProperties;@Overridepublic void initWithNiwsConfig(IClientConfig iClientConfig) {}@Overridepublic Server choose(Object key) {try {//第一步:获取当前服务所在的集群String currentClusterName = discoveryProperties.getClusterName();//第二步:获取一个负载均衡对象BaseLoadBalancer baseLoadBalancer = (BaseLoadBalancer) getLoadBalancer();//第三步:获取当前调用的微服务的名称String invokedSerivceName = baseLoadBalancer.getName();//第四步:获取nacos clinet的服务注册发现组件的apiNamingService namingService = discoveryProperties.namingServiceInstance();//第五步:获取所有的服务实例List<Instance> allInstance = namingService.getAllInstances(invokedSerivceName);List<Instance> theSameClusterNameInstList = new ArrayList<>();//第六步:过滤筛选同集群下的所有实例for (Instance instance : allInstance) {if (StringUtils.endsWithIgnoreCase(instance.getClusterName(), currentClusterName)) {theSameClusterNameInstList.add(instance);}}Instance toBeChooseInstance;//第七步:选择合适的一个实例调用if (theSameClusterNameInstList.isEmpty()) {toBeChooseInstance = ArtisanWeightedBalancer.chooseInstanceByRandomWeight(allInstance);log.info("发生跨集群调用--->当前微服务所在集群:{},被调用微服务所在集群:{},Host:{},Port:{}",currentClusterName, toBeChooseInstance.getClusterName(), toBeChooseInstance.getIp(), toBeChooseInstance.getPort());} else {toBeChooseInstance = ArtisanWeightedBalancer.chooseInstanceByRandomWeight(theSameClusterNameInstList);log.info("同集群调用--->当前微服务所在集群:{},被调用微服务所在集群:{},Host:{},Port:{}",currentClusterName, toBeChooseInstance.getClusterName(), toBeChooseInstance.getIp(), toBeChooseInstance.getPort());}return new NacosServer(toBeChooseInstance);} catch (NacosException e) {log.error("同集群优先权重负载均衡算法选择异常:{}", e);}return null;}}
随机权重策略
package com.artisan.customrules;import com.alibaba.nacos.api.naming.pojo.Instance;import com.alibaba.nacos.client.naming.core.Balancer;import java.util.List;/*** @author 小工匠* @version 1.0* @description: 根据权重选择随机选择一个* @date /2/3 0:28* @mark: show me the code , change the world*/public class ArtisanWeightedBalancer extends Balancer {public static Instance chooseInstanceByRandomWeight(List<Instance> hosts) {return getHostByRandomWeight(hosts);}}
配置
artisan-cloud-customcfg-ribbon-order 中的 nacos配置
spring:cloud:nacos:discovery:server-addr: 1.117.97.88:8848cluster-name: BeiJingClusterapplication:name: artisan-order-center
artisan-cloud-customcfg-ribbon-product 中的 nacos配置
spring: cloud:nacos:discovery:server-addr: 1.117.97.88:8848cluster-name: GuangDongClusterapplication:name: artisan-product-center
验证
artisan-cloud-customcfg-ribbon-order---- cluster-name: BeiJingCluster artisan-cloud-customcfg-ribbon-product ----cluster-name: BeiJingCluster artisan-cloud-customcfg-ribbon-product ----cluster-name: GuangDongCluster
启动一个artisan-cloud-customcfg-ribbon-order和两个 artisan-cloud-customcfg-ribbon-product 工程后，
通过artisan-cloud-customcfg-ribbon-order 【BeiJingCluster 】访问artisan-cloud-customcfg-ribbon-product
查看artisan-cloud-customcfg-ribbon-order日志
我们下线artisan-cloud-customcfg-ribbon-order 【BeiJingCluster 】
再次调用 ，观察日志
这样就实现了刚才的需求
源码
/yangshangwei/SpringCloudAlibabMaster
