[Issue 7049][WebSocket] Make WebSocketService.MaxTextFrameSize config…

…urable (apache#7155)

Motivation
fix apache#7049
Make WebSocketService.MaxTextFrameSize configurable.

Modifications
Add config of webSocketMaxTextFrameSize in websocket.conf, standalone.conf, broker.conf, and the default value is 1024*1024byte.

conf/broker.conf

@@ -947,6 +947,9 @@ webSocketConnectionsPerBroker=8
 # Time in milliseconds that idle WebSocket session times out
 webSocketSessionIdleTimeoutMillis=300000
 
+# The maximum size of a text message during parsing in WebSocket proxy
+webSocketMaxTextFrameSize=1048576
+
 ### --- Metrics --- ###
 
 # Enable topic level metrics

conf/standalone.conf

@@ -700,6 +700,9 @@ webSocketConnectionsPerBroker=8
 # Time in milliseconds that idle WebSocket session times out
 webSocketSessionIdleTimeoutMillis=300000
 
+# The maximum size of a text message during parsing in WebSocket proxy
+webSocketMaxTextFrameSize=1048576
+
 ### --- Metrics --- ###
 
 # Enable topic level metrics

conf/websocket.conf

@@ -60,6 +60,9 @@ webSocketConnectionsPerBroker=8
 # Time in milliseconds that idle WebSocket session times out
 webSocketSessionIdleTimeoutMillis=300000
 
+# The maximum size of a text message during parsing in WebSocket proxy
+webSocketMaxTextFrameSize=1048576
+
 ### --- Authentication --- ###
 
 # Enable authentication

deployment/terraform-ansible/templates/broker.conf

@@ -467,6 +467,8 @@ webSocketNumIoThreads=8
 # Number of connections per Broker in Pulsar Client used in WebSocket proxy
 webSocketConnectionsPerBroker=8
 
+# The maximum size of a text message during parsing in WebSocket proxy
+webSocketMaxTextFrameSize=1048576
 
 ### --- Metrics --- ###
 

pulsar-broker-common/src/main/java/org/apache/pulsar/broker/ServiceConfiguration.java

@@ -1563,6 +1563,12 @@ public class ServiceConfiguration implements PulsarConfiguration {
     )
     private int webSocketSessionIdleTimeoutMillis = 300000;
 
+    @FieldContext(
+        category = CATEGORY_WEBSOCKET,
+        doc = "The maximum size of a text message during parsing in WebSocket proxy."
+    )
+    private int webSocketMaxTextFrameSize = 1048576;
+
     /**** --- Metrics --- ****/
     @FieldContext(
         category = CATEGORY_METRICS,

pulsar-websocket/src/main/java/org/apache/pulsar/websocket/WebSocketConsumerServlet.java

@@ -36,7 +36,7 @@ public WebSocketConsumerServlet(WebSocketService service) {
 
     @Override
     public void configure(WebSocketServletFactory factory) {
-        factory.getPolicy().setMaxTextMessageSize(WebSocketService.MaxTextFrameSize);
+        factory.getPolicy().setMaxTextMessageSize(service.getConfig().getWebSocketMaxTextFrameSize());
         if (service.getConfig().getWebSocketSessionIdleTimeoutMillis() > 0) {
             factory.getPolicy().setIdleTimeout(service.getConfig().getWebSocketSessionIdleTimeoutMillis());
         }

pulsar-websocket/src/main/java/org/apache/pulsar/websocket/WebSocketProducerServlet.java

@@ -35,7 +35,7 @@ public WebSocketProducerServlet(WebSocketService service) {
 
     @Override
     public void configure(WebSocketServletFactory factory) {
-        factory.getPolicy().setMaxTextMessageSize(WebSocketService.MaxTextFrameSize);
+        factory.getPolicy().setMaxTextMessageSize(service.getConfig().getWebSocketMaxTextFrameSize());
         if (service.getConfig().getWebSocketSessionIdleTimeoutMillis() > 0) {
             factory.getPolicy().setIdleTimeout(service.getConfig().getWebSocketSessionIdleTimeoutMillis());
         }

pulsar-websocket/src/main/java/org/apache/pulsar/websocket/WebSocketReaderServlet.java

@@ -36,7 +36,7 @@ public WebSocketReaderServlet(WebSocketService service) {
 
     @Override
     public void configure(WebSocketServletFactory factory) {
-        factory.getPolicy().setMaxTextMessageSize(WebSocketService.MaxTextFrameSize);
+        factory.getPolicy().setMaxTextMessageSize(service.getConfig().getWebSocketMaxTextFrameSize());
         if (service.getConfig().getWebSocketSessionIdleTimeoutMillis() > 0) {
             factory.getPolicy().setIdleTimeout(service.getConfig().getWebSocketSessionIdleTimeoutMillis());
         }

pulsar-websocket/src/main/java/org/apache/pulsar/websocket/WebSocketService.java

@@ -62,8 +62,6 @@
  */
 public class WebSocketService implements Closeable {
 
-    public static final int MaxTextFrameSize = 1024 * 1024;
-
     AuthenticationService authenticationService;
     AuthorizationService authorizationService;
     PulsarClient pulsarClient;

pulsar-websocket/src/main/java/org/apache/pulsar/websocket/service/WebSocketProxyConfiguration.java

@@ -66,6 +66,8 @@ public class WebSocketProxyConfiguration implements PulsarConfiguration {
     private Optional<Integer> webServicePortTls = Optional.empty();
     // Hostname or IP address the service binds on, default is 0.0.0.0.
     private String bindAddress;
+    // The maximum size of a text message during parsing in WebSocket proxy
+    private int webSocketMaxTextFrameSize = 1024 * 1024;
     // --- Authentication ---
     // Enable authentication
     private boolean authenticationEnabled;
@@ -232,6 +234,14 @@ public void setBindAddress(String bindAddress) {
         this.bindAddress = bindAddress;
     }
 
+    public int getWebSocketMaxTextFrameSize() {
+        return webSocketMaxTextFrameSize;
+    }
+
+    public void setWebSocketMaxTextFrameSize(int webSocketMaxTextFrameSize) {
+        this.webSocketMaxTextFrameSize = webSocketMaxTextFrameSize;
+    }
+
     public boolean isAuthenticationEnabled() {
         return authenticationEnabled;
     }