feat: 新增 websocket 功能模块

This commit is contained in:
haoxr
2023-06-03 11:02:44 +08:00
parent 260e7950c5
commit 808c33789c
3 changed files with 135 additions and 0 deletions

View File

@@ -0,0 +1,32 @@
package com.youlai.system.config;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
/**
* WebSocket 配置类
*
* @author haoxr
* @since 2.3.0
*/
@Configuration
@ConditionalOnProperty(name = "system.config.websocket-enabled")// system.config.websocket-enabled = true 才会自动装配
@EnableWebSocketMessageBroker // 注解告诉Spring框架要开启WebSocket消息代理的功能并配置相关的端点和消息代理
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
@Override
public void configureMessageBroker(MessageBrokerRegistry registry) {
registry.enableSimpleBroker("/topic", "/user");
}
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
// 注册了一个 /ws 的端点用于建立WebSocket连接。
registry.addEndpoint("/ws").setAllowedOriginPatterns("*");
}
}