拦截
事件提供了STOMP连接生命周期的通知,但没有提供每个客户端消息的通知。应用程序还可以注册一个 ChannelInterceptor 来拦截任何消息以及处理链的任何部分。以下示例展示了如何拦截来自客户端的入站消息。
-
Java
-
Kotlin
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfiguration implements WebSocketMessageBrokerConfigurer {
@Override
public void configureClientInboundChannel(ChannelRegistration registration) {
registration.interceptors(new MyChannelInterceptor());
}
}
@Configuration
@EnableWebSocketMessageBroker
class WebSocketConfiguration : WebSocketMessageBrokerConfigurer {
override fun configureClientInboundChannel(registration: ChannelRegistration) {
registration.interceptors(MyChannelInterceptor())
}
}
自定义的 ChannelInterceptor 可以使用 StompHeaderAccessor 或 SimpMessageHeaderAccessor 来访问消息信息,如下例所示:
-
Java
-
Kotlin
public class MyChannelInterceptor implements ChannelInterceptor {
@Override
public Message<?> preSend(Message<?> message, MessageChannel channel) {
StompHeaderAccessor accessor = StompHeaderAccessor.wrap(message);
StompCommand command = accessor.getCommand();
// ...
return message;
}
}
class MyChannelInterceptor : ChannelInterceptor {
override fun preSend(message: Message<*>, channel: MessageChannel): Message<*> {
val accessor = StompHeaderAccessor.wrap(message)
val command = accessor.command
// ...
return message
}
}
应用程序还可以实现 ExecutorChannelInterceptor,它是 ChannelInterceptor 的子接口,在处理消息的线程中提供回调。当每个发送到通道的消息只调用一次 ChannelInterceptor 时,ExecutorChannelInterceptor 为订阅通道消息的每个 MessageHandler 的线程提供了钩子。
请注意,与前面描述的 SessionDisconnectEvent 一样,DISCONNECT 消息可以来自客户端,也可以在WebSocket会话关闭时自动生成。在某些情况下,拦截器可能会为每个会话多次拦截此消息。组件应该对多次断开连接事件具有幂等性。