Fix for #888: configure the buffer size as well

Atmosphere · Feb 11, 2013 · 5ea5722 · 5ea5722
1 parent 266c311
commit 5ea5722
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/modules/cpr/src/main/java/org/atmosphere/container/TomcatWebSocketHandler.java b/modules/cpr/src/main/java/org/atmosphere/container/TomcatWebSocketHandler.java
@@ -54,6 +54,12 @@ public TomcatWebSocketHandler(AtmosphereRequest request, AtmosphereFramework fra
         } else {
             webSocketWriteTimeout = -1;
         }
+
+        s = framework.getAtmosphereConfig().getInitParameter(ApplicationConfig.WEBSOCKET_BUFFER_SIZE);
+        if (s != null) {
+            setByteBufferMaxSize(Integer.valueOf(s));
+            setCharBufferMaxSize(getByteBufferMaxSize());
+        }
     }
 
     @Override