44 lines
1.9 KiB
Java
44 lines
1.9 KiB
Java
package sample.config;
|
|
|
|
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.messaging.simp.SimpMessageSendingOperations;
|
|
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
|
import org.springframework.session.ExpiringSession;
|
|
import org.springframework.session.web.socket.config.annotation.AbstractSessionWebSocketMessageBrokerConfigurer;
|
|
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
|
|
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
|
|
|
|
import sample.data.ActiveWebSocketUserRepository;
|
|
import sample.websocket.WebSocketConnectHandler;
|
|
import sample.websocket.WebSocketDisconnectHandler;
|
|
|
|
@Configuration
|
|
@EnableScheduling
|
|
@EnableWebSocketMessageBroker
|
|
public class WebSocketConfig<S extends ExpiringSession> extends AbstractSessionWebSocketMessageBrokerConfigurer<S> {
|
|
|
|
@Bean
|
|
public WebSocketConnectHandler<S> webSocketConnectHandler(SimpMessageSendingOperations messagingTemplate, ActiveWebSocketUserRepository repository) {
|
|
return new WebSocketConnectHandler<S>(messagingTemplate, repository);
|
|
}
|
|
|
|
@Override
|
|
protected void configureStompEndpoints(StompEndpointRegistry registry) {
|
|
registry.addEndpoint("/messages")
|
|
.withSockJS();
|
|
}
|
|
|
|
@Override
|
|
public void configureMessageBroker(MessageBrokerRegistry registry) {
|
|
registry.enableSimpleBroker("/queue/", "/topic/");
|
|
registry.setApplicationDestinationPrefixes("/app");
|
|
}
|
|
|
|
@Bean
|
|
public WebSocketDisconnectHandler<S> webSocketDisconnectHandler(SimpMessageSendingOperations messagingTemplate, ActiveWebSocketUserRepository repository) {
|
|
return new WebSocketDisconnectHandler<S>(messagingTemplate, repository);
|
|
}
|
|
} |