Netty Websocket

最新推荐文章于 2024-07-31 15:05:27 发布

王小工

最新推荐文章于 2024-07-31 15:05:27 发布

阅读量1.3k

点赞数 32

分类专栏： java 文章标签： websocket java nio

本文链接：https://blog.csdn.net/mqiqe/article/details/140483440

版权

java 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

一、WebSocket 协议概述

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。它允许服务端主动向客户端推送数据，从而实现了实时通信。WebSocket 建立在 HTTP 之上，但与 HTTP 的轮询（Polling）和长轮询（Long Polling）相比，WebSocket 只需一次握手，即可在客户端和服务器之间建立持久的连接，并通过这个连接进行双向数据传输。

二、Netty 框架简介

Netty 是一个高性能、异步事件驱动的网络应用程序框架，支持快速开发可维护的高性能协议服务器和客户端。它基于 Java NIO（非阻塞 I/O）进行封装，提供了更简单易用的 API，并解决了 NIO 编程中常见的复杂性和错误。

三、Netty WebSocket 原理

1. 握手过程

客户端发起握手请求：客户端通过 HTTP 请求与服务器建立 WebSocket 连接。这个请求与普通的 HTTP 请求不同，它包含了 WebSocket 特有的升级请求头（如 Upgrade: websocket 和 Connection: Upgrade），以及一个用于验证的 Sec-WebSocket-Key。
服务器响应握手请求：服务器接收到客户端的握手请求后，会解析这些请求头，并生成一个响应。响应中包含了 Sec-WebSocket-Accept 头，该头的值是使用 Sec-WebSocket-Key 加上一个固定的字符串（如 “258EAFA5-E914-47DA-95CA-C5AB0DC85B11”），然后通过 SHA-1 加密并 Base64 编码得到的。如果服务器同意升级协议，则响应的状态码为 101（Switching Protocols）。
握手成功：一旦服务器发送了包含 Sec-WebSocket-Accept 的响应，并且客户端验证了该响应的正确性，WebSocket 连接就建立起来了。此时，客户端和服务器之间就可以通过 TCP 连接进行双向数据交换了。

2. 数据传输

帧（Frame）：在 WebSocket 协议中，数据是通过帧来传输的。一个帧可以包含文本数据、二进制数据或控制帧（如关闭帧）。
ChannelHandler：Netty 使用 ChannelHandler 来处理网络事件和数据。在 WebSocket 应用中，通常会定义一系列的 ChannelHandler 来处理 WebSocket 的特定事件，如握手、消息接收、消息发送等。
ChannelPipeline：Netty 使用 ChannelPipeline 来管理 ChannelHandler 的链式调用。每个 Handler 都可以对经过的数据或事件进行处理，并决定是否将其传递给链中的下一个 Handler。

3. Netty 的线程模型

EventLoopGroup：Netty 使用 EventLoopGroup 来管理一组 EventLoop。每个 EventLoop 都是一个不断循环执行任务的线程。在 WebSocket 服务器中，通常会使用两个 EventLoopGroup：一个是用于接收客户端连接的 BossGroup，另一个是用于处理网络读写的 WorkerGroup。
ChannelPipeline 中的处理流程：当数据到达时，Netty 会将其封装成一个 ChannelHandlerContext 对象，并在 ChannelPipeline 中传递。每个 ChannelHandler 都可以对这个对象进行处理，并可以决定是否将其传递给链中的下一个 Handler。

四、Netty WebSocket 的优势

高性能：Netty 基于 NIO 进行封装，提供了非阻塞的 I/O 操作，能够处理大量的并发连接。
易用性：Netty 提供了丰富的编码解码器和 Handler，使得开发者可以很容易地实现 WebSocket 的功能。
稳定性：Netty 在稳定性和故障恢复方面有出色的表现，能够确保 WebSocket 连接的稳定性和可靠性。

五、Netty WebSocket 的主要组成部分

ChannelHandler：Netty 使用 ChannelHandler 来处理网络事件，如连接建立、数据读写等。在 WebSocket 应用中，你会定义一系列的 ChannelHandler 来处理 WebSocket 的特定事件，如握手（Handshake）、消息接收、消息发送等。
ChannelPipeline：Netty 使用 ChannelPipeline 来管理 ChannelHandler 的链式调用。你可以将多个 ChannelHandler 添加到同一个 ChannelPipeline 中，每个 Handler 都可以对经过的数据或事件进行处理，并决定是否将其传递给链中的下一个 Handler。
WebSocketServerProtocolHandler：这是 Netty 提供的专门用于处理 WebSocket 协议的 Handler。它负责处理 WebSocket 的握手请求，并在握手成功后将后续的 HTTP 帧转换为 WebSocket 帧。
WebSocketVersion 和 WebSocketSubprotocol：Netty 允许你指定 WebSocket 的版本（如 RFC6455, 也就是 WebSocket 1.0）以及子协议（如自定义的协议标识符）。

六、Netty WebSocket 应用场景

Netty WebSocket 的应用场景非常广泛，主要集中在需要实时、双向通信的 web 应用中。以下是一些典型的应用场景：

1. 即时聊天

应用描述：构建实时聊天应用，用户可以实时发送和接收消息，实现低延迟、高效的在线交流。
优势：WebSocket 提供了持久连接和双向通信的能力，使得聊天应用能够实时传输消息，减少延迟，提升用户体验。

2. 金融市场实时数据推送

应用描述：股票、外汇、期货等金融市场的实时报价、交易提醒等。
优势：金融市场数据变化迅速，WebSocket 能够实现服务器主动推送数据给客户端，确保用户能够实时获取最新的市场动态。

3. 新闻与社交媒体实时推送

应用描述：新闻、社交媒体的实时推送通知，如微博、今日头条等平台的实时更新。
优势：通过 WebSocket，新闻和社交媒体平台可以实时向用户推送最新的内容，提高用户粘性和活跃度。

4. 物联网（IoT）设备监控与远程控制

应用描述：智能家居、工业自动化等物联网设备的状态监控与远程控制。
优势：物联网设备通常需要实时上传数据并接收控制指令，WebSocket 的实时性和双向通信能力使其成为物联网通信的理想选择。

5. 协作工具

应用描述：在线文档编辑、白板绘图、代码协作等需要多方实时同步内容的应用。
优势：协作工具需要实时共享和同步数据，WebSocket 能够实现多方之间的实时通信和数据交换，提高协作效率。

6. 游戏

应用描述：多人在线游戏中的实时状态同步、玩家交互等。
优势：游戏需要低延迟的实时通信来确保玩家之间的同步和交互，WebSocket 的实时性和高效性使其成为游戏开发的常用技术。

7. 实时位置追踪与导航

应用描述：实时位置追踪、导航应用中的动态路线更新等。
优势：通过 WebSocket，导航应用可以实时接收用户的位置信息，并根据实时交通状况动态更新路线，提高导航的准确性和实用性。

8. 直播互动

应用描述：直播平台的实时评论、弹幕、礼物赠送等互动功能。
优势：直播需要实时处理大量的用户互动数据，WebSocket 能够实现服务器与客户端之间的实时通信，确保观众能够实时参与直播互动。

9. 数据分析与监控

应用描述：实时仪表盘、日志流处理、性能监控系统的实时数据展示与报警。
优势：在数据分析和监控领域，实时性非常重要。WebSocket 能够实现数据的实时传输和展示，帮助用户及时发现问题并采取相应的措施。

七、样例

在Netty中实现WebSocket的Demo可以分为服务端和客户端两部分。以下是一个简化的Netty WebSocket Demo示例，展示了如何搭建一个基本的WebSocket服务器和客户端。

maven

确保你已经将Netty的依赖项添加到你的项目中。如果你使用Maven，可以在pom.xml文件中添加以下依赖：

<dependencies>
    <dependency>
        <groupId>io.netty</groupId>
        <artifactId>netty-all</artifactId>
        <version>4.1.x</version> <!-- 使用你需要的Netty版本 -->
    </dependency>
</dependencies>

Netty WebSocket

WebSocket服务器的代码

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

public class WebSocketServer {

    private int port;

    public WebSocketServer(int port) {
        this.port = port;
    }

    public void run() throws Exception {
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup)
             .channel(NioServerSocketChannel.class)
             .childHandler(new ChannelInitializer<SocketChannel>() {
                 @Override
                 public void initChannel(SocketChannel ch) throws Exception {
                     ChannelPipeline pipeline = ch.pipeline();

                     // HTTP编解码器
                     pipeline.addLast(new HttpServerCodec());

                     // 聚合HTTP消息
                     pipeline.addLast(new HttpObjectAggregator(65536));

                     // 支持大消息的写操作
                     pipeline.addLast(new ChunkedWriteHandler());

                     // WebSocket协议处理
                     pipeline.addLast(new WebSocketServerProtocolHandler("/ws")); // 指定WebSocket的路径为"/ws"

                     // 自定义的WebSocket业务处理
                     pipeline.addLast(new WebSocketServerHandler());
                 }
             });

            // 开始监听端口
            ChannelFuture f = b.bind(port).sync();

            // 等待服务器套接字关闭
            f.channel().closeFuture().sync();
        } finally {
            // 关闭所有的事件循环以终止所有的线程
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }

    public static void main(String[] args) throws Exception {
        int port;
        if (args.length > 0) {
            port = Integer.parseInt(args[0]);
        } else {
            port = 8080;
        }
        new WebSocketServer(port).run();
    }

    private static class WebSocketServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {

        @Override
        protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
            // 打印接收到的消息
            System.out.println("Received message: " + msg.text());

            // 回显消息
            ctx.channel().writeAndFlush(new TextWebSocketFrame("Echo: " + msg.text()));
        }

        @Override
        public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
            // 当handler被添加到ChannelPipeline时调用
            System.out.println("WebSocket client connected: " + ctx.channel().remoteAddress());
        }

        @Override
        public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
            // 当handler从ChannelPipeline中移除时调用
            System.out.println("WebSocket client disconnected: " + ctx.channel().remoteAddress());
        }

        @Override
        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
            // 当异常被抛出时调用
            cause.printStackTrace();
            ctx.close();
        }
    }
}