add setters for executors

linkedin · Dec 22, 2023 · 328d1b4 · 328d1b4
1 parent f5a0adb
commit 328d1b4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/d2/src/main/java/com/linkedin/d2/balancer/D2ClientBuilder.java b/d2/src/main/java/com/linkedin/d2/balancer/D2ClientBuilder.java
@@ -541,6 +541,18 @@ public D2ClientBuilder setWarmUpConcurrentRequests(int warmUpConcurrentRequests)
     return this;
   }
 
+  public D2ClientBuilder setStartUpExecutorService(ScheduledExecutorService executorService)
+  {
+    _config.startUpExecutorService = executorService;
+    return this;
+  }
+
+  public D2ClientBuilder setIndisStartUpExecutorService(ScheduledExecutorService executorService)
+  {
+    _config.indisStartUpExecutorService = executorService;
+    return this;
+  }
+
   public D2ClientBuilder setDownstreamServicesFetcher(DownstreamServicesFetcher downstreamServicesFetcher)
   {
     _config.downstreamServicesFetcher = downstreamServicesFetcher;