Netty编码实战与Channel生命周期
本次将搭建一个最简单的Hello Netty服务器,并且通过这个简单的示例了解了Channel的生命周期。最后将基于Netty搭建一个Websocket网页聊天小程序,可以使用户在Web浏览器或者移动端浏览器进行消息的收发,来深入体会一下使用Netty编码NIO服务器是多么便捷。
Hello Netty服务器
- 构建一对主从线程组
- 定义服务器启动类
- 为服务器设置Channel
- 设置处理从线程池的助手类初始化器
- 监听启动和关闭服务器
1、构建主从线程组与服务启动类
首先新建一个Maven工程,引入Netty的依赖,我引入的依赖如下:
1<dependencies>
2 <dependency>
3 <groupId>io.netty</groupId>
4 <artifactId>netty-all</artifactId>
5 <version>4.1.50.Final</version>
6 </dependency>
7</dependencies>
2、设置Channel初始化器
每一个channel由多个handler共同组成管道(pipeline)
3、开始编写自定义的助手类
然后接下来启动服务器,通过Postman访问一下http://localhost:8080
得到如下结果:
如果直接在浏览器端访问的话会打印两次客户端远程地址,因为浏览器默认还访问了http://localhost:8080/favicon.ico
,或者使用在Linux环境下使用curl进行测试也是可以的。
探究Channel生命周期
我们通过重写下图所示的方法来研究一下Channel的生命周期(IDEA快捷键 Ctrl + O):
重写完成之后的CustomHandler如下:
1public class CustomHandler extends SimpleChannelInboundHandler<HttpObject> {
2 protected void channelRead0(ChannelHandlerContext ctx, HttpObject msg) throws Exception {
3 // 获取Channel
4 Channel channel = ctx.channel();
5 if(msg instanceof HttpRequest) {
6 // 显示客户端的远程地址
7 System.out.println(channel.remoteAddress());
8 // 数据Copy至缓冲区(定义发送的数据消息)
9 ByteBuf content = Unpooled.copiedBuffer("<h1>Hello Netty</h1>", CharsetUtil.UTF_8);
10 // 构建一个Http Response
11 FullHttpResponse response = new DefaultFullHttpResponse(
12 HttpVersion.HTTP_1_1, HttpResponseStatus.OK, content);
13 // 为响应增加一个数据类型和长度
14 response.headers().set(HttpHeaderNames.CONTENT_TYPE, "text.plain");
15 response.headers().set(HttpHeaderNames.CONTENT_LENGTH, content.readableBytes());
16 // 把响应刷到客户端
17 ctx.writeAndFlush(response);
18 }
19 }
20
21 @Override
22 public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
23 System.out.println("Channel-注册");
24 super.channelRegistered(ctx);
25 }
26
27 @Override
28 public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
29 System.out.println("Channel-移除");
30 super.channelUnregistered(ctx);
31 }
32
33 @Override
34 public void channelActive(ChannelHandlerContext ctx) throws Exception {
35 System.out.println("Channel-活跃");
36 super.channelActive(ctx);
37 }
38
39 @Override
40 public void channelInactive(ChannelHandlerContext ctx) throws Exception {
41 System.out.println("Channel-不活跃(断开了)");
42 super.channelInactive(ctx);
43 }
44
45 @Override
46 public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
47 System.out.println("Channel-读取数据完毕");
48 super.channelReadComplete(ctx);
49 }
50
51 @Override
52 public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
53 System.out.println("用户事件触发");
54 super.userEventTriggered(ctx, evt);
55 }
56
57 @Override
58 public void channelWritabilityChanged(ChannelHandlerContext ctx) throws Exception {
59 System.out.println("Channel-可写更改");
60 super.channelWritabilityChanged(ctx);
61 }
62
63 @Override
64 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
65 System.out.println("捕获到异常");
66 super.exceptionCaught(ctx, cause);
67 }
68
69 @Override
70 public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
71 System.out.println("助手类添加");
72 super.handlerAdded(ctx);
73 }
74
75 @Override
76 public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
77 System.out.println("助手类移除");
78 super.handlerRemoved(ctx);
79 }
80}
通过Curl访问控制台打印如下:
为什么要用CURL而不是浏览器或者PostMan呢?因为我们使用了HTTP1.1的版本,支持长连接,而且默认是开启状态,所以看不到Channel不活跃断开的状态,所以才使用CURL来访问。
网页版的WebSocket聊天室
1、Netty 服务器编码
Netty 服务器启动类WSServe.java如下:
1import io.netty.bootstrap.ServerBootstrap;
2import io.netty.channel.ChannelFuture;
3import io.netty.channel.EventLoopGroup;
4import io.netty.channel.nio.NioEventLoopGroup;
5import io.netty.channel.socket.nio.NioServerSocketChannel;
6
7public class WSServer {
8 public static void main(String[] args) throws Exception {
9 EventLoopGroup bossGroup = new NioEventLoopGroup();
10 EventLoopGroup workerGroup = new NioEventLoopGroup();
11
12 try{
13 ServerBootstrap serverBootstrap = new ServerBootstrap();
14 serverBootstrap.group(bossGroup, workerGroup)
15 .channel(NioServerSocketChannel.class)
16 .childHandler(new WSServerInitializer());
17
18 ChannelFuture channelFuture = serverBootstrap.bind(8080).sync();
19 channelFuture.channel().closeFuture().sync();
20 } finally {
21 bossGroup.shutdownGracefully();
22 workerGroup.shutdownGracefully();
23 }
24 }
25}
接下来是Channel初始化器WSServerInitializer.java
1import io.netty.channel.ChannelInitializer;
2import io.netty.channel.ChannelPipeline;
3import io.netty.channel.socket.SocketChannel;
4import io.netty.handler.codec.http.HttpObjectAggregator;
5import io.netty.handler.codec.http.HttpServerCodec;
6import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
7import io.netty.handler.stream.ChunkedWriteHandler;
8
9public class WSServerInitializer extends ChannelInitializer<SocketChannel> {
10 @Override
11 protected void initChannel(SocketChannel socketChannel) throws Exception {
12 ChannelPipeline pipeline = socketChannel.pipeline();
13 // WebSocket基于Http协议,添加Http编解码器
14 pipeline.addLast(new HttpServerCodec());
15
16 // 添加对写大数据流的支持
17 pipeline.addLast(new ChunkedWriteHandler());
18
19 // 对Http Message进行聚合,聚合成FullHttpRequest或FullHttpResponse
20 // 几乎在Netty中的编程都会使用到此Handler
21 pipeline.addLast(new HttpObjectAggregator(1024 * 64));
22
23 //-------------------- 以上是用于支持HTTP协议 ----------------------
24
25 // WebSocket服务器处理的协议,并且指定给客户端链接访问的路由
26 // 使用此Handler会直接帮你处理握手动作(Close、Ping、Pong)
27 // 对于WebSocket,都是以帧进行传输的,不同数据对应的帧也不同
28 pipeline.addLast(new WebSocketServerProtocolHandler("/ws"));
29
30 // 自定义的Handler
31 pipeline.addLast(new ChatHandler());
32 }
33}
最后是自定义的Handler,ChatHandler.java
1import io.netty.channel.Channel;
2import io.netty.channel.ChannelHandlerContext;
3import io.netty.channel.SimpleChannelInboundHandler;
4import io.netty.channel.group.ChannelGroup;
5import io.netty.channel.group.DefaultChannelGroup;
6import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
7import io.netty.util.concurrent.GlobalEventExecutor;
8
9import java.time.LocalDateTime;
10
11// 对于WebSocket,都是以帧进行传输的,不同数据对应的帧也不同 -> TextWebSocketFrame
12// TextWebSocketFrame是WebSocket专门用于处理文本的对象,Frame是消息的载体
13public class ChatHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {
14
15 // 用于记录和管理所有客户端的Channel
16 private static ChannelGroup clients = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
17
18 @Override
19 protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
20 // 从客户端获取到的消息内容
21 String content = msg.text();
22 System.out.println("接收到的数据:" + content);
23 String message = "[服务器收到消息] " + LocalDateTime.now() + "消息为:" + content;
24 for(Channel channel: clients){
25 //channel.writeAndFlush(content); ERROR 不能直接传String,而是TextWebSocketFrame载体
26 channel.writeAndFlush(new TextWebSocketFrame(message));
27 }
28
29 // 下面这种方式与For循环一致
30 //clients.writeAndFlush(new TextWebSocketFrame(message));
31 }
32
33
34 @Override
35 public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
36 Channel channel = ctx.channel();
37 // 当客户端打开链接后,获取客户端的Channel并且添加Channel至ChannelGroup中进行管理
38 clients.add(channel);
39 }
40
41 @Override
42 public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
43 // 当触发handlerRemoved,ChannelGroup会自动移除客户端的Channel
44 System.out.println("客户端断开, Channel对应的长ID:" + ctx.channel().id().asLongText());
45 System.out.println("客户端断开, Channel对应的短ID:" + ctx.channel().id().asShortText());
46 }
47}
2、前端JavaScript编码
下面是前端需要用到的WebSocket API:
1<html>
2 <head>
3 <meta charset="utf-8" />
4 <title></title>
5 </head>
6 <body>
7 <div>发送消息</div>
8 <input type="text" id="msgContent"/>
9 <input type="button" onclick="CHAT.chat()" value="发送"/>
10
11 <div>接收消息</div>
12 <div id="receiveMsg" style="background-color: darkturquoise;"></div>
13 <script type="application/javascript">
14 window.CHAT = {
15 socket: null,
16 init: function(){
17 if(window.WebSocket){
18 CHAT.socket = new WebSocket("ws://127.0.0.1:8080/ws");
19 CHAT.socket.onopen = function(){
20 console.log('连接建立成功...');
21 },
22 CHAT.socket.onclose = function(){
23 console.log('连接建立关闭...');
24 },
25 CHAT.socket.onerror = function(){
26 console.log('连接建立发生错误...');
27 },
28 CHAT.socket.onmessage = function(e){
29 console.log('收到消息...' + e.data);
30 var receiveMsg = document.getElementById('receiveMsg');
31 var html = receiveMsg.innerHTML;
32 receiveMsg.innerHTML = html + "<br/>" + e.data;
33 }
34 }else{
35 alert('不支持WebSocket');
36 }
37 },
38 chat: function(){
39 var msg = document.getElementById("msgContent");
40 CHAT.socket.send(msg.value);
41 }
42 }
43 CHAT.init();
44 </script>
45 </body>
46</html>
3、效果展示
Netty编码的小总结
首先是流程,先新建主从线程组,编写启动类,因为Netty官方推荐的模式也是主从线程模型。接下来是编写Channel初始化器,继承自ChannelInitializer,Channel注册后会执行里面的相应的初始化方法,通过Channel获取管道,然后添加需要的Handler,最后添加自己的自定义的Handler来处理请求。