43 lines
2.0 KiB
Java
43 lines
2.0 KiB
Java
package com.baeldung.sendtouser;
|
|
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.http.server.ServerHttpRequest;
|
|
import org.springframework.http.server.ServerHttpResponse;
|
|
import org.springframework.http.server.ServletServerHttpRequest;
|
|
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
|
|
import org.springframework.web.socket.WebSocketHandler;
|
|
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
|
|
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
|
|
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
|
|
import org.springframework.web.socket.server.support.DefaultHandshakeHandler;
|
|
|
|
import javax.servlet.http.HttpSession;
|
|
import java.util.Map;
|
|
|
|
@Configuration
|
|
@EnableWebSocketMessageBroker
|
|
public class WebSocketSendToUserConfig implements WebSocketMessageBrokerConfigurer {
|
|
|
|
@Override
|
|
public void configureMessageBroker(MessageBrokerRegistry config) {
|
|
config.enableSimpleBroker("/topic/", "/queue/");
|
|
config.setApplicationDestinationPrefixes("/app");
|
|
}
|
|
|
|
@Override
|
|
public void registerStompEndpoints(StompEndpointRegistry registry) {
|
|
registry.addEndpoint("/greeting").setHandshakeHandler(new DefaultHandshakeHandler() {
|
|
|
|
//Get sessionId from request and set it in Map attributes
|
|
public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
|
|
Map attributes) throws Exception {
|
|
if (request instanceof ServletServerHttpRequest) {
|
|
ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
|
|
HttpSession session = servletRequest.getServletRequest().getSession();
|
|
attributes.put("sessionId", session.getId());
|
|
}
|
|
return true;
|
|
}}).withSockJS();
|
|
}
|
|
}
|