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);