package org.ruoyi.common.chat.config;
|
|
import cn.hutool.core.util.StrUtil;
|
import org.ruoyi.common.chat.config.properties.WebSocketProperties;
|
import org.ruoyi.common.chat.handler.PlusWebSocketHandler;
|
import org.ruoyi.common.chat.interceptor.PlusWebSocketInterceptor;
|
import org.ruoyi.common.chat.listener.WebSocketTopicListener;
|
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.web.socket.WebSocketHandler;
|
import org.springframework.web.socket.config.annotation.EnableWebSocket;
|
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
|
import org.springframework.web.socket.server.HandshakeInterceptor;
|
|
/**
|
* WebSocket 配置
|
*
|
* @author zendwang
|
*/
|
@AutoConfiguration
|
@ConditionalOnProperty(value = "websocket.enabled", havingValue = "true")
|
@EnableConfigurationProperties(WebSocketProperties.class)
|
@EnableWebSocket
|
public class WebSocketConfig {
|
|
@Bean
|
public WebSocketConfigurer webSocketConfigurer(HandshakeInterceptor handshakeInterceptor,
|
WebSocketHandler webSocketHandler,
|
WebSocketProperties webSocketProperties) {
|
// 如果WebSocket的路径为空，则设置默认路径为 "/websocket"
|
if (StrUtil.isBlank(webSocketProperties.getPath())) {
|
webSocketProperties.setPath("/websocket");
|
}
|
// 如果允许跨域访问的地址为空，则设置为 "*"，表示允许所有来源的跨域请求
|
if (StrUtil.isBlank(webSocketProperties.getAllowedOrigins())) {
|
webSocketProperties.setAllowedOrigins("*");
|
}
|
// 返回一个WebSocketConfigurer对象，用于配置WebSocket
|
return registry -> registry
|
// 添加WebSocket处理程序和拦截器到指定路径，设置允许的跨域来源
|
.addHandler(webSocketHandler, webSocketProperties.getPath())
|
.addInterceptors(handshakeInterceptor)
|
.setAllowedOrigins(webSocketProperties.getAllowedOrigins());
|
}
|
|
@Bean
|
public HandshakeInterceptor handshakeInterceptor() {
|
return new PlusWebSocketInterceptor();
|
}
|
|
@Bean
|
public WebSocketHandler webSocketHandler() {
|
return new PlusWebSocketHandler();
|
}
|
|
@Bean
|
public WebSocketTopicListener topicListener() {
|
return new WebSocketTopicListener();
|
}
|
}
|
