(自用)WebSocket

最新推荐文章于 2025-10-31 15:16:11 发布

原创最新推荐文章于 2025-10-31 15:16:11 发布 · 633 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#websocket #网络协议 #网络

一,websocket是什么？

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。在介绍它之前，先来了解一下传统的 HTTP 通信方式。传统的 HTTP 协议是无状态的，客户端向服务器发送请求，服务器接收到请求后进行处理并返回响应，这种通信是单向的，即由客户端主动发起请求，服务器被动响应，服务器无法主动向客户端发送消息。而 WebSocket 则打破了这种限制，以下从几个方面详细介绍 WebSocket：

特点

全双工通信：在 WebSocket 连接建立后，客户端和服务器可以在任意时刻相互发送数据，无需客户端不断地发送请求来获取信息。比如在实时聊天应用中，当一方发送消息后，服务器能立刻将消息推送给另一方，无需接收方不断询问是否有新消息。
较少的开销：与 HTTP 请求相比，WebSocket 连接建立后的数据包头部开销小，HTTP 请求通常会携带大量的头部信息，而 WebSocket 只需要较小的协议头部，这在需要频繁交换数据的场景下，能显著减少网络带宽的消耗。
持久连接：WebSocket 连接一旦建立，除非一方主动关闭，否则连接会一直保持。这使得服务器可以在有新数据时立即推送给客户端，非常适合实时性要求高的应用，如股票行情实时更新、在线游戏等。

应用场景

实时通信：如在线聊天、即时通讯工具等，能够实现消息的即时发送和接收，让用户之间的交流更加流畅。
实时数据更新：像股票交易系统、体育赛事比分直播等，服务器可以实时将最新的数据推送给客户端，让用户及时获取最新信息。
多人协作：在多人在线文档编辑、在线游戏等场景中，多个用户的操作需要实时同步，WebSocket 可以确保数据的实时性和一致性。

工作原理

握手阶段：客户端向服务器发送一个 HTTP 请求，请求中包含一些特殊的头部信息，表明这是一个 WebSocket 连接请求。服务器收到请求后，如果支持 WebSocket 协议，会返回一个状态码为 101 的响应，表示同意升级协议。
数据传输阶段：握手成功后，HTTP 连接会升级为 WebSocket 连接，客户端和服务器可以通过这个连接进行全双工通信，数据以帧的形式在两者之间传输。
关闭连接阶段：当一方想要关闭连接时，会发送一个关闭帧给对方，对方收到关闭帧后，会发送一个确认关闭帧，然后双方关闭连接。

二，实现WebSocket代码

1，新建idea项目，导入spring boot，WebSocket依赖

<parent>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-parent</artifactId>
    <version>3.4.3</version>
</parent>
<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>
<build>
    <plugins>
        <plugin>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-maven-plugin</artifactId>
        </plugin>
    </plugins>
</build>

2，配置websocket端口：实现WebSocketConfigurer，加入注解@EnableWebSocket

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myWebSocketHandler(),  "/ws")
                .setAllowedOrigins("*"); // 允许跨域
    }
    @Bean
    public MyWebSocketHandler myWebSocketHandler() {
        return new MyWebSocketHandler();
    }
}

3，建立ws处理类，注意这里只处理文本类型

public class MyWebSocketHandler extends TextWebSocketHandler {
    //    private static final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
    private static final List<WebSocketSession> sessions = new ArrayList<>();
    @Override
    public void afterConnectionEstablished(WebSocketSession session) {
        sessions.add(session);
    }
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) {
        // 处理客户端消息
        System.out.println("收到===="+message.getPayload());
        //broadcast("Received: " + message.getPayload());
        try {
            session.sendMessage(new TextMessage("您好"));
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
    public static void broadcast(String message) {
        sessions.forEach(session  -> {
            try {
                session.sendMessage(new  TextMessage(message+"我的返回"));
            } catch (IOException e) {
                // 异常处理
            }
        });
    }
    ///加入了一个定时器，注意在启动类上加上@EnableScheduling才有效
    @Scheduled(fixedRate = 5000)
    public void sendUpdates() {
        for (WebSocketSession session : sessions) {
            try {
                session.sendMessage(new TextMessage("好的---收到"));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }

        }
//        MyWebSocketHandler.broadcast("Server  Time: " + new Date());
    }
}

4，在启动类上加上@EnableScheduling才有效启动定时器

@SpringBootApplication
@EnableScheduling
public class Main {
    public static void main(String[] args) {
        SpringApplication.run(Main.class, args);
    }
}

5，前端使用postman模拟即可，新建 websocket,,url 为ws://localhost:8080/ws，