Adding a benchmark for websockets

Motivation:

It is often helpful to measure the performance of connections, e.g. the
latency and the throughput. This can be performed through benchmarks.

Modification:

This adds a simple but configurable benchmark for websockets into the
example directory. The Netty WebSocket server will echo all received
websocket frames and will provide an HTML/JS page which serves as the
client for the benchmark.
The benchmark also provides a verification mode that verifies the sent
against the received data. This can be used for the verification ob
websocket frame encoding and decoding funtionality.

Result:

A benchmark is added in form a further Netty websocket example.
With this benchmark it is easily possible to measure the performance between Netty and a browser
This commit is contained in:
Matthias Einwag 2014-10-01 22:02:22 +02:00 committed by Norman Maurer
parent 9f077b41b0
commit feda7afc0b
6 changed files with 495 additions and 1 deletions

View File

@ -0,0 +1,70 @@
/*
* Copyright 2014 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License,
* version 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package io.netty.example.http.websocketx.benchmarkserver;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.Channel;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.util.SelfSignedCertificate;
/**
* A Benchmark application for websocket which is served at:
*
* http://localhost:8080/websocket
*
* Open your browser at http://localhost:8080/, then the benchmark page will be loaded and a Web Socket connection will
* be made automatically.
*/
public final class WebSocketServer {
static final boolean SSL = System.getProperty("ssl") != null;
static final int PORT = Integer.parseInt(System.getProperty("port", SSL? "8443" : "8080"));
public static void main(String[] args) throws Exception {
// Configure SSL.
final SslContext sslCtx;
if (SSL) {
SelfSignedCertificate ssc = new SelfSignedCertificate();
sslCtx = SslContext.newServerContext(ssc.certificate(), ssc.privateKey());
} else {
sslCtx = null;
}
EventLoopGroup bossGroup = new NioEventLoopGroup(1);
EventLoopGroup workerGroup = new NioEventLoopGroup();
try {
ServerBootstrap b = new ServerBootstrap();
b.group(bossGroup, workerGroup)
.channel(NioServerSocketChannel.class)
.childHandler(new WebSocketServerInitializer(sslCtx));
Channel ch = b.bind(PORT).sync().channel();
System.out.println("Open your web browser and navigate to " +
(SSL? "https" : "http") + "://127.0.0.1:" + PORT + '/');
ch.closeFuture().sync();
} finally {
bossGroup.shutdownGracefully();
workerGroup.shutdownGracefully();
}
}
}

View File

@ -0,0 +1,191 @@
/*
* Copyright 2014 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License,
* version 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package io.netty.example.http.websocketx.benchmarkserver;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.util.CharsetUtil;
/**
* Generates the benchmark HTML page which is served at http://localhost:8080/
*/
public final class WebSocketServerBenchmarkPage {
private static final String NEWLINE = "\r\n";
public static ByteBuf getContent(String webSocketLocation) {
return Unpooled.copiedBuffer(
"<html><head><title>Web Socket Performance Test</title></head>" + NEWLINE +
"<body>" + NEWLINE +
"<h2>WebSocket Performance Test</h2>" + NEWLINE +
"<label>Connection Status:</label>" + NEWLINE +
"<label id=\"connectionLabel\"></label><br />" + NEWLINE +
"<form onsubmit=\"return false;\">" + NEWLINE +
"Message size:" +
"<input type=\"text\" id=\"messageSize\" value=\"1024\"/><br>" + NEWLINE +
"Number of messages:" +
"<input type=\"text\" id=\"nrMessages\" value=\"100000\"/><br>" + NEWLINE +
"Data Type:" +
"<input type=\"radio\" name=\"type\" id=\"typeText\" value=\"text\" checked>text" +
"<input type=\"radio\" name=\"type\" id=\"typeBinary\" value=\"binary\">binary<br>" + NEWLINE +
"Mode:<br>" + NEWLINE +
"<input type=\"radio\" name=\"mode\" id=\"modeSingle\" value=\"single\" checked>" +
"Wait for response after each messages<br>" + NEWLINE +
"<input type=\"radio\" name=\"mode\" id=\"modeAll\" value=\"all\">" +
"Send all messages and then wait for all responses<br>" + NEWLINE +
"<input type=\"checkbox\" id=\"verifiyResponses\">Verify responded messages<br>" + NEWLINE +
"<input type=\"button\" value=\"Start Benchmark\"" + NEWLINE +
" onclick=\"startBenchmark()\" />" + NEWLINE +
"<h3>Output</h3>" + NEWLINE +
"<textarea id=\"output\" style=\"width:500px;height:300px;\"></textarea>" + NEWLINE +
"<br>" + NEWLINE +
"<input type=\"button\" value=\"Clear\" onclick=\"clearText()\">" + NEWLINE +
"</form>" + NEWLINE +
"<script type=\"text/javascript\">" + NEWLINE +
"var benchRunning = false;" + NEWLINE +
"var messageSize = 0;" + NEWLINE +
"var totalMessages = 0;" + NEWLINE +
"var rcvdMessages = 0;" + NEWLINE +
"var isBinary = true;" + NEWLINE +
"var isSingle = true;" + NEWLINE +
"var verifiyResponses = false;" + NEWLINE +
"var benchData = null;" + NEWLINE +
"var startTime;" + NEWLINE +
"var endTime;" + NEWLINE +
"var socket;" + NEWLINE +
"var output = document.getElementById('output');" + NEWLINE +
"var connectionLabel = document.getElementById('connectionLabel');" + NEWLINE +
"if (!window.WebSocket) {" + NEWLINE +
" window.WebSocket = window.MozWebSocket;" + NEWLINE +
'}' + NEWLINE +
"if (window.WebSocket) {" + NEWLINE +
" socket = new WebSocket(\"" + webSocketLocation + "\");" + NEWLINE +
" socket.binaryType = 'arraybuffer';" + NEWLINE +
" socket.onmessage = function(event) {" + NEWLINE +
" if (verifiyResponses) {" + NEWLINE +
" if (isBinary) {" + NEWLINE +
" if (!(event.data instanceof ArrayBuffer) || " + NEWLINE +
" event.data.byteLength != benchData.byteLength) {" + NEWLINE +
" onInvalidResponse(benchData, event.data);" + NEWLINE +
" return;" + NEWLINE +
" } else {" + NEWLINE +
" var v = new Uint8Array(event.data);" + NEWLINE +
" for (var j = 0; j < benchData.byteLength; j++) {" + NEWLINE +
" if (v[j] != benchData[j]) {" + NEWLINE +
" onInvalidResponse(benchData, event.data);" + NEWLINE +
" return;" + NEWLINE +
" }" + NEWLINE +
" }" + NEWLINE +
" }" + NEWLINE +
" } else {" + NEWLINE +
" if (event.data != benchData) {" + NEWLINE +
" onInvalidResponse(benchData, event.data);" + NEWLINE +
" return;" + NEWLINE +
" }" + NEWLINE +
" }" + NEWLINE +
" }" + NEWLINE +
" rcvdMessages++;" + NEWLINE +
" if (rcvdMessages == totalMessages) {" + NEWLINE +
" onFinished();" + NEWLINE +
" } else if (isSingle) {" + NEWLINE +
" socket.send(benchData);" + NEWLINE +
" }" + NEWLINE +
" };" + NEWLINE +
" socket.onopen = function(event) {" + NEWLINE +
" connectionLabel.innerHTML = \"Connected\";" + NEWLINE +
" };" + NEWLINE +
" socket.onclose = function(event) {" + NEWLINE +
" benchRunning = false;" + NEWLINE +
" connectionLabel.innerHTML = \"Disconnected\";" + NEWLINE +
" };" + NEWLINE +
"} else {" + NEWLINE +
" alert(\"Your browser does not support Web Socket.\");" + NEWLINE +
'}' + NEWLINE +
NEWLINE +
"function onInvalidResponse(sent,recvd) {" + NEWLINE +
" socket.close();" + NEWLINE +
" alert(\"Error: Sent data did not match the received data!\");" + NEWLINE +
"}" + NEWLINE +
NEWLINE +
"function clearText() {" + NEWLINE +
" output.value=\"\";" + NEWLINE +
"}" + NEWLINE +
NEWLINE +
"function createBenchData() {" + NEWLINE +
" if (isBinary) {" + NEWLINE +
" benchData = new Uint8Array(messageSize);" + NEWLINE +
" for (var i=0; i < messageSize; i++) {" + NEWLINE +
" benchData[i] += Math.floor(Math.random() * 255);" + NEWLINE +
" }" + NEWLINE +
" } else { " + NEWLINE +
" benchData = \"\";" + NEWLINE +
" for (var i=0; i < messageSize; i++) {" + NEWLINE +
" benchData += String.fromCharCode(Math.floor(Math.random() * (123 - 65) + 65));" + NEWLINE +
" }" + NEWLINE +
" }" + NEWLINE +
"}" + NEWLINE +
NEWLINE +
"function startBenchmark(message) {" + NEWLINE +
" if (!window.WebSocket || benchRunning) { return; }" + NEWLINE +
" if (socket.readyState == WebSocket.OPEN) {" + NEWLINE +
" isBinary = document.getElementById('typeBinary').checked;" + NEWLINE +
" isSingle = document.getElementById('modeSingle').checked;" + NEWLINE +
" verifiyResponses = document.getElementById('verifiyResponses').checked;" + NEWLINE +
" messageSize = parseInt(document.getElementById('messageSize').value);" + NEWLINE +
" totalMessages = parseInt(document.getElementById('nrMessages').value);" + NEWLINE +
" if (isNaN(messageSize) || isNaN(totalMessages)) return;" + NEWLINE +
" createBenchData();" + NEWLINE +
" output.value = output.value + '\\nStarting Benchmark';" + NEWLINE +
" rcvdMessages = 0;" + NEWLINE +
" benchRunning = true;" + NEWLINE +
" startTime = new Date();" + NEWLINE +
" if (isSingle) {" + NEWLINE +
" socket.send(benchData);" + NEWLINE +
" } else {" + NEWLINE +
" for (var i = 0; i < totalMessages; i++) socket.send(benchData);" + NEWLINE +
" }" + NEWLINE +
" } else {" + NEWLINE +
" alert(\"The socket is not open.\");" + NEWLINE +
" }" + NEWLINE +
'}' + NEWLINE +
NEWLINE +
"function onFinished() {" + NEWLINE +
" endTime = new Date();" + NEWLINE +
" var duration = (endTime - startTime) / 1000.0;" + NEWLINE +
" output.value = output.value + '\\nTest took: ' + duration + 's';" + NEWLINE +
" var messagesPerS = totalMessages / duration;" + NEWLINE +
" output.value = output.value + '\\nPerformance: ' + messagesPerS + ' Messages/s';" + NEWLINE +
" output.value = output.value + ' in each direction';" + NEWLINE +
" output.value = output.value + '\\nRound trip: ' + 1000.0/messagesPerS + 'ms';" + NEWLINE +
" var throughput = messageSize * totalMessages / duration;" + NEWLINE +
" var throughputText;" + NEWLINE +
" if (isBinary) throughputText = throughput / (1024*1024) + ' MB/s';" + NEWLINE +
" else throughputText = throughput / (1000*1000) + ' MChars/s';" + NEWLINE +
" output.value = output.value + '\\nThroughput: ' + throughputText;" + NEWLINE +
" output.value = output.value + ' in each direction';" + NEWLINE +
" benchRunning = false;" + NEWLINE +
"}" + NEWLINE +
"</script>" + NEWLINE +
"</body>" + NEWLINE +
"</html>" + NEWLINE, CharsetUtil.US_ASCII);
}
private WebSocketServerBenchmarkPage() {
// Unused
}
}

View File

@ -0,0 +1,161 @@
/*
* Copyright 2014 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License,
* version 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package io.netty.example.http.websocketx.benchmarkserver;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.DefaultFullHttpResponse;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.FullHttpResponse;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.websocketx.BinaryWebSocketFrame;
import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame;
import io.netty.handler.codec.http.websocketx.PingWebSocketFrame;
import io.netty.handler.codec.http.websocketx.PongWebSocketFrame;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketServerHandshaker;
import io.netty.handler.codec.http.websocketx.WebSocketServerHandshakerFactory;
import io.netty.util.CharsetUtil;
import static io.netty.handler.codec.http.HttpHeaders.Names.*;
import static io.netty.handler.codec.http.HttpMethod.*;
import static io.netty.handler.codec.http.HttpResponseStatus.*;
import static io.netty.handler.codec.http.HttpVersion.*;
/**
* Handles handshakes and messages
*/
public class WebSocketServerHandler extends SimpleChannelInboundHandler<Object> {
private static final String WEBSOCKET_PATH = "/websocket";
private WebSocketServerHandshaker handshaker;
@Override
public void channelRead0(ChannelHandlerContext ctx, Object msg) {
if (msg instanceof FullHttpRequest) {
handleHttpRequest(ctx, (FullHttpRequest) msg);
} else if (msg instanceof WebSocketFrame) {
handleWebSocketFrame(ctx, (WebSocketFrame) msg);
}
}
@Override
public void channelReadComplete(ChannelHandlerContext ctx) {
ctx.flush();
}
private void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest req) {
// Handle a bad request.
if (!req.getDecoderResult().isSuccess()) {
sendHttpResponse(ctx, req, new DefaultFullHttpResponse(HTTP_1_1, BAD_REQUEST));
return;
}
// Allow only GET methods.
if (req.getMethod() != GET) {
sendHttpResponse(ctx, req, new DefaultFullHttpResponse(HTTP_1_1, FORBIDDEN));
return;
}
// Send the demo page and favicon.ico
if ("/".equals(req.getUri())) {
ByteBuf content = WebSocketServerBenchmarkPage.getContent(getWebSocketLocation(req));
FullHttpResponse res = new DefaultFullHttpResponse(HTTP_1_1, OK, content);
res.headers().set(CONTENT_TYPE, "text/html; charset=UTF-8");
HttpHeaders.setContentLength(res, content.readableBytes());
sendHttpResponse(ctx, req, res);
return;
}
if ("/favicon.ico".equals(req.getUri())) {
FullHttpResponse res = new DefaultFullHttpResponse(HTTP_1_1, NOT_FOUND);
sendHttpResponse(ctx, req, res);
return;
}
// Handshake
WebSocketServerHandshakerFactory wsFactory = new WebSocketServerHandshakerFactory(
getWebSocketLocation(req), null, true, 5 * 1024 * 1024);
handshaker = wsFactory.newHandshaker(req);
if (handshaker == null) {
WebSocketServerHandshakerFactory.sendUnsupportedVersionResponse(ctx.channel());
} else {
handshaker.handshake(ctx.channel(), req);
}
}
private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame) {
// Check for closing frame
if (frame instanceof CloseWebSocketFrame) {
handshaker.close(ctx.channel(), (CloseWebSocketFrame) frame.retain());
return;
}
if (frame instanceof PingWebSocketFrame) {
ctx.write(new PongWebSocketFrame(frame.content().retain()));
return;
}
if (frame instanceof TextWebSocketFrame) {
// Echo the frame
ctx.write(frame.retain());
return;
}
if (frame instanceof BinaryWebSocketFrame) {
// Echo the frame
ctx.write(frame.retain());
return;
}
}
private static void sendHttpResponse(
ChannelHandlerContext ctx, FullHttpRequest req, FullHttpResponse res) {
// Generate an error page if response getStatus code is not OK (200).
if (res.getStatus().code() != 200) {
ByteBuf buf = Unpooled.copiedBuffer(res.getStatus().toString(), CharsetUtil.UTF_8);
res.content().writeBytes(buf);
buf.release();
HttpHeaders.setContentLength(res, res.content().readableBytes());
}
// Send the response and close the connection if necessary.
ChannelFuture f = ctx.channel().writeAndFlush(res);
if (!HttpHeaders.isKeepAlive(req) || res.getStatus().code() != 200) {
f.addListener(ChannelFutureListener.CLOSE);
}
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
cause.printStackTrace();
ctx.close();
}
private static String getWebSocketLocation(FullHttpRequest req) {
String location = req.headers().get(HOST) + WEBSOCKET_PATH;
if (WebSocketServer.SSL) {
return "wss://" + location;
} else {
return "ws://" + location;
}
}
}

View File

@ -0,0 +1,45 @@
/*
* Copyright 2014 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License,
* version 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package io.netty.example.http.websocketx.benchmarkserver;
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.ssl.SslContext;
/**
*/
public class WebSocketServerInitializer extends ChannelInitializer<SocketChannel> {
private final SslContext sslCtx;
public WebSocketServerInitializer(SslContext sslCtx) {
this.sslCtx = sslCtx;
}
@Override
public void initChannel(SocketChannel ch) throws Exception {
ChannelPipeline pipeline = ch.pipeline();
if (sslCtx != null) {
pipeline.addLast(sslCtx.newHandler(ch.alloc()));
}
pipeline.addLast(new HttpServerCodec());
pipeline.addLast(new HttpObjectAggregator(65536));
pipeline.addLast(new WebSocketServerHandler());
}
}

View File

@ -0,0 +1,27 @@
/*
* Copyright 2014 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License,
* version 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
/**
* <p>This package contains a benchmark application for websockets.
* <p>The websocket server will accept websocket upgrade requests and simply echo
* all incoming frames.
* <p>The supplied index page contains a benchmark client that runs in a browser.
* <p>Once started, you can start benchmarking by by navigating
* to http://localhost:8080/ with your browser.
* <p>You can also test it with a web socket client. Send web socket traffic to
* ws://localhost:8080/websocket.
*/
package io.netty.example.http.websocketx.benchmarkserver;

View File

@ -70,7 +70,7 @@ public final class WebSocketServer {
Channel ch = b.bind(PORT).sync().channel();
System.err.println("Open your web browser and navigate to " +
System.out.println("Open your web browser and navigate to " +
(SSL? "https" : "http") + "://127.0.0.1:" + PORT + '/');
ch.closeFuture().sync();