Restore the old constructor to avoid incompatible issue

linkedin · Oct 17, 2024 · c239573 · c239573
1 parent 9510c64
commit c239573
Showing 1 changed file with 78 additions and 0 deletions.
diff --git a/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperConnectionManager.java b/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperConnectionManager.java
@@ -184,6 +184,75 @@ protected None convertResponse(None none) throws Exception
     }
   }
 
+  /**
+   * deprecated, use {@link  ConnectionManager#markDownAllServers(Callback)} instead.
+   */
+  @Deprecated
+  public void markDownAllServers(final Callback<None> callback)
+  {
+    Callback<None> markDownCallback;
+    if (callback != null)
+    {
+      markDownCallback = callback;
+    }
+    else
+    {
+      markDownCallback = new Callback<None>()
+      {
+        @Override
+        public void onError(Throwable e)
+        {
+          LOG.error("failed to mark down servers", e);
+        }
+
+        @Override
+        public void onSuccess(None result)
+        {
+          LOG.info("mark down all servers successful");
+        }
+      };
+    }
+    Callback<None> multiCallback = Callbacks.countDown(markDownCallback, _servers.length);
+    for (ZooKeeperAnnouncer server : _servers)
+    {
+      server.markDown(multiCallback);
+    }
+  }
+
+  /**
+   * deprecated, use {@link  ConnectionManager#markUpAllServers(Callback)} instead.
+   */
+  public void markUpAllServers(final Callback<None> callback)
+  {
+    Callback<None> markUpCallback;
+    if (callback != null)
+    {
+      markUpCallback = callback;
+    }
+    else
+    {
+      markUpCallback = new Callback<None>()
+      {
+        @Override
+        public void onError(Throwable e)
+        {
+          LOG.error("failed to mark up servers", e);
+        }
+
+        @Override
+        public void onSuccess(None result)
+        {
+          LOG.info("mark up all servers successful");
+        }
+      };
+    }
+    Callback<None> multiCallback = Callbacks.countDown(markUpCallback, _servers.length);
+    for (ZooKeeperAnnouncer server : _servers)
+    {
+      server.markUp(multiCallback);
+    }
+  }
+
   private class Listener implements ZKPersistentConnection.EventListener
   {
     @Override
@@ -295,6 +364,15 @@ public interface ZKStoreFactory<P, Z extends ZooKeeperStore<P>>
     Z createStore(ZKConnection connection, String path);
   }
 
+  /**
+   * @deprecated Use {@link #ConnectionManager#getAnnouncers()} instead.
+   */
+  @Deprecated
+  public ZooKeeperAnnouncer[] getAnnouncers()
+  {
+    return _servers;
+  }
+
   @Override
   public String getAnnouncementTargetIdentifier()
   {