study/Spring-Cloud-Alibaba
1
0
Fork 0
mirror of https://gitee.com/mirrors/Spring-Cloud-Alibaba.git synced 2021-06-26 13:25:11 +08:00
Code Issues Releases Wiki Activity

NacosRule code format

Browse Source
This commit is contained in:
fangjian0423 2019-07-03 16:55:16 +08:00
parent bbf7bdee52
commit 3998ea23f7
1 changed files with 55 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
spring-cloud-alibaba-nacos-discovery/src/main/java/org/springframework/cloud/alibaba/nacos/ribbon/NacosRule.java
View File

@ -1,11 +1,9 @@
package org.springframework.cloud.alibaba.nacos.ribbon; package org.springframework.cloud.alibaba.nacos.ribbon;
import com.alibaba.nacos.api.naming.NamingService; import java.util.List;
import com.alibaba.nacos.api.naming.pojo.Instance; import java.util.Objects;
import com.netflix.client.config.IClientConfig; import java.util.stream.Collectors;
import com.netflix.loadbalancer.AbstractLoadBalancerRule;
import com.netflix.loadbalancer.DynamicServerListLoadBalancer;
import com.netflix.loadbalancer.Server;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
@ -13,12 +11,17 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.alibaba.nacos.NacosDiscoveryProperties; import org.springframework.cloud.alibaba.nacos.NacosDiscoveryProperties;
import org.springframework.util.CollectionUtils; import org.springframework.util.CollectionUtils;
import java.util.List; import com.alibaba.nacos.api.naming.NamingService;
import java.util.Objects; import com.alibaba.nacos.api.naming.pojo.Instance;
import java.util.stream.Collectors;
import com.netflix.client.config.IClientConfig;
import com.netflix.loadbalancer.AbstractLoadBalancerRule;
import com.netflix.loadbalancer.DynamicServerListLoadBalancer;
import com.netflix.loadbalancer.Server;
/** /**
* 支持优先调用同集群实例的ribbon负载均衡规则. * Supports preferentially calling the ribbon load balancing rules of the same cluster
* instance.
* *
* @author itmuch.com * @author itmuch.com
*/ */
@ -35,30 +38,36 @@ public class NacosRule extends AbstractLoadBalancerRule {
DynamicServerListLoadBalancer loadBalancer = (DynamicServerListLoadBalancer) getLoadBalancer(); DynamicServerListLoadBalancer loadBalancer = (DynamicServerListLoadBalancer) getLoadBalancer();
String name = loadBalancer.getName(); String name = loadBalancer.getName();
NamingService namingService = this.nacosDiscoveryProperties.namingServiceInstance(); NamingService namingService = this.nacosDiscoveryProperties
.namingServiceInstance();
List<Instance> instances = namingService.selectInstances(name, true); List<Instance> instances = namingService.selectInstances(name, true);
if (CollectionUtils.isEmpty(instances)) { if (CollectionUtils.isEmpty(instances)) {
LOGGER.warn("{}服务当前无任何实例", name); LOGGER.warn("no instance in service {}", name);
return null; return null;
} }
List<Instance> instancesToChoose = instances; List<Instance> instancesToChoose = instances;
if (StringUtils.isNotBlank(clusterName)) { if (StringUtils.isNotBlank(clusterName)) {
List<Instance> sameClusterInstances = instances.stream() List<Instance> sameClusterInstances = instances.stream()
.filter(instance -> Objects.equals(clusterName, instance.getClusterName())) .filter(instance -> Objects.equals(clusterName,
instance.getClusterName()))
.collect(Collectors.toList()); .collect(Collectors.toList());
if (!CollectionUtils.isEmpty(sameClusterInstances)) { if (!CollectionUtils.isEmpty(sameClusterInstances)) {
instancesToChoose = sameClusterInstances; instancesToChoose = sameClusterInstances;
} else { }
LOGGER.warn("发生跨集群的调用，name = {}, clusterName = {}, instance = {}", name, clusterName, instances); else {
LOGGER.warn(
"A cross-cluster call occurs，name = {}, clusterName = {}, instance = {}",
name, clusterName, instances);
} }
} }
Instance instance = ExtendBalancer.getHostByRandomWeight2(instancesToChoose); Instance instance = ExtendBalancer.getHostByRandomWeight2(instancesToChoose);
return new NacosServer(instance); return new NacosServer(instance);
} catch (Exception e) { }
LOGGER.warn("NacosRule发生异常", e); catch (Exception e) {
LOGGER.warn("NacosRule error", e);
return null; return null;
} }
} }