Merge pull request AsyncHttpClient#13

Author: jfarcand

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

@@ -80,6 +80,7 @@ public class AsyncHttpClientConfig {
     private final boolean useRawUrl;
     private final boolean removeQueryParamOnRedirect;
     private final HostnameVerifier hostnameVerifier;
+    private final int ioThreadMultiplier;
 
     private AsyncHttpClientConfig(int maxTotalConnections,
                                   int maxConnectionPerHost,
@@ -106,7 +107,8 @@ private AsyncHttpClientConfig(int maxTotalConnections,
                                   boolean allowSslConnectionCaching,
                                   boolean useRawUrl,
                                   boolean removeQueryParamOnRedirect,
-                                  HostnameVerifier hostnameVerifier) {
+                                  HostnameVerifier hostnameVerifier,
+                                  int ioThreadMultiplier) {
 
         this.maxTotalConnections = maxTotalConnections;
         this.maxConnectionPerHost = maxConnectionPerHost;
@@ -132,6 +134,7 @@ private AsyncHttpClientConfig(int maxTotalConnections,
         this.allowSslConnectionPool = allowSslConnectionCaching;
         this.removeQueryParamOnRedirect = removeQueryParamOnRedirect;
         this.hostnameVerifier = hostnameVerifier;
+        this.ioThreadMultiplier = ioThreadMultiplier;
 
         if (applicationThreadPool == null) {
             this.applicationThreadPool = Executors.newCachedThreadPool();
@@ -426,6 +429,13 @@ public boolean isClosed(){
     public HostnameVerifier getHostnameVerifier() {
         return hostnameVerifier;
     }
+    /***
+     *
+     * @return number to multiply by availableProcessors() that will determine # of NioWorkers to use
+     */
+    public int getIoThreadMultiplier() {
+        return ioThreadMultiplier;
+    }
 
     /**
      * Builder for an {@link AsyncHttpClient}
@@ -476,6 +486,7 @@ public boolean verify( String s, SSLSession sslSession ) {
                 return true;
             }
         };
+        private int ioThreadMultiplier = 2;
 
         public Builder() {
         }
@@ -865,6 +876,11 @@ public Builder setUseProxyProperties(boolean useProxyProperties) {
             return this;
         }
 
+        public Builder setIOThreadMultiplier(int multiplier){
+            this.ioThreadMultiplier = multiplier;
+            return this;
+        }
+
         /**
          * Set the {@link HostnameVerifier}
          * @param hostnameVerifier {@link HostnameVerifier}
@@ -903,6 +919,7 @@ public Builder(AsyncHttpClientConfig prototype) {
             requestFilters.addAll(prototype.getRequestFilters());
             responseFilters.addAll(prototype.getResponseFilters());
             useRawUrl = prototype.isUseRawUrl();
+            ioThreadMultiplier = prototype.getIoThreadMultiplier();
         }
 
         /**
@@ -946,7 +963,8 @@ public AsyncHttpClientConfig build() {
                     allowSslConnectionPool,
                     useRawUrl,
                     removeQueryParamOnRedirect,
-                    hostnameVerifier);
+                    hostnameVerifier,
+                    ioThreadMultiplier);
         }
     }
 }

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

@@ -194,7 +194,9 @@ public NettyAsyncHttpProvider(AsyncHttpClientConfig config) {
             } else {
                 e = Executors.newCachedThreadPool();
             }
-            socketChannelFactory = new NioClientSocketChannelFactory(e, config.executorService());
+            int numWorkers = config.getIoThreadMultiplier() * Runtime.getRuntime().availableProcessors();
+            log.info("Number of application's worked threads is {}", numWorkers);
+            socketChannelFactory = new NioClientSocketChannelFactory(e, config.executorService(), numWorkers);
         }
         plainBootstrap = new ClientBootstrap(socketChannelFactory);
         secureBootstrap = new ClientBootstrap(socketChannelFactory);