Spring Stomp Websocket -太大的发送消息生成1009错误并断开连接

在使用 Spring 的 STOMP WebSocket 实现时，如果发送的消息过大，可能会导致客户端收到 1009 错误并被断开连接。这是因为 WebSocket 协议对消息大小有一定的限制，具体限制取决于服务器和客户端的配置。以下是解决此问题的详细步骤和建议：

1. 理解错误代码 `1009`

WebSocket 错误代码 1009 表示 "Policy Violation"（策略违规）。这通常意味着发送的消息超过了服务器或客户端设定的最大允许大小。

2. 检查和调整服务器配置

a. Spring 配置

确保在 Spring 的 WebSocket 配置中设置了适当的消息大小限制。可以通过 WebSocketMessageBrokerConfigurer 接口进行配置：

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.*;

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic", "/queue");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws").withSockJS();
    }

    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(new ChannelInterceptor() {
            @Override
            public Message<?> preSend(Message<?> message, MessageChannel channel) {
                StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);
                if (StompCommand.SEND.equals(accessor.getCommand())) {
                    // 设置最大消息大小，例如 10MB
                    int maxMessageSize = 10 * 1024 * 1024;
                    if (message.getPayload() instanceof byte[]) {
                        if (((byte[]) message.getPayload()).length > maxMessageSize) {
                            throw new MessageSizeExceededException("Message size exceeded: " + maxMessageSize);
                        }
                    }
                }
                return message;
            }
        });
    }
}

b. Tomcat 配置（如果使用嵌入式 Tomcat）

在 application.properties 或 application.yml 中配置最大消息大小：

# application.properties
server.tomcat.max-http-post-size=10MB

或者在 application.yml 中：

# application.yml
server:
  tomcat:
    max-http-post-size: 10485760 # 10MB

3. 调整客户端配置

确保客户端在建立 WebSocket 连接时设置了适当的消息大小限制。例如，使用 JavaScript 客户端时，可以在连接 URL 中添加参数：

const socket = new WebSocket('ws://localhost:8080/ws?maxFramePayloadSize=10485760');

4. 分片传输大消息

如果确实需要发送较大的消息，可以考虑将消息分片（chunking）传输。这可以通过自定义消息处理逻辑实现，将大消息拆分成多个较小的消息进行发送，并在客户端重新组装。

5. 检查网络中间件

如果使用了负载均衡器、代理服务器（如 Nginx）或其他网络中间件，确保这些组件没有对 WebSocket 消息大小设置额外的限制。例如，在 Nginx 中，可以调整以下配置：

http {
    map $http_upgrade $connection_upgrade {
        default upgrade;
        ''      close;
    }

    server {
        listen 80;
        server_name example.com;

        location /ws {
            proxy_pass http://backend;
            proxy_http_version 1.1;
            proxy_set_header Upgrade $http_upgrade;
            proxy_set_header Connection $connection_upgrade;
            proxy_read_timeout 86400;

            # 设置最大消息大小，例如 10MB
            client_max_body_size 10M;
        }
    }
}