专业做网站较好的公司建程网手机版建设建筑工程网

结合Spring Boot和WebSocket实现一个基本示例，并且使用Spring Bean注入的方式来组织代码。

1. 创建Spring Boot项目

首先，确保你有一个Spring Boot项目，并在pom.xml文件中引入了WebSocket相关的依赖。

<dependencies><!-- Spring Boot WebSocket --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-websocket</artifactId></dependency><!-- Spring Boot Starter Web for Rest Controllers --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><!-- Spring Boot Starter for Logging and other utilities --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-logging</artifactId></dependency>
</dependencies>

2. WebSocket 配置类

创建一个配置类，启用WebSocket支持，并且在这个类中配置一个TextWebSocketHandler来处理消息的传递。

import org.springframework.context.annotation.Configuration;
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.config.annotation.WebSocketHandlerRegistry;@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {private final MyWebSocketHandler myWebSocketHandler;// 注入自定义的 WebSocketHandlerpublic WebSocketConfig(MyWebSocketHandler myWebSocketHandler) {this.myWebSocketHandler = myWebSocketHandler;}@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {registry.addHandler(myWebSocketHandler, "/ws").setAllowedOrigins("*");}
}

3. 创建自定义WebSocketHandler

MyWebSocketHandler 类继承自TextWebSocketHandler，用来处理WebSocket消息的接收和发送。我们将使用Spring的@Component注解让这个类成为一个Spring Bean，并通过构造器注入来引入服务逻辑。

import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
import org.springframework.web.socket.TextMessage;@Component
public class MyWebSocketHandler extends TextWebSocketHandler {private final MyService myService;// 构造器注入MyServicepublic MyWebSocketHandler(MyService myService) {this.myService = myService;}@Overridepublic void handleTextMessage(WebSocketSession session, TextMessage message) {// 通过MyService处理消息String response = myService.processMessage(message.getPayload());try {// 发送处理后的消息session.sendMessage(new TextMessage(response));} catch (Exception e) {e.printStackTrace();}}
}

4. 创建一个简单的Service

我们创建一个简单的Service类MyService，用来处理消息的业务逻辑。它会被注入到MyWebSocketHandler中。

import org.springframework.stereotype.Service;@Service
public class MyService {public String processMessage(String message) {// 这里是消息处理的业务逻辑，可以根据实际需要修改return "处理后的消息: " + message;}
}

5. 创建Controller（可选）

如果你想通过HTTP请求来访问WebSocket服务，或者提供一个WebSocket客户端连接的页面，你可以创建一个简单的Controller。

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;@RestController
public class WebSocketController {@GetMapping("/ws")public String hello() {return "WebSocket 服务正在运行";}
}

6. 启动应用

创建Application类来启动Spring Boot应用：

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplication
public class WebSocketApplication {public static void main(String[] args) {SpringApplication.run(WebSocketApplication.class, args);}
}

7. 测试WebSocket连接

使用WebSocket客户端（例如浏览器控制台、Postman、WebSocket客户端插件等）连接到ws://localhost:8080/ws，并发送一些消息。你应该能收到由MyService处理后的消息。

例如，发送消息"Hello WebSocket"，应该会收到类似 "处理后的消息: Hello WebSocket" 的响应。

---

说明：

• 在WebSocketConfig中，我们将MyWebSocketHandler注册为WebSocket处理器，并指定WebSocket的URL /ws。
• 通过构造器注入，MyWebSocketHandler可以访问Spring管理的MyService Bean，并在WebSocket连接中调用它处理消息。
• MyService封装了消息的业务逻辑，保持了代码的清晰与解耦。

这个示例展示了Spring Boot与WebSocket的基本集成，同时也体现了如何使用Spring的依赖注入来组织代码。