Nacos客户端磁盘降级需求 issues:11053

client/pom.xml

@@ -119,6 +119,10 @@
  <groupId>org.yaml</groupId>
  <artifactId>snakeyaml</artifactId>
  </dependency>
+ <dependency>
+ <groupId>io.micrometer</groupId>
+ <artifactId>micrometer-core</artifactId>
+ </dependency>
  </dependencies>
 
  <build>

client/src/main/java/com/alibaba/nacos/client/naming/NacosNamingService.java

@@ -39,6 +39,7 @@
 import com.alibaba.nacos.client.utils.PreInitUtils;
 import com.alibaba.nacos.client.utils.ValidatorUtils;
 import com.alibaba.nacos.common.notify.NotifyCenter;
+import com.alibaba.nacos.common.utils.JacksonUtils;
 import com.alibaba.nacos.common.utils.StringUtils;
 
 import java.util.ArrayList;
@@ -47,6 +48,8 @@
 import java.util.Properties;
 import java.util.UUID;
 
+import static com.alibaba.nacos.client.utils.LogUtils.NAMING_LOGGER;
+
 /**
  * Nacos Naming Service.
  *
@@ -235,17 +238,8 @@ public List<Instance> getAllInstances(String serviceName, List<String> clusters,
  @Override
  public List<Instance> getAllInstances(String serviceName, String groupName, List<String> clusters,
  boolean subscribe) throws NacosException {
- ServiceInfo serviceInfo;
- String clusterString = StringUtils.join(clusters, ",");
- if (subscribe) {
- serviceInfo = serviceInfoHolder.getServiceInfo(serviceName, groupName, clusterString);
- if (null == serviceInfo || !clientProxy.isSubscribed(serviceName, groupName, clusterString)) {
- serviceInfo = clientProxy.subscribe(serviceName, groupName, clusterString);
- }
- } else {
- serviceInfo = clientProxy.queryInstancesOfService(serviceName, groupName, clusterString, false);
- }
  List<Instance> list;
+ ServiceInfo serviceInfo = getServiceInfo(serviceName, groupName, clusters, subscribe);
  if (serviceInfo == null || CollectionUtils.isEmpty(list = serviceInfo.getHosts())) {
  return new ArrayList<>();
  }
@@ -295,17 +289,7 @@ public List<Instance> selectInstances(String serviceName, List<String> clusters,
  @Override
  public List<Instance> selectInstances(String serviceName, String groupName, List<String> clusters, boolean healthy,
  boolean subscribe) throws NacosException {
-
- ServiceInfo serviceInfo;
- String clusterString = StringUtils.join(clusters, ",");
- if (subscribe) {
- serviceInfo = serviceInfoHolder.getServiceInfo(serviceName, groupName, clusterString);
- if (null == serviceInfo || !clientProxy.isSubscribed(serviceName, groupName, clusterString)) {
- serviceInfo = clientProxy.subscribe(serviceName, groupName, clusterString);
- }
- } else {
- serviceInfo = clientProxy.queryInstancesOfService(serviceName, groupName, clusterString, false);
- }
+ ServiceInfo serviceInfo = getServiceInfo(serviceName, groupName, clusters, subscribe);
  return selectInstances(serviceInfo, healthy);
  }
 
@@ -326,6 +310,41 @@ private List<Instance> selectInstances(ServiceInfo serviceInfo, boolean healthy)
  return list;
  }
 
+ private ServiceInfo getServiceInfoByFailover(String serviceName, String groupName, String clusterString) {
+ return serviceInfoHolder.getFailoverServiceInfo(serviceName, groupName, clusterString);
+ }
+
+ private ServiceInfo getServiceInfoBySubscribe(String serviceName, String groupName, String clusterString,
+ boolean subscribe) throws NacosException {
+ ServiceInfo serviceInfo;
+ if (subscribe) {
+ serviceInfo = serviceInfoHolder.getServiceInfo(serviceName, groupName, clusterString);
+ if (null == serviceInfo || !clientProxy.isSubscribed(serviceName, groupName, clusterString)) {
+ serviceInfo = clientProxy.subscribe(serviceName, groupName, clusterString);
+ }
+ } else {
+ serviceInfo = clientProxy.queryInstancesOfService(serviceName, groupName, clusterString, false);
+ }
+ return serviceInfo;
+ }
+
+ private ServiceInfo getServiceInfo(String serviceName, String groupName, List<String> clusters, boolean subscribe)
+ throws NacosException {
+ ServiceInfo serviceInfo;
+ String clusterString = StringUtils.join(clusters, ",");
+ if (serviceInfoHolder.isFailoverSwitch()) {
+ serviceInfo = getServiceInfoByFailover(serviceName, groupName, clusterString);
+ if (serviceInfo != null && serviceInfo.getHosts().size() > 0) {
+ NAMING_LOGGER.debug("getServiceInfo from failover,serviceName: {} data:{}", serviceName,
+ JacksonUtils.toJson(serviceInfo.getHosts()));
+ return serviceInfo;
+ }
+ }
+
+ serviceInfo = getServiceInfoBySubscribe(serviceName, groupName, clusterString, subscribe);
+ return serviceInfo;
+ }
+
  @Override
  public Instance selectOneHealthyInstance(String serviceName) throws NacosException {
  return selectOneHealthyInstance(serviceName, new ArrayList<>());
@@ -367,17 +386,8 @@ public Instance selectOneHealthyInstance(String serviceName, List<String> cluste
  @Override
  public Instance selectOneHealthyInstance(String serviceName, String groupName, List<String> clusters,
  boolean subscribe) throws NacosException {
- String clusterString = StringUtils.join(clusters, ",");
- if (subscribe) {
- ServiceInfo serviceInfo = serviceInfoHolder.getServiceInfo(serviceName, groupName, clusterString);
- if (null == serviceInfo || !clientProxy.isSubscribed(serviceName, groupName, clusterString)) {
- serviceInfo = clientProxy.subscribe(serviceName, groupName, clusterString);
- }
- return Balancer.RandomByWeight.selectHost(serviceInfo);
- } else {
- ServiceInfo serviceInfo = clientProxy.queryInstancesOfService(serviceName, groupName, clusterString, false);
- return Balancer.RandomByWeight.selectHost(serviceInfo);
- }
+ ServiceInfo serviceInfo = getServiceInfo(serviceName, groupName, clusters, subscribe);
+ return Balancer.RandomByWeight.selectHost(serviceInfo);
  }
 
  @Override

client/src/main/java/com/alibaba/nacos/client/naming/backups/FailoverData.java

@@ -0,0 +1,67 @@
+/*
+ * Copyright 1999-2018 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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.alibaba.nacos.client.naming.backups;
+
+/**
+ * Failover Data.
+ *
+ * @author zongkang.guo
+ */
+public class FailoverData {
+
+ /**
+ * failover type,naming or config.
+ */
+ private DataType dataType;
+
+ /**
+ * failover data.
+ */
+ private Object data;
+
+ public FailoverData(DataType dataType, Object data) {
+ this.data = data;
+ this.dataType = dataType;
+ }
+
+ public enum DataType {
+ /**
+ * naming.
+ */
+ naming,
+ /**
+ * config.
+ */
+ config
+ }
+
+ public DataType getDataType() {
+ return dataType;
+ }
+
+ public void setDataType(DataType dataType) {
+ this.dataType = dataType;
+ }
+
+ public Object getData() {
+ return data;
+ }
+
+ public void setData(Object data) {
+ this.data = data;
+ }
+}

client/src/main/java/com/alibaba/nacos/client/naming/backups/FailoverDataSource.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright 1999-2018 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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.alibaba.nacos.client.naming.backups;
+
+import java.util.Map;
+
+/**
+ * Failover Service Interface.
+ *
+ * @author Nacos
+ */
+public interface FailoverDataSource {
+
+ /**
+ * Get current disaster recovery switch.
+ *
+ * @return
+ */
+ FailoverSwitch getSwitch();
+
+
+ /**
+ * Get current disaster recovery data.
+ *
+ * @return map
+ */
+ Map<String, FailoverData> getFailoverData();
+
+}