springboot stomp从连接中获取参数

在Spring Boot中使用STOMP协议时，可以从STOMP消息的头部（headers）中获取参数。下面是一个示例，展示如何从连接中获取参数：

1. 在WebSocket配置类中，可以通过`configureClientInboundChannel`方法获取到`ChannelRegistration`对象，然后注册自定义的`ChannelInterceptor`实现类。

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
    
    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(new MyChannelInterceptor());
    }
}

2. 自定义`ChannelInterceptor`实现类中的`preSend`方法可以获取到`Message<?>`对象，从而可以获取到STOMP消息的头部。

public class MyChannelInterceptor extends ChannelInterceptorAdapter {
 
    @Override
    public Message<?> preSend(Message<?> message, MessageChannel channel) {
        StompHeaderAccessor accessor = StompHeaderAccessor.wrap(message);
        String sessionId = accessor.getSessionId();
        String destination = accessor.getDestination();
        
        // 从Stomp消息头部获取参数
        String customParam = accessor.getFirstNativeHeader("custom-param");
        
        // TODO: 处理自定义逻辑
        
        return super.preSend(message, channel);
    }
}

3. 在客户端连接WebSocket时，可以在连接URL的查询参数中添加自定义参数，例如：

var socket = new SockJS('/websocket?custom-param=my-custom-param');

这样，在`MyChannelInterceptor`中的`preSend`方法中，就可以获取到这个参数的值。