netty处理客户端传过来的get、post、websocket数据例子
用neety作为http服务器，在服务器端读取客户端传过来的get和post参数和websocket数据例子，netty版本3.6.6
package com.penngo.http; import java.net.InetSocketAddress; import java.util.concurrent.Executors; import org.jboss.netty.bootstrap.ServerBootstrap; import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory; public class HttpServer { private final int port; public HttpServer(int port) { this.port = port; } public void run() { ServerBootstrap bootstrap = new ServerBootstrap(new NioServerSocketChannelFactory( Executors.newCachedThreadPool(), Executors.newCachedThreadPool())); bootstrap.setPipelineFactory(new HttpServerPipelineFactory()); bootstrap.bind(new InetSocketAddress(port)); System.out.println("Web socket server started at port " + port + '.'); System.out.println("Open your browser and navigate to http://localhost:" + port + '/'); } public static void main(String[] args) { int port; if (args.length > 0) { port = Integer.parseInt(args[0]); } else { port = 8080; } new HttpServer(port).run(); } }
package com.penngo.http; import java.util.List; import java.util.Map; import org.jboss.netty.buffer.ChannelBuffer; import org.jboss.netty.buffer.ChannelBuffers; import org.jboss.netty.channel.ChannelFuture; import org.jboss.netty.channel.ChannelFutureListener; import org.jboss.netty.channel.ChannelHandlerContext; import org.jboss.netty.channel.ExceptionEvent; import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.SimpleChannelUpstreamHandler; import org.jboss.netty.handler.codec.http.DefaultHttpResponse; import org.jboss.netty.handler.codec.http.HttpRequest; import org.jboss.netty.handler.codec.http.HttpResponse; import org.jboss.netty.handler.codec.http.QueryStringDecoder; import org.jboss.netty.handler.codec.http.multipart.Attribute; import org.jboss.netty.handler.codec.http.multipart.DefaultHttpDataFactory; import org.jboss.netty.handler.codec.http.multipart.HttpPostRequestDecoder; import org.jboss.netty.handler.codec.http.multipart.InterfaceHttpData; import org.jboss.netty.handler.codec.http.multipart.InterfaceHttpData.HttpDataType; import org.jboss.netty.handler.codec.http.websocketx.CloseWebSocketFrame; import org.jboss.netty.handler.codec.http.websocketx.PingWebSocketFrame; import org.jboss.netty.handler.codec.http.websocketx.PongWebSocketFrame; import org.jboss.netty.handler.codec.http.websocketx.TextWebSocketFrame; import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; import org.jboss.netty.handler.codec.http.websocketx.WebSocketServerHandshaker; import org.jboss.netty.handler.codec.http.websocketx.WebSocketServerHandshakerFactory; import org.jboss.netty.util.CharsetUtil; import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.*; import static org.jboss.netty.handler.codec.http.HttpHeaders.*; import static org.jboss.netty.handler.codec.http.HttpMethod.*; import static org.jboss.netty.handler.codec.http.HttpResponseStatus.*; import static org.jboss.netty.handler.codec.http.HttpVersion.*; public class HttpServerHandler extends SimpleChannelUpstreamHandler { private WebSocketServerHandshaker handshaker; private static final String WEBSOCKET_PATH = "/websocket"; @Override public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception { Object msg = e.getMessage(); if (msg instanceof HttpRequest) { handleHttpRequest(ctx, (HttpRequest) msg); } else if (msg instanceof WebSocketFrame) { handleWebSocketFrame(ctx, (WebSocketFrame) msg); } } private void handleHttpRequest(ChannelHandlerContext ctx, HttpRequest req) throws Exception { System.out.println("handleHttpRequest method==========" + req.getMethod()); System.out.println("handleHttpRequest uri==========" + req.getUri()); if (req.getMethod() == GET) { // 处理get请求 if(req.getUri().equals("/websocket")){ // Handshake WebSocketServerHandshakerFactory wsFactory = new WebSocketServerHandshakerFactory( getWebSocketLocation(req), null, false); handshaker = wsFactory.newHandshaker(req); if (handshaker == null) { wsFactory.sendUnsupportedWebSocketVersionResponse(ctx.getChannel()); } else { handshaker.handshake(ctx.getChannel(), req).addListener(WebSocketServerHandshaker.HANDSHAKE_LISTENER); } } else{ QueryStringDecoder decoder = new QueryStringDecoder(req.getUri()); Map<String, List<String>> parame = decoder.getParameters(); List<String> q = parame.get("q"); // 读取从客户端传过来的参数 String question = q.get(0); if (question != null && !question.equals("")) { System.out.println("r :" + question); HttpResponse res = new DefaultHttpResponse(HTTP_1_1, OK); String data = "<html><body>你好，GET</body><html>"; ChannelBuffer content = ChannelBuffers.copiedBuffer(data, CharsetUtil.UTF_8); res.setHeader(CONTENT_TYPE, "text/html; charset=UTF-8"); res.setHeader(ACCESS_CONTROL_ALLOW_ORIGIN, "*"); setContentLength(res, content.readableBytes()); res.setContent(content); sendHttpResponse(ctx, req, res); } } } if (req.getMethod() == POST) { // 处理POST请求 HttpPostRequestDecoder decoder = new HttpPostRequestDecoder( new DefaultHttpDataFactory(false), req); InterfaceHttpData postData = decoder.getBodyHttpData("q"); // // // 读取从客户端传过来的参数 String question = ""; if (postData.getHttpDataType() == HttpDataType.Attribute) { Attribute attribute = (Attribute) postData; question = attribute.getValue(); System.out.println("q:" + question); } if (question != null && !question.equals("")) { HttpResponse res = new DefaultHttpResponse(HTTP_1_1, OK); String data = "<html><body>你好，POST</body><html>"; ChannelBuffer content = ChannelBuffers.copiedBuffer(data, CharsetUtil.UTF_8); res.setHeader(CONTENT_TYPE, "text/html; charset=UTF-8"); res.setHeader(ACCESS_CONTROL_ALLOW_ORIGIN, "*"); setContentLength(res, content.readableBytes()); res.setContent(content); sendHttpResponse(ctx, req, res); } return; } } @Override public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) throws Exception { e.getCause().printStackTrace(); e.getChannel().close(); } private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame) { if (frame instanceof CloseWebSocketFrame) { handshaker.close(ctx.getChannel(), (CloseWebSocketFrame) frame); return; } if (frame instanceof PingWebSocketFrame) { ctx.getChannel().write(new PongWebSocketFrame(frame.getBinaryData())); return; } if (!(frame instanceof TextWebSocketFrame)) { throw new UnsupportedOperationException(String.format("%s frame types not supported", frame.getClass() .getName())); } String request = ((TextWebSocketFrame) frame).getText(); System.out.println("收到socket msg=" + request); request = "这是来自服务器端的数据"; ctx.getChannel().write(new TextWebSocketFrame(request.toUpperCase())); } private static void sendHttpResponse(ChannelHandlerContext ctx, HttpRequest req, HttpResponse res) { if (res.getStatus().getCode() != 200) { res.setContent(ChannelBuffers.copiedBuffer(res.getStatus() .toString(), CharsetUtil.UTF_8)); setContentLength(res, res.getContent().readableBytes()); } ChannelFuture f = ctx.getChannel().write(res); if (!isKeepAlive(req) || res.getStatus().getCode() != 200) { f.addListener(ChannelFutureListener.CLOSE); } } private static String getWebSocketLocation(HttpRequest req) { return "ws://" + req.getHeader(HOST) + WEBSOCKET_PATH; } }
package com.penngo.http; import static org.jboss.netty.channel.Channels.*; import org.jboss.netty.channel.ChannelPipeline; import org.jboss.netty.channel.ChannelPipelineFactory; import org.jboss.netty.handler.codec.http.HttpChunkAggregator; import org.jboss.netty.handler.codec.http.HttpRequestDecoder; import org.jboss.netty.handler.codec.http.HttpResponseEncoder; public class HttpServerPipelineFactory implements ChannelPipelineFactory { public ChannelPipeline getPipeline() throws Exception { // Create a default pipeline implementation. ChannelPipeline pipeline = pipeline(); pipeline.addLast("decoder", new HttpRequestDecoder()); // pipeline.addLast("aggregator", new HttpChunkAggregator(65536)); pipeline.addLast("encoder", new HttpResponseEncoder()); pipeline.addLast("handler", new HttpServerHandler()); return pipeline; } }
测试代码
<!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>Web Socket Example</title> <script src="http://ajax.aspnetcdn.com/ajax/jQuery/jquery-1.8.3.min.js" type="text/javascript"></script> <style> textarea { width: 500px; height: 100px; } </style> <script> $(function(){ var Sock = function() { var socket; if (!window.WebSocket) { window.WebSocket = window.MozWebSocket; } if (window.WebSocket) { socket = new WebSocket("ws://localhost:8080/websocket"); socket.onopen = onopen; socket.onmessage = onmessage; socket.onclose = onclose; } else { alert("Your browser does not support Web Socket."); } function onopen(event) { getTextAreaElement().value = "Web Socket opened!"; } function onmessage(event) { appendTextArea(event.data); } function onclose(event) { appendTextArea("Web Socket closed"); } function appendTextArea(newData) { var el = getTextAreaElement(); el.value = el.value + '\n' + newData; } function getTextAreaElement() { return document.getElementById('responseText'); } function send(event) { event.preventDefault(); if (window.WebSocket) { if (socket.readyState == WebSocket.OPEN) { socket.send(event.target.message.value); } else { alert("The socket is not open."); } } } document.forms.inputform.addEventListener('submit', send, false); } window.addEventListener('load', function() { new Sock(); }, false); $("#btnGet").click( function () { $.get("http://localhost:8080/", { q: "John"}, function(data){ $("#responseTextGet").val($("#responseTextGet").val() + data) }); }); $("#btnPost").click( function () { $.post("http://localhost:8080/", { q: "John"}, function(data){ $("#responseTextPost").val($("#responseTextGet").val() + data) }); }); }); </script> </head> <body> <h3>输入内容</h3> <form name="inputform"> <input type="text" name="message" id="message" placeholder="Enter text to be sent" autofocus> <input type="submit" value="发送WebSocket数据"> </form> <h3>服务端返回</h3> <textarea id="responseText"></textarea> <h3>输入内容</h3> <form name="inputform"> <input type="text" name="message" id="messageGet" placeholder="Enter text to be sent" autofocus> <input type="button" value="Get数据" id="btnGet"> </form> <h3>服务端返回</h3> <textarea id="responseTextGet"></textarea> <h3>输入内容</h3> <form name="inputform"> <input type="text" name="message" id="messagePost" placeholder="Enter text to be sent" autofocus> <input type="button" value="Post数据" id="btnPost"> </form> <h3>服务端返回</h3> <textarea id="responseTextPost"></textarea> </body> </html>