WebSocket——学习

WebSocket 是一种全双工通信协议，允许在客户端和服务器之间进行实时的双向通信。它是在 HTML5 标准中引入的，旨在解决传统 HTTP 协议在实时通信中的一些限制。

WebSocket 的基本概念

什么是 WebSocket？

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。它与 HTTP 协议的主要区别在于 WebSocket 允许服务器主动向客户端推送消息，而不仅仅是响应客户端的请求。

WebSocket 的优点

• 低延迟：WebSocket 是一种持久化的连接，客户端和服务器之间只需要一次握手，之后的数据交换都通过这个单一连接进行，减少了延迟。
• 双向通信：允许服务器和客户端随时发送消息，适合实时应用。
• 减少开销：WebSocket 连接建立后，通信报文头信息较小，减少了数据传输的开销。

WebSocket 的工作原理

握手过程

WebSocket 连接是通过标准 HTTP/HTTPS 请求发起的。客户端发送一个带有特定头信息的 HTTP 请求到服务器，服务器在接受并识别到这个请求是 WebSocket 请求后，会返回一个 HTTP 101 状态码响应，表示协议切换成功。

客户端握手请求：

GET /chat HTTP/1.1
Host: server.example.com
Upgrade: websocket
Connection: Upgrade
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
Sec-WebSocket-Version: 13

服务器握手响应：

HTTP/1.1 101 Switching Protocols
Upgrade: websocket
Connection: Upgrade
Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=

数据传输

握手完成后，WebSocket 连接建立，客户端和服务器可以相互发送消息。消息可以是文本或二进制数据。WebSocket 数据帧具有较小的头信息，减少了传输开销。

示例数据帧：

• 帧头信息包含了帧的类型（文本、二进制、关闭、Ping、Pong 等）和长度等信息。
• 数据部分包含实际的消息内容。

WebSocket API

JavaScript WebSocket API

WebSocket 在浏览器中通过 JavaScript API 使用。以下是基本用法示例：

// 创建 WebSocket 连接
const socket = new WebSocket('ws://localhost:8080/ws');

// 连接成功回调
socket.onopen = function(event) {
    console.log('Connected to WebSocket server.');
    socket.send('Hello Server!');
};

// 接收消息回调
socket.onmessage = function(event) {
    console.log('Received message from server:', event.data);
};

// 连接关闭回调
socket.onclose = function(event) {
    console.log('WebSocket connection closed.');
};

// 错误处理回调
socket.onerror = function(error) {
    console.error('WebSocket error:', error);
};

WebSocket 服务器端实现

使用 Java 实现 WebSocket 服务器

Java 提供了多种实现 WebSocket 服务器的方法，包括原生的 javax.websocket API 和基于 Netty 的实现。

使用 javax.websocket API：

@ServerEndpoint("/ws")
public class WebSocketServer {
    
    @OnOpen
    public void onOpen(Session session) {
        System.out.println("Connected: " + session.getId());
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("Received: " + message);
        // Echo the message back to the client
        session.getAsyncRemote().sendText("Echo: " + message);
    }

    @OnClose
    public void onClose(Session session, CloseReason closeReason) {
        System.out.println("Disconnected: " + session.getId());
    }

    @OnError
    public void onError(Session session, Throwable throwable) {
        throwable.printStackTrace();
    }
}

使用 Netty 实现 WebSocket 服务器

Netty 是一个高性能的网络应用框架，适合实现复杂的 WebSocket 服务器。

Netty WebSocket 服务器示例：

public class NettyWebSocketServer {
    public static void main(String[] args) throws InterruptedException {
        EventLoopGroup bossGroup = new NioEventLoopGroup(1);
        EventLoopGroup workerGroup = new NioEventLoopGroup();

        try {
            ServerBootstrap serverBootstrap = new ServerBootstrap();
            serverBootstrap.group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) {
                            ChannelPipeline pipeline = ch.pipeline();
                            pipeline.addLast(new HttpServerCodec());
                            pipeline.addLast(new HttpObjectAggregator(64 * 1024));
                            pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
                            pipeline.addLast(new SimpleChannelInboundHandler<TextWebSocketFrame>() {
                                @Override
                                protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) {
                                    ctx.channel().writeAndFlush(new TextWebSocketFrame("Hello! " + msg.text()));
                                }

                                @Override
                                public void handlerAdded(ChannelHandlerContext ctx) {
                                    System.out.println("Client connected: " + ctx.channel().id());
                                }

                                @Override
                                public void handlerRemoved(ChannelHandlerContext ctx) {
                                    System.out.println("Client disconnected: " + ctx.channel().id());
                                }

                                @Override
                                public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
                                    cause.printStackTrace();
                                    ctx.close();
                                }
                            });
                        }
                    });

            ChannelFuture channelFuture = serverBootstrap.bind(8080).sync();
            channelFuture.channel().closeFuture().sync();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }
}

WebSocket 应用场景

WebSocket 适用于需要实时双向通信的场景，包括但不限于：

• 实时聊天应用：如微信、Slack、WhatsApp 等。
• 在线游戏：需要实时交互的多人游戏。
• 实时通知系统：如股票价格、新闻推送等。
• 实时协作应用：如在线文档编辑、协作白板等。

总结

WebSocket 是一种高效的双向通信协议，适用于需要实时数据交互的应用场景。它通过建立持久的连接，减少了网络延迟和开销，提供了更流畅的用户体验。WebSocket 在浏览器中可以通过 JavaScript API 使用，在服务器端可以使用各种编程语言和框架实现。通过 WebSocket，可以轻松构建实时聊天、在线游戏、实时通知等应用。