springboot websocket 客户端 自定义请求头

Spring Boot WebSocket客户端允许你在连接到服务器时添加自定义请求头。这通常通过`HandshakeInterceptor`来实现，这是WebSocket握手过程的一部分。以下是一个简单的步骤：

1. 首先，创建一个实现了`HandshakeHandler`接口的类，并实现`handle`方法。这个方法会在建立WebSocket连接前被调用，你可以在这个时候添加自定义的请求头。

import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.socket.HandshakeHandler;

public class CustomHeaderHandler implements HandshakeHandler {
    @Override
    public void handle(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler handler, Map<String, Object> attributes) throws Exception {
        // 获取当前请求头
        HttpHeaders headers = request.getHeaders();

        // 添加自定义头部
        headers.add("Custom-Header", "YourValue");

        // 设置新的请求头并继续握手流程
        request.setHeaders(headers);
        handler.handle(request, response);
    }
}

2. 然后，在`WebMvcConfigurerAdapter`或`WebFluxConfigurer`中注册你的`HandshakeHandler`。

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {

    @Autowired
    private CustomHeaderHandler customHeaderHandler;

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        //...
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        //...
    }

    @Override
    public void addHandshakeInterceptors(List<HandshakeInterceptor> interceptors) {
        interceptors.add(customHeaderHandler);
    }
}