EasyChatting/src/main/java/com/dx/easychat/easychatting/configuration/WebSocketConfiguration.java

26 lines
1007 B
Java
Raw Normal View History

2024-11-04 15:11:00 +08:00
package com.dx.easychat.easychatting.configuration;
import com.dx.easychat.easychatting.handler.WebSocketReceiveHandler;
import jakarta.annotation.Resource;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;
@Configuration
@EnableWebSocket
public class WebSocketConfiguration implements WebSocketConfigurer {
@Resource
private WebSocketReceiveHandler webSocketReceiveHandler;
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler( webSocketReceiveHandler, "/websocket/message")
.setAllowedOrigins("*")
.addInterceptors(new HttpSessionHandshakeInterceptor());
}
}