大纲
1.Netty实现HTTP服务器
2.Netty实现WebSocket
3.Netty实现的消息推送系统
(1)基于WebSocket的消息推送系统说明
(2)消息推送系统的PushServer
(3)消息推送系统的连接管理封装
(4)消息推送系统的ping-pong探测
(5)消息推送系统的全连接推送
(6)消息推送系统的HTTP响应和握手
(7)消息推送系统的运营客户端
(8)运营客户端连接PushServer
(9)运营客户端的Handler处理器
(10)运营客户端发送推送消息
(11)浏览器客户端接收推送消息
1.Netty实现HTTP服务器
(1)HTTP请求消息和响应消息
(2)Netty实现的HTTP协议栈的优势
(3)Netty实现的HTTP服务器
(4)请求的解析处理和响应的编码处理
(1)HTTP请求消息和响应消息
HTTP请求消息由三部分组成：请求行、请求头、请求体。HTTP响应消息也由三部分组成：响应行、响应头、响应体。
(2)Netty实现的HTTP协议栈的优势
Netty实现的HTTP协议栈无论在性能还是在可靠性上，都表现优异，非常适合在非Web容器下的场景使用。相比于传统的Tomcat、Jetty等Web容器，它更加轻量和小巧，灵活性和定制性也更好。
(3)Netty实现的HTTP服务器
public class NettyHttpServer { private static final Logger logger = LogManager.getLogger(NettyHttpServer.class); private static final int DEFAULT_PORT = 8998; private int port; public NettyHttpServer(int port) { this.port = port; } public void start() throws Exception { logger.info("Netty Http Server is starting."); EventLoopGroup bossEventLoopGroup = new NioEventLoopGroup(); EventLoopGroup workerEventLoopGroup = new NioEventLoopGroup(); try { ServerBootstrap serverBootstrap = new ServerBootstrap(); serverBootstrap.group(bossEventLoopGroup, workerEventLoopGroup) .channel(NioServerSocketChannel.class) .childHandler(new ChannelInitializer<SocketChannel>() { protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline() //数据进来是自上而下，数据回去时自下而上 .addLast("http-decoder", new HttpRequestDecoder()) .addLast("http-aggregator", new HttpObjectAggregator(65536)) .addLast("http-encoder", new HttpResponseEncoder()) .addLast("http-chunked", new ChunkedWriteHandler()) .addLast("netty-http-server-handler", new NettyHttpServerHandler()); } }); ChannelFuture channelFuture = serverBootstrap.bind(port).sync(); logger.info("Netty Http Server is started, listened[" + port + "]."); channelFuture.channel().closeFuture().sync(); } finally { bossEventLoopGroup.shutdownGracefully(); workerEventLoopGroup.shutdownGracefully(); } } public static void main(String[] args) throws Exception { NettyHttpServer nettyHttpServer = new NettyHttpServer(DEFAULT_PORT); nettyHttpServer.start(); } } public class NettyHttpServerHandler extends SimpleChannelInboundHandler<FullHttpRequest> { private static final Logger logger = LogManager.getLogger(NettyHttpServerHandler.class); protected void channelRead0(ChannelHandlerContext ctx, FullHttpRequest request) throws Exception { logger.info(request); } }启动NettyHttpServer，然后在浏览器进行访问，就可以看到输出如下：
Netty Http Server is starting. Netty Http Server is started, listened[8998]. HttpObjectAggregator$AggregatedFullHttpRequest(decodeResult: success, version: HTTP/1.1, content: CompositeByteBuf(ridx: 0, widx: 0, cap: 0, components=0)) GET / HTTP/1.1 Host: localhost:8998 Connection: keep-alive sec-ch-ua: ".Not/A)Brand";v="99", "Google Chrome";v="103", "Chromium";v="103" sec-ch-ua-mobile: ?0 sec-ch-ua-platform: "macOS" Upgrade-Insecure-Requests: 1 User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36 Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9 Sec-Fetch-Site: none Sec-Fetch-Mode: navigate Sec-Fetch-User: ?1 Sec-Fetch-Dest: document Accept-Encoding: gzip, deflate, br Accept-Language: zh-CN,zh;q=0.9,en;q=0.8,id;q=0.7,ar;q=0.6 Cookie: _ga=GA1.1.629604539.1641093986 content-length: 0(4)请求的解析处理和响应的编码处理
HTTP服务器最关键的就是请求的解析处理和响应的编码处理，比如会向ChannelPipeline中先后添加不同的解码器和编码器。
首先添加HTTP请求消息解码器HttpRequestDecoder，因为浏览器会把按照HTTP协议组织起来的请求数据序列化成字节数组发送给服务器，而HttpRequestDecoder可以按照HTTP协议从接收到的字节数组中读取出一个完整的请求数据。
然后添加HttpObjectAggregator解码器，它的作用是将多个消息转换为单一的FullHttpRequest或者FullHttpResponse。
接着添加HTTP响应消息编码器HttpResponseEncoder，它的作用是对HTTP响应消息进行编码。
以及添加ChunkedWriteHandler处理器，用来支持异步发送大的码流时也不会占用过多的内存，防止内存溢出。
最后添加NettyHttpServerHandler处理器，用于处理HTTP服务器的响应输出。
public class NettyHttpServerHandler extends SimpleChannelInboundHandler<FullHttpRequest> { private static final Logger logger = LogManager.getLogger(NettyHttpServerHandler.class); protected void channelRead0(ChannelHandlerContext ctx, FullHttpRequest request) throws Exception { String method = request.getMethod().name(); String uri = request.getUri(); logger.info("Receives Http Request: " + method + " " + uri + "."); String html = "<html><body>Hello, I am Netty Http Server.</body></html>"; ByteBuf byteBuf = Unpooled.copiedBuffer(html, CharsetUtil.UTF_8); FullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.OK); response.headers().set("content-type", "text/html;charset=UTF-8"); response.content().writeBytes(byteBuf); byteBuf.release(); ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE); } }
2.Netty实现WebSocket
(1)HTTP协议的弊端
(2)消息推送之Ajax短轮询
(3)消息推送之WebSocket
(4)WebSocket连接的建立
(5)基于WebSocket协议开发NettyServer
(6)WebSocketServer的请求数据处理逻辑开发
(7)WebSocketServer的HTTP与chunk处理分析
(8)WebSocket网页客户端代码开发与实现
(1)HTTP协议的弊端
一.HTTP协议为半双工通信
半双工通信指数据可以在客户端和服务端两个方向传输，但是不能同时传输。它意味着同一时刻，只能有一个方向上的数据在进行传输。
二.HTTP消息冗长而繁琐
HTTP消息包含消息头、消息体、换行符等，通常情况下采用文本方式传输。相比于其他的二进制通信协议，冗长而繁琐。
(2)消息推送之Ajax短轮询
Ajax短轮询是基于HTTP短连接的。具体就是用一个定时器由浏览器对服务器发出HTTP请求，由服务器返回数据给客户端浏览器。
Ajax短轮询的代码实现简单。但由于HTTP请求的Header非常冗长，里面可用数据的比例非常低，所以比较占用带宽和服务器资源，且数据同步不及时。
(3)消息推送之WebSocket
WebSocket本质是一个TCP连接，采用的是全双工通信。
WebSocket中，浏览器和服务器只需要做一个握手动作，两者之间就会形成一条快速通道进行直接数据传输。由于WebSocket基于TCP双向全双工进行消息传递，所以在同一时刻既可以发送消息，也可以接收消息，比HTTP半双工性能好。
WebSocket通过ping/pong帧来保持链路激活，实时性很高，但是浏览器支持度低、代码实现复杂。
(4)WebSocket连接的建立
建立WebSocket连接时，需要通过客户端或者浏览器发出握手请求，这个握手请求是一个HTTP请求。该HTTP请求包含了附加头信息"Upgrade:WebSocket"，表明这是一个申请协议升级的HTTP请求。
服务端解析这些附加的头信息，然后生成应答信息返回给客户端。这样客户端和服务端的WebSocket连接就建立起来了，双方可以通过这个连接通道自由地传递信息，并且这个连接会持续存在直到一方主动关闭连接。
(5)基于WebSocket协议开发NettyServer
可以基于TCP协议用Netty来开发客户端和服务端进行相互通信，但粘包半包问题需要手动进行处理。
可以基于HTTP协议用Netty来开发一个HTTP服务器，服务器接收浏览器发送过来的HTTP请求后，返回HTTP响应回浏览器。
也可以基于WebSocket协议来开发一个Netty服务器，此时前端HTML代码会基于socket协议和Netty服务器建立长连接。这样Netty服务器就可以和浏览器里运行的HTML通过WebSocket协议建立长连接，从而使得Netty服务器可以主动推送数据给浏览器里的HTML页面。
WebSocket协议底层也是基于TCP协议来实现的，只不过是在TCP协议的基础上封装了一层更高层次的WebSocket协议。
public class NettyWebSocketServer { private static final Logger logger = LogManager.getLogger(NettyWebSocketServer.class); private static final int DEFAULT_PORT = 8998; private int port; public NettyWebSocketServer(int port) { this.port = port; } public void start() throws Exception { logger.info("Netty WebSocket Server is starting."); EventLoopGroup bossEventLoopGroup = new NioEventLoopGroup(); EventLoopGroup workerEventLoopGroup = new NioEventLoopGroup(); try { ServerBootstrap serverBootstrap = new ServerBootstrap(); serverBootstrap.group(bossEventLoopGroup, workerEventLoopGroup) .channel(NioServerSocketChannel.class) .childHandler(new ChannelInitializer<SocketChannel>() { protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline() .addLast(new HttpServerCodec()) .addLast(new ChunkedWriteHandler()) .addLast(new HttpObjectAggregator(1024 * 32)) .addLast(new WebSocketServerProtocolHandler("/websocket")) .addLast("netty-web-socket-server-handler", new NettyWebSocketServerHandler()); } }); ChannelFuture channelFuture = serverBootstrap.bind(port).sync(); logger.info("Netty WebSocket Server server is started, listened[" + port + "]."); channelFuture.channel().closeFuture().sync(); } finally { bossEventLoopGroup.shutdownGracefully(); workerEventLoopGroup.shutdownGracefully(); } } public static void main(String[] args) throws Exception { NettyWebSocketServer nettyHttpServer = new NettyWebSocketServer(DEFAULT_PORT); nettyHttpServer.start(); } }(6)WebSocketServer的请求数据处理逻辑开发
public class NettyWebSocketServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> { private static final Logger logger = LogManager.getLogger(NettyWebSocketServerHandler.class); private static ChannelGroup webSocketClients = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception { //WebSocket网页代码里发送过来的数据 String request = msg.text(); logger.info("Netty Server receives request: " + request + "."); TextWebSocketFrame response = new TextWebSocketFrame("Hello, I am Netty Server."); ctx.writeAndFlush(response); } //如果一个网页WebSocket客户端跟Netty Server建立了连接之后，会触发这个方法 @Override public void handlerAdded(ChannelHandlerContext ctx) throws Exception { webSocketClients.add(ctx.channel()); } @Override public void handlerRemoved(ChannelHandlerContext ctx) throws Exception { logger.info("websocket client is closed, channel id: " + ctx.channel().id().asLongText() + "[" + ctx.channel().id().asShortText() + "]"); } }(7)WebSocketServer的HTTP与chunk处理分析
浏览器里面运行的是HTML网页代码，WebSocket就是在HTML网页代码里嵌入WebSocket代码，可以让浏览器里的HTML网页代码跟NettyServer建立连接，并且是基于长连接发送数据。
浏览器会按照WebSocket协议(底层基于HTTP协议)组织请求数据格式，然后把数据序列化成字节数组，接着通过网络连接把字节数组传输发送给NettyServer，NettyServer会通过数据处理链条来进行处理接收到的字节数组数据。
protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline() //浏览器的字节数组数据进来以后，字节数组数据先用http协议来处理，把字节数组转换为一个HttpRequest请求对象 //最后数据返回给浏览器前，又会对HttpResponse对象进行编码成字节数组 .addLast(new HttpServerCodec()) //chunked write用于大量数据流时的分chunk块，也就是数据实在是太大了就必须得分chunk //大量数据流进来时，可以分chunk块来读；大量数据流出去时，可以分chunk块来写； .addLast(new ChunkedWriteHandler()) //如果想要让很多http不要拆分成很多段过来，可以把完整的请求数据聚合到一起再给过来 .addLast(new HttpObjectAggregator(1024 * 32)) //基于前面已经转换好的请求数据对象，会在这里基于WebSocket协议再次做一个处理 //由于传输时是基于http协议传输过来的，而封装的内容是按webSocket协议来封装的http请求数据 //所以必须在这里提取http请求里面的数据，然后按照WebSocket协议来进行解析处理，把数据提取出来作为WebSocket的数据片段 .addLast(new WebSocketServerProtocolHandler("/websocket")) //响应数据输出时，顺序是反的，第一步原始数据必须先经过WebSocket协议转换 //WebSocket协议数据，必须经过HTTP协议处理，但最终会encode编码成一个HTTP协议的响应数据 //然后服务端将HTTP响应数据序列化的字节数组，传输给浏览器 //浏览器拿到字节数组后进行反序列化，拿到一个HTTP协议响应数据，提取出内容再按照WebSocket协议来处理 //最终把普通的数据给WebSocket代码 .addLast("netty-web-socket-server-handler", new NettyWebSocketServerHandler()); }(8)WebSocket网页客户端代码开发与实现
<html lang="en"> <head> <meta http-equiv="content-type" content="text/html; charset=utf-8" /> <title>websocket网页</title> </head> <body onload="connectServer();"> <script type="text/javascript"> var websocket; function connectServer() { if ("WebSocket" in window) { console.log("Your browser supports websocket!"); websocket = new WebSocket("ws://localhost:8998/websocket"); websocket.onopen = function() { console.log("Send request to netty server."); websocket.send("I am websocket client."); } websocket.onmessage = function(ev) { var response = ev.data; console.log("Receive response from netty server: " + response); } } } </script> </body> </html>启动Netty Server，然后打开该HTML即可看到console的输出。
WebSocket和Netty Server配合起来开发说明：
如果Server端要主动推送一些通知(push)给网页端正在浏览网页的用户，如推送用户可能感兴趣的商品、关注的新闻。那么在用户进入网页后可以询问用户，是否愿意收到服务端发送的xx提示和通知。如果用户愿意，那么网页里的WebSocket完全可以跟NettyServer构建一个长连接。这样NettyServer在必要时，就可以反向推送通知(push)给用户，浏览器里的网页可能会弹出一个push通知用户xx讯息。
3.Netty实现的消息推送系统
(1)基于WebSocket的消息推送系统说明
(2)消息推送系统的PushServer
(3)消息推送系统的连接管理封装
(4)消息推送系统的ping-pong探测
(5)消息推送系统的全连接推送
(6)消息推送系统的HTTP响应和握手
(7)消息推送系统的运营客户端
(8)运营客户端连接PushServer
(9)运营客户端的Handler处理器
(10)运营客户端发送推送消息
(11)浏览器客户端接收推送消息
(1)基于WebSocket的消息推送系统说明
首先需要一个运营系统能够基于NettyClient和PushServer建立WebSocket长连接，然后浏览器客户端也要和PushServer建立好WebSocket长连接，接着运营系统会让NettyClient发送Push推送消息给PushServer，最后PushServer再把推送消息发送给浏览器客户端。
首先启动PushServer，然后打开多个网页客户端查看console，接着启动运营客系统在控制台输入消息，这样就可以完成一个完整的消息推送的交互了。
(2)消息推送系统的PushServer
public class NettyPushServer { private static final Logger logger = LogManager.getLogger(NettyPushServer.class); private static final int DEFAULT_PORT = 8998; private int port; public NettyPushServer(int port) { this.port = port; } public void start() throws Exception { logger.info("Netty Push Server is starting."); EventLoopGroup bossEventLoopGroup = new NioEventLoopGroup(); EventLoopGroup workerEventLoopGroup = new NioEventLoopGroup(); try { ServerBootstrap serverBootstrap = new ServerBootstrap(); serverBootstrap.group(bossEventLoopGroup, workerEventLoopGroup) .channel(NioServerSocketChannel.class) .childHandler(new ChannelInitializer<SocketChannel>() { protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline() .addLast("logging", new LoggingHandler("DEBUG")) .addLast("http-codec", new HttpServerCodec()) .addLast("aggregator", new HttpObjectAggregator(65536)) .addLast("http-chunked", new ChunkedWriteHandler()) .addLast("netty-push-server-handler", new NettyPushServerHandler()); } }); ChannelFuture channelFuture = serverBootstrap.bind(port).sync(); logger.info("Netty Push Server is started, listened[" + port + "]."); channelFuture.channel().closeFuture().sync(); } finally { bossEventLoopGroup.shutdownGracefully(); workerEventLoopGroup.shutdownGracefully(); } } public static void main(String[] args) throws Exception { NettyPushServer nettyHttpServer = new NettyPushServer(DEFAULT_PORT); nettyHttpServer.start(); } } public class NettyPushServerHandler extends SimpleChannelInboundHandler<Object> { private static final Logger logger = LogManager.getLogger(NettyPushServerHandler.class); public void channelActive(ChannelHandlerContext ctx) throws Exception { logger.info("Client Connection Established: " + ctx.channel()); } public void channelInactive(ChannelHandlerContext ctx) throws Exception { logger.info("Client Disconnected: " + ctx.channel()); } protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof FullHttpRequest) { handleHttpRequest(ctx, (FullHttpRequest) msg); } else if(msg instanceof WebSocketFrame) { handleWebSocketFrame(ctx, (WebSocketFrame) msg); } } public void channelReadComplete(ChannelHandlerContext ctx) throws Exception { ctx.flush(); } private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame webSocketFrame) { } private void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest request) { } }(3)消息推送系统的连接管理封装
//用来管理连接 public class ChannelManager { private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); private static ConcurrentHashMap<String, ChannelId> channelIds = new ConcurrentHashMap<String, ChannelId>(); public static void add(Channel channel) { channelGroup.add(channel); channelIds.put(channel.id().asShortText(), channel.id()); } public static void remove(Channel channel) { channelGroup.remove(channel); channelIds.remove(channel.id().asShortText()); } public static Channel get(String id) { return channelGroup.find(channelIds.get(id)); } public static void pushToAllChannels(TextWebSocketFrame webSocketFrame) { channelGroup.writeAndFlush(webSocketFrame); } } public class NettyPushServerHandler extends SimpleChannelInboundHandler<Object> { ... @Override public void channelActive(ChannelHandlerContext ctx) throws Exception { logger.info("Client Connection Established: " + ctx.channel()); ChannelManager.add(ctx.channel()); } @Override public void channelInactive(ChannelHandlerContext ctx) throws Exception { logger.info("Client Disconnected: " + ctx.channel()); ChannelManager.remove(ctx.channel()); } ... }(4)消息推送系统的ping-pong探测
public class NettyPushServerHandler extends SimpleChannelInboundHandler<Object> { ... private WebSocketServerHandshaker webSocketServerHandshaker; ... protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof FullHttpRequest) { handleHttpRequest(ctx, (FullHttpRequest) msg); } else if(msg instanceof WebSocketFrame) { handleWebSocketFrame(ctx, (WebSocketFrame) msg); } } private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame webSocketFrame) { //WebSocket网页客户端发送的是ping消息，它会不停的ping服务端，看看长连接是否存活和有效 if (webSocketFrame instanceof PingWebSocketFrame) { logger.info("Receive ping frame from client: " + ctx.channel()); WebSocketFrame pongWebSocketFrame = new PongWebSocketFrame(webSocketFrame.content().retain()); ctx.channel().write(pongWebSocketFrame); return; } //WebSocket网页客户端发送一个请求过来，请求关闭这个WebSocket连接 if (webSocketFrame instanceof CloseWebSocketFrame) { logger.info("Receive close WebSocket request from client: " + ctx.channel()); webSocketServerHandshaker.close(ctx.channel(), ((CloseWebSocketFrame) webSocketFrame).retain()); return; } ... } ... }(5)消息推送系统的全连接推送
public class NettyPushServerHandler extends SimpleChannelInboundHandler<Object> { ... protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof FullHttpRequest) { handleHttpRequest(ctx, (FullHttpRequest) msg); } else if(msg instanceof WebSocketFrame) { handleWebSocketFrame(ctx, (WebSocketFrame) msg); } } private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame webSocketFrame) { //WebSocket网页客户端发送的是ping消息，它会不停的ping服务端，看看长连接是否存活和有效 if (webSocketFrame instanceof PingWebSocketFrame) { logger.info("Receive ping frame from client: " + ctx.channel()); WebSocketFrame pongWebSocketFrame = new PongWebSocketFrame(webSocketFrame.content().retain()); ctx.channel().write(pongWebSocketFrame); return; } //WebSocket网页客户端发送一个请求过来，请求关闭这个WebSocket连接 if (webSocketFrame instanceof CloseWebSocketFrame) { logger.info("Receive close WebSocket request from client: " + ctx.channel()); webSocketServerHandshaker.close(ctx.channel(), ((CloseWebSocketFrame) webSocketFrame).retain()); return; } //WebSocket网页客户端发送请求，但它不是text文本请求 if (!(webSocketFrame instanceof TextWebSocketFrame)) { logger.error("Netty Push Server only support text frame, does not support other type frame."); String errorMsg = String.format("%s type frame is not supported.", webSocketFrame.getClass().getName()); throw new UnsupportedOperationException(errorMsg); } //WebSocket网页客户端发送一个文本请求过来，是TextFrame类型的 String request = ((TextWebSocketFrame)webSocketFrame).text(); logger.info("Receive text frame[" + request + "] from client: " + ctx.channel()); //构建响应 TextWebSocketFrame response = new TextWebSocketFrame(request); //发送给所有连接，全连接推送 ChannelManager.pushToAllChannels(response); } ... }(6)消息推送系统的HTTP响应和握手
public class NettyPushServerHandler extends SimpleChannelInboundHandler<Object> { ... private void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest request) { if (!request.decoderResult().isSuccess() || (!"websocket".equals(request.headers().get("Upgrade")))) { DefaultFullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.BAD_REQUEST); sendHttpResponse(ctx, request, response); return; } logger.info("Receive handshake request from client: " + ctx.channel()); //握手建立 WebSocketServerHandshakerFactory factory = new WebSocketServerHandshakerFactory("ws://localhost:8998/push", null, false); webSocketServerHandshaker = factory.newHandshaker(request); if (webSocketServerHandshaker == null) { WebSocketServerHandshakerFactory.sendUnsupportedVersionResponse(ctx.channel()); } else { webSocketServerHandshaker.handshake(ctx.channel(), request); logger.info("Netty push server handshake with client: " + ctx.channel()); } } //HTTP响应 private void sendHttpResponse(ChannelHandlerContext ctx, FullHttpRequest request, DefaultFullHttpResponse response) { if (response.status().code() != RESPONSE_CODE_OK) { ByteBuf byteBuf = Unpooled.copiedBuffer(response.status().toString(), CharsetUtil.UTF_8); response.content().writeBytes(byteBuf); logger.info("Http Response is not ok: " + byteBuf.toString(CharsetUtil.UTF_8)); byteBuf.release(); } ChannelFuture channelFuture = ctx.channel().writeAndFlush(response); if (response.status().code() != RESPONSE_CODE_OK) { channelFuture.addListener(ChannelFutureListener.CLOSE); } } ... }(7)消息推送系统的运营客户端
public class OperationNettyClient { private static final Logger logger = LogManager.getLogger(OperationNettyClient.class); private static final String WEB_SOCKET_SCHEME = "ws"; private static final String WSS_SCHEME = "wss"; private static final String LOCAL_HOST = "127.0.0.1"; private static final String PUSH_SERVER_URI = System.getProperty("url", "ws://127.0.0.1:8998/push"); private static URI uri; private static String scheme; private static String host; private static int port; private static SslContext sslContext; private EventLoopGroup eventLoopGroup; public void start() throws Exception { //... } public static void main(String[] args) throws Exception { uri = new URI(PUSH_SERVER_URI); scheme = getScheme(uri); host = getHost(uri); port = getPort(uri, scheme); checkScheme(scheme); initSslContext(scheme); } private static String getScheme(URI pushServerUri) { return pushServerUri.getScheme() == null ? WEB_SOCKET_SCHEME : pushServerUri.getScheme(); } private static String getHost(URI pushServerUri) { return pushServerUri.getHost() == null ? LOCAL_HOST : pushServerUri.getHost(); } private static int getPort(URI pushServerUri, String scheme) { int port; if (pushServerUri.getPort() == -1) { if (WEB_SOCKET_SCHEME.equals(scheme)) { port = 80; } else if(WSS_SCHEME.equals(scheme)) { port = 443; } else { port = -1; } } else { port = pushServerUri.getPort(); } return port; } //检查scheme是否是ws或wss private static void checkScheme(String scheme) { if (!WEB_SOCKET_SCHEME.equals(scheme) && !WSS_SCHEME.equals(scheme)) { logger.error("Only Support ws or wss scheme."); throw new RuntimeException("Only Support ws or wss scheme."); } } //如果WebSocket使用了SSL，也就是wss，那么初始化对应的sslContext private static void initSslContext(String scheme) throws Exception { boolean enableSSL = WSS_SCHEME.equals(scheme); if (enableSSL) { sslContext = SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build(); } else { sslContext = null; } } }(8)运营客户端连接PushServer
public class OperationNettyClient { private static final Logger logger = LogManager.getLogger(OperationNettyClient.class); private static final String WEB_SOCKET_SCHEME = "ws"; private static final String WSS_SCHEME = "wss"; private static final String LOCAL_HOST = "127.0.0.1"; private static final String PUSH_SERVER_URI = System.getProperty("url", "ws://127.0.0.1:8998/push"); private static final String INPUT_MESSAGE_QUIT = "quit"; private static final String INPUT_MESSAGE_CLOSE = "close"; private static final String INPUT_MESSAGE_PING = "ping"; private static URI uri; private static String scheme; private static String host; private static int port; private static SslContext sslContext; private EventLoopGroup eventLoopGroup; public Channel start() throws Exception { logger.info("Operation Netty Client is connecting."); eventLoopGroup = new NioEventLoopGroup(); WebSocketClientHandshaker webSocketClientHandshaker = WebSocketClientHandshakerFactory.newHandshaker(uri, WebSocketVersion.V13, null, true, new DefaultHttpHeaders()); final OperationNettyClientHandler operationNettyClientHandler = new OperationNettyClientHandler(webSocketClientHandshaker); Bootstrap bootstrap = new Bootstrap(); bootstrap.group(eventLoopGroup) .channel(NioSocketChannel.class) .handler(new ChannelInitializer<SocketChannel>() { protected void initChannel(SocketChannel ch) throws Exception { ChannelPipeline channelPipeline = ch.pipeline(); if (sslContext != null) { channelPipeline.addLast(sslContext.newHandler(ch.alloc(), host, port)); } channelPipeline.addLast(new HttpClientCodec()) .addLast(new HttpObjectAggregator(65536)) .addLast(WebSocketClientCompressionHandler.INSTANCE) .addLast(operationNettyClientHandler); } }); Channel channel = bootstrap.connect(uri.getHost(), port).sync().channel(); logger.info("Operation Netty Client connected to push server."); operationNettyClientHandler.channelFuture().sync(); return channel; } public void shutdownGracefully() { eventLoopGroup.shutdownGracefully(); } public static void main(String[] args) throws Exception { uri = new URI(PUSH_SERVER_URI); scheme = getScheme(uri); host = getHost(uri); port = getPort(uri, scheme); checkScheme(scheme); initSslContext(scheme); OperationNettyClient operationNettyClient = new OperationNettyClient(); try { Channel channel = operationNettyClient.start(); } finally { operationNettyClient.shutdownGracefully(); } } ... }(9)运营客户端的Handler处理器
public class OperationNettyClientHandler extends SimpleChannelInboundHandler<Object> { private static final Logger logger = LogManager.getLogger(OperationNettyClientHandler.class); private WebSocketClientHandshaker webSocketClientHandshaker; private ChannelFuture channelFuture; public OperationNettyClientHandler(WebSocketClientHandshaker webSocketClientHandshaker) { this.webSocketClientHandshaker = webSocketClientHandshaker; } public ChannelFuture channelFuture() { return channelFuture; } public void handlerAdded(ChannelHandlerContext ctx) throws Exception { channelFuture = ctx.newPromise(); } public void channelActive(ChannelHandlerContext ctx) throws Exception { webSocketClientHandshaker.handshake(ctx.channel()); logger.info("Operation Netty Client send WebSocket handshake request."); } public void channelInactive(ChannelHandlerContext ctx) throws Exception { logger.info("netty client disconnected."); } protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception { Channel channel = ctx.channel(); if (!webSocketClientHandshaker.isHandshakeComplete()) { try { webSocketClientHandshaker.finishHandshake(channel, (FullHttpResponse) msg); logger.info("Netty Client connected."); ((ChannelPromise)channelFuture).setSuccess(); } catch(WebSocketHandshakeException e) { logger.error("WebSocket handshake failed.", e); ((ChannelPromise)channelFuture).setFailure(e); } return; } if (msg instanceof FullHttpResponse) { FullHttpResponse response = (FullHttpResponse) msg; throw new IllegalStateException("Not Supported HTTP Response."); } WebSocketFrame webSocketFrame = (WebSocketFrame) msg; if (webSocketFrame instanceof TextWebSocketFrame) { TextWebSocketFrame textWebSocketFrame = (TextWebSocketFrame) webSocketFrame; logger.info("Receives text frame: " + textWebSocketFrame.text()); } else if(webSocketFrame instanceof PongWebSocketFrame) { logger.info("Receives pong frame: " + webSocketFrame); } else if(webSocketFrame instanceof CloseWebSocketFrame) { logger.info("Receives close WebSocket frame, Netty Client is closing."); channel.close(); } } public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception { logger.error("Operation Netty client handler exception caught.", cause); if (!channelFuture.isDone()) { ((ChannelPromise)channelFuture).setFailure(cause); } ctx.close(); } }(10)运营客户端发送推送消息
public class OperationNettyClient { ... public void waitInputMessage(Channel channel) throws Exception { BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in)); while(true) { logger.info("Wait for input message."); String message = bufferedReader.readLine(); if (INPUT_MESSAGE_QUIT.equals(message)) { break; } else if(INPUT_MESSAGE_CLOSE.equals(message)) { channel.writeAndFlush(new CloseWebSocketFrame()); channel.closeFuture().sync(); break; } else if(INPUT_MESSAGE_PING.equals(message)) { WebSocketFrame webSocketFrame = new PingWebSocketFrame(Unpooled.wrappedBuffer(new byte[] {8, 1, 8, 1})); channel.writeAndFlush(webSocketFrame); } else { WebSocketFrame webSocketFrame = new TextWebSocketFrame(message); channel.writeAndFlush(webSocketFrame); } } } public static void main(String[] args) throws Exception { uri = new URI(PUSH_SERVER_URI); scheme = getScheme(uri); host = getHost(uri); port = getPort(uri, scheme); checkScheme(scheme); initSslContext(scheme); OperationNettyClient operationNettyClient = new OperationNettyClient(); try { Channel channel = operationNettyClient.start(); //运营客户端发送消息入口 operationNettyClient.waitInputMessage(channel); } finally { operationNettyClient.shutdownGracefully(); } } ... }(11)浏览器客户端接收推送消息
<html lang="en"> <head> <meta http-equiv="content-type" content="text/html; charset=utf-8" /> <title>websocket网页</title> </head> <body onload="connectServer();"> <script type="text/javascript"> var websocket; function connectServer() { if ("WebSocket" in window) { console.log("your browser supports websocket!"); websocket = new WebSocket("ws://localhost:8998/push"); websocket.onopen = function() { console.log("established connection with push server."); } websocket.onmessage = function(ev) { var response = ev.data; console.log("receives push message from netty server: " + response); } } } </script> </body> </html>
