扩展Ribbon支持Nacos集群配置

1     
2     
3     
4     
5     
6     

spring:     
  cloud:     
    nacos:     
      discovery:     
        # 北京机房集群     
        cluster-name: BJ     

1     
2     
3     
4     
5     
6     
7     
8     
9     
10     
11     
12     
13     
14     
15     
16     
17     
18     
19     
20     
21     
22     
23     
24     
25     
26     
27     
28     
29     
30     
31     
32     
33     
34     
35     
36     
37     
38     
39     
40     
41     
42     
43     
44     
45     
46     
47     
48     
49     

/**     
 * 支持优先调用同集群实例的ribbon负载均衡规则.     
 *     
 * @author itmuch.com     
 */     
@Slf4j     
public class NacosRule extends AbstractLoadBalancerRule {     
    @Autowired     
    private NacosDiscoveryProperties nacosDiscoveryProperties;     
     
    @Override     
    public Server choose(Object key) {     
        try {     
            String clusterName = this.nacosDiscoveryProperties.getClusterName();     
            DynamicServerListLoadBalancer loadBalancer = (DynamicServerListLoadBalancer) getLoadBalancer();     
            String name = loadBalancer.getName();     
     
            NamingService namingService = this.nacosDiscoveryProperties.namingServiceInstance();     
     
            List<Instance> instances = namingService.selectInstances(name, true);     
            if (CollectionUtils.isEmpty(instances)) {     
                return null;     
            }     
     
            List<Instance> instancesToChoose = instances;     
            if (StringUtils.isNotBlank(clusterName)) {     
                List<Instance> sameClusterInstances = instances.stream()     
                        .filter(instance -> Objects.equals(clusterName, instance.getClusterName()))     
                        .collect(Collectors.toList());     
                if (!CollectionUtils.isEmpty(sameClusterInstances)) {     
                    instancesToChoose = sameClusterInstances;     
                } else {     
                    log.warn("发生跨集群的调用，name = {}, clusterName = {}, instance = {}", name, clusterName, instances);     
                }     
            }     
     
            Instance instance = ExtendBalancer.getHostByRandomWeight2(instancesToChoose);     
     
            return new NacosServer(instance);     
        } catch (Exception e) {     
            log.warn("NacosRule发生异常", e);     
            return null;     
        }     
    }     
     
    @Override     
    public void initWithNiwsConfig(IClientConfig iClientConfig) {     
    }     
}     

1     
2     
3     
4     
5     
6     
7     
8     
9     
10     
11     
12     

// Balancer来自于com.alibaba.nacos.client.naming.core.Balancer，也就是Nacos Client自带的基于权重的负载均衡算法。     
public class ExtendBalancer extends Balancer {     
    /**     
     * 根据权重，随机选择实例     
     *     
     * @param instances 实例列表     
     * @return 选择的实例     
     */     
    public static Instance getHostByRandomWeight2(List<Instance> instances) {     
        return getHostByRandomWeight(instances);     
    }     
}     

1     
2     
3     

microservice-provider-user:     
  ribbon:     
    NFLoadBalancerRuleClassName: com.itmuch.cloud.study.ribbon.NacosClusterAwareWeightedRule