Fix WS timeout.

Author: rlubke

src/main/java/com/ning/http/client/providers/grizzly/GrizzlyAsyncHttpProvider.java

@@ -306,7 +306,7 @@ protected <T> ListenableFuture<T> execute(final Connection c,
     }
 
 
-    protected void initializeTransport(AsyncHttpClientConfig clientConfig) {
+    protected void initializeTransport(final AsyncHttpClientConfig clientConfig) {
 
         final FilterChainBuilder fcb = FilterChainBuilder.stateless();
         fcb.add(new AsyncHttpClientTransportFilter());
@@ -326,6 +326,9 @@ public long getTimeout(FilterChainContext ctx) {
                             final HttpTransactionContext context =
                                     GrizzlyAsyncHttpProvider.this.getHttpTransactionContext(ctx.getConnection());
                             if (context != null) {
+                                if (context.isWSRequest) {
+                                    return clientConfig.getWebSocketIdleTimeoutInMs();
+                                }
                                 final PerRequestConfig config = context.request.getPerRequestConfig();
                                 if (config != null) {
                                     final long timeout = config.getRequestTimeoutInMs();