add io thread config param

Author: jloomis

src/main/java/com/ning/http/client/AsyncHttpClientConfig.java

@@ -77,6 +77,7 @@ public class AsyncHttpClientConfig {
     private final boolean allowSslConnectionPool;
     private final boolean useRawUrl;
     private final boolean removeQueryParamOnRedirect;
+    private final int ioThreadMultiplier;
 
     private AsyncHttpClientConfig(int maxTotalConnections,
                                   int maxConnectionPerHost,
@@ -102,7 +103,8 @@ private AsyncHttpClientConfig(int maxTotalConnections,
                                   int maxRequestRetry,
                                   boolean allowSslConnectionCaching,
                                   boolean useRawUrl,
-                                  boolean removeQueryParamOnRedirect) {
+                                  boolean removeQueryParamOnRedirect,
+                                  int ioThreadMultiplier) {
 
         this.maxTotalConnections = maxTotalConnections;
         this.maxConnectionPerHost = maxConnectionPerHost;
@@ -127,6 +129,7 @@ private AsyncHttpClientConfig(int maxTotalConnections,
         this.reaper = reaper;
         this.allowSslConnectionPool = allowSslConnectionCaching;
         this.removeQueryParamOnRedirect = removeQueryParamOnRedirect;
+        this.ioThreadMultiplier = ioThreadMultiplier;
 
         if (applicationThreadPool == null) {
             this.applicationThreadPool = Executors.newCachedThreadPool();
@@ -406,6 +409,13 @@ public boolean isRemoveQueryParamOnRedirect() {
         return removeQueryParamOnRedirect;
     }
 
+    /***
+     *
+     * @return number to multiply by availableProcessors() that will determine # of NioWorkers to use
+     */
+    public int getIoThreadMultiplier() {
+        return ioThreadMultiplier;
+    }
 
     /**
      * Builder for an {@link AsyncHttpClient}
@@ -450,6 +460,7 @@ public Thread newThread(Runnable r) {
         private boolean allowSslConnectionPool = true;
         private boolean useRawUrl = false;
         private boolean removeQueryParamOnRedirect = true;
+        private int ioThreadMultiplier = 2;
 
         public Builder() {
         }
@@ -839,6 +850,11 @@ public Builder setUseProxyProperties(boolean useProxyProperties) {
             return this;
         }
 
+        public Builder setIOThreadMultiplier(int multiplier){
+            this.ioThreadMultiplier = multiplier;
+            return this;
+        }
+
         /**
          * Create a config builder with values taken from the given prototype configuration.
          *
@@ -867,6 +883,7 @@ public Builder(AsyncHttpClientConfig prototype) {
             requestFilters.addAll(prototype.getRequestFilters());
             responseFilters.addAll(prototype.getResponseFilters());
             useRawUrl = prototype.isUseRawUrl();
+            ioThreadMultiplier = prototype.getIoThreadMultiplier();
         }
 
         /**
@@ -904,7 +921,8 @@ public AsyncHttpClientConfig build() {
                     maxRequestRetry,
                     allowSslConnectionPool,
                     useRawUrl,
-                    removeQueryParamOnRedirect);
+                    removeQueryParamOnRedirect,
+                    ioThreadMultiplier);
         }
     }
 }

src/main/java/com/ning/http/client/providers/netty/NettyAsyncHttpProvider.java

@@ -195,7 +195,9 @@ public NettyAsyncHttpProvider(AsyncHttpClientConfig config) {
             } else {
                 e = Executors.newCachedThreadPool();
             }
-            socketChannelFactory = new NioClientSocketChannelFactory(e, config.executorService());
+            int numWorkers = config.getIoThreadMultiplier()*Runtime.getRuntime().availableProcessors();
+            log.info("numWorkers=" + numWorkers);
+            socketChannelFactory = new NioClientSocketChannelFactory(e, config.executorService(), numWorkers);
         }
         plainBootstrap = new ClientBootstrap(socketChannelFactory);
         secureBootstrap = new ClientBootstrap(socketChannelFactory);