diff --git a/d2/src/main/java/com/linkedin/d2/balancer/D2ClientBuilder.java b/d2/src/main/java/com/linkedin/d2/balancer/D2ClientBuilder.java index fbb07fe9ee..d3183aea39 100644 --- 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;