springboot 使用websocket 实现消息提示

Spring Boot是一个用于创建基于Spring的独立的、生产级别的应用程序的框架。它通过提供一种快速、方便的方式来构建应用程序，可以大大提高开发效率和应用程序的质量。

WebSocket是一种基于TCP协议的全双工通信协议，在浏览器和服务器之间建立一个持久化的连接，实现实时数据的传输。在Web应用程序中，WebSocket通常用于实时消息推送、在线聊天、游戏等场景。

下面是使用Spring Boot实现WebSocket消息提示的步骤：

1. 添加依赖

在pom.xml文件中添加以下依赖：

<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2. 创建WebSocket处理器

创建一个类，实现WebSocketHandler接口，用于处理WebSocket连接、发送消息等操作。例如：

@Component
public class MyWebSocketHandler implements WebSocketHandler {

  private List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();

  @Override
  public void afterConnectionEstablished(WebSocketSession session) throws Exception {
    sessions.add(session);
  }

  @Override
  public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
    // 处理收到的消息
  }

  @Override
  public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
    sessions.remove(session);
  }

  @Override
  public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
    sessions.remove(session);
  }

  public void sendMessage(String message) {
    for (WebSocketSession session : sessions) {
      try {
        session.sendMessage(new TextMessage(message));
      } catch (IOException e) {
        e.printStackTrace();
      }
    }
  }
}

这个处理器维护了一个WebSocketSession列表，处理连接建立、消息处理、连接关闭、异常处理等操作。其中sendMessage方法用于向所有连接的客户端发送消息。

3. 配置WebSocket

在Spring Boot的配置类中添加WebSocket配置。例如：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

  @Autowired
  private MyWebSocketHandler myWebSocketHandler;

  @Override
  public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
    registry.addHandler(myWebSocketHandler, "/ws").setAllowedOrigins("*");
  }
}

这里使用@EnableWebSocket注解开启WebSocket支持，注册MyWebSocketHandler处理器，并设置访问路径和跨域。

4. 发送消息

在需要发送WebSocket消息的地方，注入MyWebSocketHandler，并调用sendMessage方法发送消息。例如：

@Component
public class MyComponent {

  @Autowired
  private MyWebSocketHandler myWebSocketHandler;

  public void sendWebSocketMessage(String message) {
    myWebSocketHandler.sendMessage(message);
  }
}

这里的sendWebSocketMessage方法调用MyWebSocketHandler的sendMessage方法，向所有连接的客户端发送消息。

至此，使用Spring Boot实现WebSocket消息提示的步骤完成。通过WebSocket连接，可以实现实时消息推送、在线聊天、游戏等功能。