拦截

事件 提供了 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 可以使用StompHeaderAccessorSimpMessageHeaderAccessor 来访问有关消息的信息,如下例所示

  • 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 会话关闭时自动生成。在某些情况下,拦截器可能会针对每个会话多次拦截此消息。组件应该对多个断开连接事件具有幂等性。