基于netty的webstock

### 使用 Netty 实现 WebSocket 功能 #### 添加依赖 为了使用 Netty 构建 WebSocket 应用程序，需先在项目的 `pom.xml` 文件中加入必要的依赖项。对于 Maven 项目而言，应包含如下所示的配置： ```xml <dependencies> <!-- Netty All --> <dependency> <groupId>io.netty</groupId> <artifactId>netty-all</artifactId> <version>4.1.77.Final</version> </dependency> <!-- Spring Boot WebSocket Starter (如果需要集成Spring框架)--> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> </dependencies> ``` 上述代码片段展示了如何通过 Maven 来获取最新版本的 Netty 和 Spring Boot 的 WebSocket 支持[^2]。 #### 创建 WebSocket 客户端和服务端 ##### 初始化服务器端口并启动监听 创建一个类来初始化 Netty 并设置好用于处理 HTTP 请求升级到 WebSocket 连接的相关参数。下面是一个简单的例子展示怎样建立一个基本的服务端实例： ```java import io.netty.bootstrap.ServerBootstrap; import io.netty.channel.ChannelFuture; import io.netty.channel.EventLoopGroup; import io.netty.channel.nio.NioEventLoopGroup; import io.netty.channel.socket.nio.NioServerSocketChannel; public class WSServer { private final int port; public static void main(String[] args) throws Exception { new WSServer(8080).start(); } public WSServer(int port){ this.port = port; } public void start() throws InterruptedException{ EventLoopGroup bossGroup = new NioEventLoopGroup(); // 负责接收连接请求 EventLoopGroup workerGroup = new NioEventLoopGroup(); // 处理已接受的连接 try { ServerBootstrap b = new ServerBootstrap() .group(bossGroup, workerGroup) .channel(NioServerSocketChannel.class); ChannelFuture f = b.bind(port).sync(); System.out.println("WebSocket server started at ws://localhost:" + port + '/'); f.channel().closeFuture().sync(); } finally { bossGroup.shutdownGracefully(); workerGroup.shutdownGracefully(); } } } ``` 这段 Java 代码定义了一个名为 `WSServer` 的类，该类负责启动 WebSocket 服务端，并绑定指定端口号等待客户端发起握手请求[^1]。 ##### 配置管道处理器 为了让 Netty 正确解析 WebSocket 帧数据包以及管理会话状态，还需要自定义通道初始化器以添加特定于 WebSocket 协议的消息解码器和编码器等组件。这通常涉及到继承 `ChannelInitializer` 类重写其中的方法完成相应逻辑编写工作。 ```java import io.netty.channel.ChannelInitializer; import io.netty.channel.ChannelPipeline; import io.netty.channel.socket.SocketChannel; import io.netty.handler.codec.http.HttpObjectAggregator; import io.netty.handler.codec.http.HttpServerCodec; import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler; import io.netty.handler.stream.ChunkedWriteHandler; public class WSChannelInitializer extends ChannelInitializer<SocketChannel> { @Override protected void initChannel(SocketChannel ch) throws Exception { ChannelPipeline pipeline = ch.pipeline(); // Http编解码器 pipeline.addLast(new HttpServerCodec()); // 分块写入支持大文件传输 pipeline.addLast(new ChunkedWriteHandler()); // 将HTTP消息聚合为FullHttpRequest/FullHttpResponse对象 pipeline.addLast(new HttpObjectAggregator(65536)); // WebSocket协议处理器 String websocketPath = "/ws"; pipeline.addLast(new WebSocketServerProtocolHandler(websocketPath)); // 自定义业务逻辑处理器 pipeline.addLast(new MyWebSocketFrameHandler()); } } ``` 此部分实现了对传入的数据流进行预处理的功能，确保能够按照 WebSocket 规范正确地交换信息。 ##### 编写业务逻辑处理器 最后一步就是实现具体的通信行为——即当收到新消息时要执行的操作。可以创建一个新的 Handler 继承自 `SimpleChannelInboundHandler<WebSocketFrame>` 或者其他合适的基类来进行定制化开发。 ```java import io.netty.channel.ChannelHandlerContext; import io.netty.channel.SimpleChannelInboundHandler; import io.netty.handler.codec.http.websocketx.TextWebSocketFrame; import io.netty.handler.codec.http.websocketx.WebSocketFrame; public class MyWebSocketFrameHandler extends SimpleChannelInboundHandler<WebSocketFrame> { @Override protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception { if(frame instanceof TextWebSocketFrame){ String request = ((TextWebSocketFrame)frame).text(); System.out.println("Received message from client: " + request); // Echo back the received text to the sender. ctx.channel().writeAndFlush(new TextWebSocketFrame("Echo: "+request)); }else{ String errorMessage="Unsupported frame type!"; System.err.println(errorMessage); ctx.channel().writeAndFlush(new TextWebSocketFrame(errorMessage)); } } @Override public void handlerAdded(ChannelHandlerContext ctx)throws Exception { super.handlerAdded(ctx); System.out.println("New connection established."); } @Override public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception { cause.printStackTrace(); ctx.close(); } } ``` 以上便是整个基于 Netty 框架搭建 WebSocket 服务端的过程概述。当然实际应用可能更加复杂，涉及更多细节上的调整优化。