setup cluster for ServiceDiscoveryDemo (#2812)

setup cluster for ServiceDiscoveryDemo
apache · Jul 11, 2024 · c0c64a4 · c0c64a4
1 parent bf20e2a
commit c0c64a4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...rvice-discovery/src/main/java/org/apache/helix/servicediscovery/ServiceDiscoveryDemo.java b/...rvice-discovery/src/main/java/org/apache/helix/servicediscovery/ServiceDiscoveryDemo.java
@@ -24,6 +24,7 @@
 import java.util.ArrayList;
 import java.util.List;
 
+import org.apache.helix.manager.zk.ZKHelixAdmin;
 import org.apache.helix.servicediscovery.ServiceDiscovery.Mode;
 
 public class ServiceDiscoveryDemo {
@@ -34,13 +35,19 @@ public static void main(String[] args) throws Exception {
     String serviceName = "myServiceName";
     int numServices = 5;
 
+    ZKHelixAdmin admin = new ZKHelixAdmin(zkAddress);
+    // create cluster
+    System.out.println("Creating cluster: " + clusterName);
+    admin.addCluster(clusterName, true);
+
     // registration + zk watch
     demo(clusterName, zkAddress, serviceName, numServices, Mode.WATCH);
     // registration + periodic poll
     demo(clusterName, zkAddress, serviceName, numServices, Mode.POLL);
     // only registration + ondemand
     demo(clusterName, zkAddress, serviceName, numServices, Mode.NONE);
 
+    admin.close();
   }
 
   private static void demo(String clusterName, String zkAddress, String serviceName,