在 Spring Boot 中,要在与 HTTP 不同的端口上配置 WebSocket,您需要创建一个单独的 Server
实例来处理 WebSocket 连接。以下是如何在 Spring Boot 中配置 WebSocket 以侦听不同端口的步骤:
pom.xml
文件中添加 Spring Boot WebSocket 依赖: <groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
AbstractWebSocketMessageBrokerConfigurer
,并重写 registerStompEndpoints
和 configureMessageBroker
方法。在这个配置类中,您可以指定 WebSocket 服务器的端口。import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {
@Override
public void configureMessageBroker(MessageBrokerRegistry config) {
config.enableSimpleBroker("/topic");
config.setApplicationDestinationPrefixes("/app");
}
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
registry.addEndpoint("/websocket").setAllowedOrigins("*").withSockJS();
}
}
WebServerFactoryCustomizer<ConfigurableWebServerFactory>
接口。在这个类中,您可以指定 WebSocket 服务器的端口。import org.springframework.boot.web.server.ConfigurableWebServerFactory;
import org.springframework.boot.web.server.WebServerFactoryCustomizer;
import org.springframework.stereotype.Component;
@Component
public class WebSocketServerCustomizer implements WebServerFactoryCustomizer<ConfigurableWebServerFactory> {
@Override
public void customize(ConfigurableWebServerFactory factory) {
factory.setPort(8081); // 指定 WebSocket 服务器的端口
}
}
在这个示例中,我们将 WebSocket 服务器的端口设置为 8081。您可以根据需要更改此端口。
@MessageMapping
注解来处理特定的消息。import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;
@Controller
public class WebSocketController {
@MessageMapping("/chat")
@SendTo("/topic/messages")
public String processMessage(String message) {
return "Received: " + message;
}
}
完成以上步骤后,您应该已经成功地在 Spring Boot 中配置了一个监听不同端口的 WebSocket 服务器。
领取专属 10元无门槛券
手把手带您无忧上云