Expose numConnections from HttpClient

Author: liori

src/main/java/com/simplaex/clients/druid/DruidClientConfig.java

@@ -24,6 +24,7 @@ public class DruidClientConfig {
 
   private final DruidClient.EventEmitter eventEmitter;
 
+  private final Integer numConnections;
 
   public ObjectMapper getObjectMapper() {
     if (objectMapper == null) {
@@ -58,4 +59,8 @@ public int getPort() {
     return port != null && port > 0 ? port : 8080;
   }
 
+  @Nonnegative
+  public int getNumConnections() {
+    return numConnections != null && numConnections > 0 ? numConnections : 1;
+  }
 }

src/main/java/com/simplaex/clients/druid/DruidClientImpl.java

@@ -71,7 +71,7 @@ public final class DruidClientImpl implements DruidClient {
     this.queryManager = new QueryManager();
     this.executorService = config.getExecutorService();
     this.druidClient = createDruidClient(config.getHost(), config.getPort(), queryManager, serviceEmitter, executorService,
-      config.getObjectMapper());
+      config.getObjectMapper(), config.getNumConnections());
   }
 
   private static DirectDruidClient createDruidClient(
@@ -80,14 +80,15 @@ private static DirectDruidClient createDruidClient(
     final QueryWatcher queryWatcher,
     final ServiceEmitter serviceEmitter,
     final ExecutorService executorService,
-    final ObjectMapper objectMapper
+    final ObjectMapper objectMapper,
+    final int numConnections
   ) {
     final String host = String.format("%s:%d", hostname, port);
     return new DirectDruidClient(
       createQueryToolChestWarehouse(objectMapper, serviceEmitter, queryWatcher, executorService),
       queryWatcher,
       objectMapper,
-      createHttpClient(),
+      createHttpClient(numConnections),
       "http",
       host, serviceEmitter
     );
@@ -292,8 +293,8 @@ public String getFormatString() {
     return new MapQueryToolChestWarehouse(chestMap);
   }
 
-  private static HttpClient createHttpClient() {
-    final HttpClientConfig httpClientConfig = HttpClientConfig.builder().build();
+  private static HttpClient createHttpClient(final int numConnections) {
+    final HttpClientConfig httpClientConfig = HttpClientConfig.builder().withNumConnections(numConnections).build();
     final Lifecycle lifecycle = new Lifecycle();
     return HttpClientInit.createClient(
       httpClientConfig,