Web Socket简单尝试，实现服务器向客户端的主动消息推送

2025-04-29 19:00:10 0 阅读

使用Web Socket的原因

首先，在常规情况下，我们使用的接口是http协议，在这种情况下，只有客户端向服务器发送请求的情况下，服务器才能向客户端发送消息。但是在一些特殊场景下，服务器是需要向客户端主动发送消息，例如聊天、推送信息等功能。

在这种情况下我们就需要用到Web Socket来实现我们的需求，Web Socket是全双工通信，允许客户端和服务器同时进行数据发送和接收，这使得交互更加高效。

原理：

握手：WebSocket首先通过HTTP协议进行握手，成功后会升级为WebSocket协议。
数据传输：连接建立后，客户端和服务器可以自由地交换信息，直到某一方主动关闭连接。

工作过程：

1. 握手过程

WebSocket的连接是通过HTTP协议进行握手的。这个过程包括以下几个步骤：

客户端发起请求：客户端发送一个HTTP请求到服务器，请求升级到WebSocket协议。这个请求包含一些特殊的头部字段（详细可以看Web Socket协议）
服务器响应：如果服务器支持WebSocket协议，它会返回一个HTTP 101状态码，表示协议切换。响应头中也会包含一些信息。

2. 建立持久连接

一旦握手成功，WebSocket会在客户端和服务器之间建立一个持久的TCP连接。这意味着在这个连接建立后，数据可以随时从客户端发送到服务器，反之亦然，而不需要重新建立连接。

3. 数据传输

WebSocket使用帧（frames）来传输数据。数据帧的格式相对简单，包含：

帧头：指示帧的类型（文本帧、二进制帧、关闭帧等）。
有效载荷：实际要传输的数据。

WebSocket允许消息在两端同时流动，双方可以随时发送消息。

4. 关闭连接

当一方要关闭连接时，会发送一个关闭帧，包括一个状态码（如1000表示正常关闭）。另一方收到后也会发送一个关闭帧，完成关闭操作。

个人的简单实现

本文章的实现仅为个人的见解，如有谬误请指正

第一步建立一个SpringBoot项目

SpringBoot中提供了Web Socket插件，我们可以直接使用这个

在这里我们使用了Lombok、Spring Web、WebSocket

创建包

名字可以自己起，只要自己找得到就行

编写两个类，

第一个类继承TextWebSocketHandler类，这个类主要用来实现类中的方法

import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

/**
 * @Author: 百花杀
 * @Motto: LESS TALKING、MORE BUILDING、MORE THINkING
 * @Date: 2024/11/26/16:19
 */
public class MyWebSocketHandler extends TextWebSocketHandler {
    @Override
//    这个方法是WebSocket连接建立成功后调用，在这里可以做一些初始化操作
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("Web Socket 连接session : " + session.getId() + " 连接成功");
    }

    @Override
//    这个方法是处理WebSocket消息的，在这里可以接收客户端发送的消息并进行处理
    public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理接收到的消息
        String incomingMessage = message.getPayload();
        System.out.println("接收到的消息：" + incomingMessage);

        // 回复消息
        session.sendMessage(new TextMessage("服务器接收到消息： " + incomingMessage + " 这是服务器返回的消息： 百花杀"));
    }

    @Override
//    这个方法是WebSocket连接关闭时调用，在这里可以做一些清理操作
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        System.out.println("Web Socket 连接session : " + session.getId() + " 连接关闭");
    }
}

这样我们就实现了处理类，接下来我们来实现配置类，配置类需要实现WebSocketConfigurer接口

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

/**
 * @Author: 百花杀
 * @Motto: LESS TALKING、MORE BUILDING、MORE THINkING
 * @Date: 2024/11/26/16:19
 */
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    // 这里我们注册了一个MyWebSocketHandler，它是WebSocketHandler的一个实现类，负责处理WebSocket请求。
    // 本例中，websocket请求路径为/WEBSOCKET/ws,setAllowedOrigins("*")表示允许所有来源的WebSocket请求,可以不用处理跨域问题。
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myWebSocketHandler(), "/WEBSOCKET/ws").setAllowedOrigins("*");
    }

    public MyWebSocketHandler myWebSocketHandler() {
        return new MyWebSocketHandler();
    }
}

这样的话我们就实现了服务器的基础配置，接下来是前端部分：

我使用的是Vue2来实现这个功能