Submit a feature that applies graceful up and down

core/src/main/java/org/apache/servicecomb/core/SCBEngine.java

@@ -371,7 +371,7 @@ private void doRun() throws Exception {
     registrationManager.run();
     discoveryManager.run();
     // ensure can invoke services in AFTER_REGISTRY
-    registrationManager.updateMicroserviceInstanceStatus(MicroserviceInstanceStatus.UP);
+//    registrationManager.updateMicroserviceInstanceStatus(MicroserviceInstanceStatus.UP);
     status = SCBStatus.UP;
     triggerEvent(EventType.AFTER_REGISTRY);
 

foundations/foundation-config/src/main/java/org/apache/servicecomb/config/DataCenterProperties.java

@@ -19,11 +19,11 @@
 public class DataCenterProperties {
   public static final String PREFIX = "servicecomb.datacenter";
 
-  private String name;
+  private String name = "local";
 
-  private String region;
+  private String region = "local";
 
-  private String availableZone;
+  private String availableZone = "local";
 
   public String getName() {
     return name;

foundations/foundation-registry/src/main/java/org/apache/servicecomb/registry/DiscoveryManager.java

@@ -178,7 +178,8 @@ private VersionedCache calcAvailableInstance(String application, String serviceN
         -> new ConcurrentHashMapEx<>());
     List<StatefulDiscoveryInstance> result = new ArrayList<>();
     for (StatefulDiscoveryInstance instance : statefulInstances.values()) {
-      if (instance.getHistoryStatus() == HistoryStatus.CURRENT) {
+      if (instance.getHistoryStatus() == HistoryStatus.CURRENT
+              && instance.getMicroserviceInstanceStatus() == MicroserviceInstanceStatus.UP) {
         result.add(instance);
         continue;
       }

foundations/foundation-registry/src/main/java/org/apache/servicecomb/registry/api/MicroserviceInstance.java

@@ -93,4 +93,10 @@ public interface MicroserviceInstance {
   default String getServiceId() {
     return "";
   }
+
+  /**
+   * Service status(Required): status of this microservice.
+   * @return
+   */
+  public MicroserviceInstanceStatus getStatus();
 }

foundations/foundation-registry/src/main/java/org/apache/servicecomb/registry/discovery/StatefulDiscoveryInstance.java

@@ -62,6 +62,7 @@ public enum HistoryStatus {
 
   public StatefulDiscoveryInstance(DiscoveryInstance discoveryInstance) {
     this.discoveryInstance = discoveryInstance;
+    this.microserviceInstanceStatus = discoveryInstance.getStatus();
   }
 
   public MicroserviceInstanceStatus getMicroserviceInstanceStatus() {

service-registry/registry-etcd/src/main/java/org/apache/servicecomb/registry/etcd/EtcdInstance.java

@@ -23,6 +23,7 @@
 
 import org.apache.servicecomb.registry.api.DataCenterInfo;
 import org.apache.servicecomb.registry.api.MicroserviceInstance;
+import org.apache.servicecomb.registry.api.MicroserviceInstanceStatus;
 
 public class EtcdInstance implements MicroserviceInstance {
   private String serviceId;
@@ -188,6 +189,11 @@ public String getServiceId() {
     return serviceId;
   }
 
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return MicroserviceInstanceStatus.UP;
+  }
+
   @Override
   public String toString() {
     return "EtcdInstance{" +

service-registry/registry-local/src/main/java/org/apache/servicecomb/localregistry/LocalRegistrationInstance.java

@@ -118,6 +118,11 @@ public String getInstanceId() {
     return this.instanceId;
   }
 
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return MicroserviceInstanceStatus.UP;
+  }
+
   @Override
   public MicroserviceInstanceStatus getInitialStatus() {
     return MicroserviceInstanceStatus.STARTING;

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosConst.java

@@ -57,4 +57,6 @@ public class NacosConst {
   public static final String NAMING_LOAD_CACHE_AT_START = "namingLoadCacheAtStart";
 
   public static final String NACOS_REGISTRY_NAME = "nacos-registry";
+
+  public static final String NACOS_STATUS= "nacos-status";
 }

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosDiscoveryInstance.java

@@ -57,7 +57,11 @@ public NacosDiscoveryInstance(Instance instance, String application, String serv
 
   @Override
   public MicroserviceInstanceStatus getStatus() {
-    return instance.isEnabled() ? MicroserviceInstanceStatus.UP : MicroserviceInstanceStatus.DOWN;
+    if (instance.isEnabled()){
+      return MicroserviceInstanceStatus.valueOf(instance.getMetadata().get(NacosConst.NACOS_STATUS));
+    } else {
+      return  MicroserviceInstanceStatus.DOWN;
+    }
   }
 
   @Override

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosDiscoveryProperties.java

@@ -49,6 +49,8 @@ public class NacosDiscoveryProperties {
 
   private boolean enableSwaggerRegistration = false;
 
+  private String initialStatus = "UP";
+
   public String getServerAddr() {
     return serverAddr;
   }
@@ -160,4 +162,12 @@ public boolean isEnableSwaggerRegistration() {
   public void setEnableSwaggerRegistration(boolean enableSwaggerRegistration) {
     this.enableSwaggerRegistration = enableSwaggerRegistration;
   }
+
+  public String getInitialStatus() {
+    return initialStatus;
+  }
+
+  public void setInitialStatus(String initialStatus) {
+    this.initialStatus = initialStatus;
+  }
 }

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosMicroserviceHandler.java

@@ -63,6 +63,7 @@ public static Instance createMicroserviceInstance(
     metadata.put(NacosConst.PROPERTY_DATACENTER, dataCenterProperties.getName());
     metadata.put(NacosConst.PROPERTY_REGION, dataCenterProperties.getRegion());
     metadata.put(NacosConst.PROPERTY_ZONE, dataCenterProperties.getAvailableZone());
+    metadata.put(NacosConst.NACOS_STATUS, properties.getInitialStatus());
     if (!StringUtils.isEmpty(environment.getProperty(VERSION_MAPPING)) &&
         !StringUtils.isEmpty(environment.getProperty(environment.getProperty(VERSION_MAPPING)))) {
       metadata.put("version", environment.getProperty(environment.getProperty(VERSION_MAPPING)));

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosRegistration.java

@@ -132,9 +132,21 @@ public NacosRegistrationInstance getMicroserviceInstance() {
 
   @Override
   public boolean updateMicroserviceInstanceStatus(MicroserviceInstanceStatus status) {
-    // Do not support Nacos update status now. Because update status will fail
-    // due to some unknown reasons(Maybe different constrains in register and maintain api).
-    return true;
+    try {
+      if (MicroserviceInstanceStatus.UP == status){
+        instance.getMetadata().put(NacosConst.NACOS_STATUS, MicroserviceInstanceStatus.UP.name());
+        namingService.registerInstance(nacosRegistrationInstance.getServiceName(),
+                nacosRegistrationInstance.getApplication(), instance);
+      }
+      if (MicroserviceInstanceStatus.DOWN == status){
+        instance.getMetadata().put(NacosConst.NACOS_STATUS, MicroserviceInstanceStatus.DOWN.name());
+        namingService.deregisterInstance(nacosRegistrationInstance.getServiceName(),
+                nacosRegistrationInstance.getApplication(), instance);
+      }
+      return true;
+    } catch (Exception e) {
+      throw new RuntimeException(e);
+    }
   }
 
   @Override

service-registry/registry-nacos/src/main/java/org/apache/servicecomb/registry/nacos/NacosRegistrationInstance.java

@@ -103,6 +103,11 @@ public String getInstanceId() {
     return instance.getInstanceId();
   }
 
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return MicroserviceInstanceStatus.valueOf(instance.getMetadata().get(NacosConst.NACOS_STATUS));
+  }
+
   @Override
   public MicroserviceInstanceStatus getInitialStatus() {
     return MicroserviceInstanceStatus.STARTING;

service-registry/registry-service-center/src/main/java/org/apache/servicecomb/registry/sc/MicroserviceHandler.java

@@ -113,13 +113,13 @@ public static MicroserviceInstance createMicroserviceInstance(
     }
     microserviceInstance.setHostName(hostName);
 
+    DataCenterInfo dataCenterInfo = new DataCenterInfo();
     if (StringUtils.isNotEmpty(dataCenterProperties.getName())) {
-      DataCenterInfo dataCenterInfo = new DataCenterInfo();
       dataCenterInfo.setName(dataCenterProperties.getName());
       dataCenterInfo.setRegion(dataCenterProperties.getRegion());
       dataCenterInfo.setAvailableZone(dataCenterProperties.getAvailableZone());
-      microserviceInstance.setDataCenterInfo(dataCenterInfo);
     }
+    microserviceInstance.setDataCenterInfo(dataCenterInfo);
 
     HealthCheck healthCheck = new HealthCheck();
     healthCheck.setMode(HealthCheckMode.push);

service-registry/registry-service-center/src/main/java/org/apache/servicecomb/registry/sc/SCConfigurationProperties.java

@@ -49,7 +49,7 @@ public class SCConfigurationProperties {
 
   private boolean autoDiscovery = false;
 
-  private String initialStatus = "STARTING";
+  private String initialStatus = "UP";
 
   private boolean watch = false;
 

service-registry/registry-service-center/src/main/java/org/apache/servicecomb/registry/sc/SCDiscoveryInstance.java

@@ -76,9 +76,9 @@ public String getVersion() {
 
   @Override
   public DataCenterInfo getDataCenterInfo() {
-    return new DataCenterInfo(microserviceInstance.getDataCenterInfo().getName(),
-        microserviceInstance.getDataCenterInfo().getRegion(),
-        microserviceInstance.getDataCenterInfo().getAvailableZone());
+      return new DataCenterInfo(microserviceInstance.getDataCenterInfo().getName(),
+              microserviceInstance.getDataCenterInfo().getRegion(),
+              microserviceInstance.getDataCenterInfo().getAvailableZone());
   }
 
   @Override

service-registry/registry-service-center/src/main/java/org/apache/servicecomb/registry/sc/SCRegistration.java

@@ -195,4 +195,15 @@ public Microservice getBackendMicroservice() {
   public MicroserviceInstance getBackendMicroserviceInstance() {
     return microserviceInstance;
   }
+
+  /**
+   * Returns the latest instance for checking when updating the instance status from STARTING to UP.
+   * @return MicroserviceInstance
+   */
+  public MicroserviceInstance getLatestMicroserviceInstance() {
+    MicroserviceInstance latestInstance = serviceCenterClient.getMicroserviceInstance(microserviceInstance.getServiceId(),
+            microserviceInstance.getInstanceId());
+    microserviceInstance.setStatus(latestInstance.getStatus());
+    return latestInstance;
+  }
 }

service-registry/registry-service-center/src/main/java/org/apache/servicecomb/registry/sc/SCRegistrationInstance.java

@@ -34,8 +34,7 @@ public class SCRegistrationInstance implements RegistrationInstance {
 
   private final ServiceCenterRegistration serviceCenterRegistration;
 
-  public SCRegistrationInstance(Microservice microservice,
-      MicroserviceInstance microserviceInstance,
+  public SCRegistrationInstance(Microservice microservice, MicroserviceInstance microserviceInstance,
       ServiceCenterRegistration serviceCenterRegistration) {
     this.microservice = microservice;
     this.microserviceInstance = microserviceInstance;
@@ -106,6 +105,11 @@ public String getServiceId() {
     return microservice.getServiceId();
   }
 
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return MicroserviceInstanceStatus.valueOf(microserviceInstance.getStatus().name());
+  }
+
   @Override
   public MicroserviceInstanceStatus getInitialStatus() {
     return MicroserviceInstanceStatus.valueOf(microserviceInstance.getStatus().name());

service-registry/registry-zero-config/src/main/java/org/apache/servicecomb/zeroconfig/ZeroConfigRegistrationInstance.java

@@ -86,6 +86,11 @@ public String getInstanceId() {
     return this.self.getInstanceId();
   }
 
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return MicroserviceInstanceStatus.UP;
+  }
+
   @Override
   public MicroserviceInstanceStatus getInitialStatus() {
     return MicroserviceInstanceStatus.STARTING;

service-registry/registry-zookeeper/src/main/java/org/apache/servicecomb/registry/zookeeper/ZookeeperDiscoveryInstance.java

@@ -24,11 +24,6 @@ public ZookeeperDiscoveryInstance(ZookeeperInstance other) {
     super(other);
   }
 
-  @Override
-  public MicroserviceInstanceStatus getStatus() {
-    return MicroserviceInstanceStatus.UP;
-  }
-
   @Override
   public String getRegistryName() {
     return ZookeeperConst.ZOOKEEPER_REGISTRY_NAME;

service-registry/registry-zookeeper/src/main/java/org/apache/servicecomb/registry/zookeeper/ZookeeperInstance.java

@@ -23,6 +23,7 @@
 
 import org.apache.servicecomb.registry.api.DataCenterInfo;
 import org.apache.servicecomb.registry.api.MicroserviceInstance;
+import org.apache.servicecomb.registry.api.MicroserviceInstanceStatus;
 
 public class ZookeeperInstance implements MicroserviceInstance {
   private String serviceId;
@@ -49,6 +50,8 @@ public class ZookeeperInstance implements MicroserviceInstance {
 
   private Map<String, String> properties = new HashMap<>();
 
+  private MicroserviceInstanceStatus status;
+
   public ZookeeperInstance() {
 
   }
@@ -66,6 +69,8 @@ public ZookeeperInstance(ZookeeperInstance other) {
     this.endpoints = other.endpoints;
     this.schemas = other.schemas;
     this.properties = other.properties;
+    this.status = other.status;
+
   }
 
   public void setServiceId(String serviceId) {
@@ -116,6 +121,10 @@ public void setProperties(Map<String, String> properties) {
     this.properties = properties;
   }
 
+  public void setStatus(MicroserviceInstanceStatus status) {
+    this.status = status;
+  }
+
   @Override
   public String getEnvironment() {
     return this.environment;
@@ -187,4 +196,9 @@ public String getInstanceId() {
   public String getServiceId() {
     return serviceId;
   }
+
+  @Override
+  public MicroserviceInstanceStatus getStatus() {
+    return this.status;
+  }
 }

service-registry/registry-zookeeper/src/main/java/org/apache/servicecomb/registry/zookeeper/ZookeeperRegistration.java

@@ -79,6 +79,8 @@ public AppConfigurationEntry[] getAppConfigurationEntry(String name) {
 
   private RegistrationId registrationId;
 
+  private ServiceDiscovery<ZookeeperInstance> dis;
+
   @Autowired
   @SuppressWarnings("unused")
   public void setEnvironment(Environment environment) {
@@ -125,6 +127,8 @@ public void init() {
     }
     zookeeperInstance.setProperties(BootStrapProperties.readServiceProperties(environment));
     zookeeperInstance.setVersion(BootStrapProperties.readServiceVersion(environment));
+
+    zookeeperInstance.setStatus(MicroserviceInstanceStatus.valueOf(zookeeperRegistryProperties.getInitialStatus()));
     try {
       this.instance = ServiceInstance.<ZookeeperInstance>builder().name(zookeeperInstance.getServiceName())
           .id(zookeeperInstance.getInstanceId()).payload(zookeeperInstance).build();
@@ -155,7 +159,7 @@ public void run() {
     client.start();
     JsonInstanceSerializer<ZookeeperInstance> serializer =
         new JsonInstanceSerializer<>(ZookeeperInstance.class);
-    ServiceDiscovery<ZookeeperInstance> dis = ServiceDiscoveryBuilder.builder(ZookeeperInstance.class)
+    dis = ServiceDiscoveryBuilder.builder(ZookeeperInstance.class)
         .client(client)
         .basePath(basePath + "/" + BootStrapProperties.readApplication(environment))
         .serializer(serializer)
@@ -187,8 +191,13 @@ public ZookeeperRegistrationInstance getMicroserviceInstance() {
 
   @Override
   public boolean updateMicroserviceInstanceStatus(MicroserviceInstanceStatus status) {
-    // not support yet
-    return true;
+      this.instance.getPayload().setStatus(status);
+      try {
+        dis.updateService(instance);
+      } catch (Exception e){
+          throw new IllegalStateException(e);
+      }
+      return true;
   }
 
   @Override
@@ -212,4 +221,21 @@ public void addProperty(String key, String value) {
   public boolean enabled() {
     return zookeeperRegistryProperties.isEnabled();
   }
+
+  /**
+   * Returns the latest instance for checking when updating the instance status from STARTING to UP.
+   * @return ZookeeperInstance
+   */
+  public ZookeeperInstance getZookeeperInstance(){
+    try {
+      ServiceInstance<ZookeeperInstance> zkInstance = dis.queryForInstance(instance.getName(), instance.getId());
+      if (zkInstance != null){
+        return zkInstance.getPayload();
+      }
+    } catch (Exception e) {
+      throw new RuntimeException(e);
+    }
+    return null;
+  }
+
 }