1 /*
2 * Copyright 2014 The Netty Project
3 *
4 * The Netty Project licenses this file to you under the Apache License,
5 * version 2.0 (the "License"); you may not use this file except in compliance
6 * with the License. You may obtain a copy of the License at:
7 *
8 * https://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 * License for the specific language governing permissions and limitations
14 * under the License.
15 */
16 package io.netty.example.http.websocketx.benchmarkserver;
17
18 import io.netty.buffer.ByteBuf;
19 import io.netty.buffer.Unpooled;
20 import io.netty.util.CharsetUtil;
21
22 /**
23 * Generates the benchmark HTML page which is served at http://localhost:8080/
24 */
25 public final class WebSocketServerBenchmarkPage {
26
27 private static final String NEWLINE = "\r\n";
28
29 public static ByteBuf getContent(String webSocketLocation) {
30 return Unpooled.copiedBuffer(
31 "<html><head><title>Web Socket Performance Test</title></head>" + NEWLINE +
32 "<body>" + NEWLINE +
33 "<h2>WebSocket Performance Test</h2>" + NEWLINE +
34 "<label>Connection Status:</label>" + NEWLINE +
35 "<label id=\"connectionLabel\"></label><br />" + NEWLINE +
36
37 "<form onsubmit=\"return false;\">" + NEWLINE +
38 "Message size:" +
39 "<input type=\"text\" id=\"messageSize\" value=\"1024\"/><br>" + NEWLINE +
40 "Number of messages:" +
41 "<input type=\"text\" id=\"nrMessages\" value=\"100000\"/><br>" + NEWLINE +
42 "Data Type:" +
43 "<input type=\"radio\" name=\"type\" id=\"typeText\" value=\"text\" checked>text" +
44 "<input type=\"radio\" name=\"type\" id=\"typeBinary\" value=\"binary\">binary<br>" + NEWLINE +
45 "Mode:<br>" + NEWLINE +
46 "<input type=\"radio\" name=\"mode\" id=\"modeSingle\" value=\"single\" checked>" +
47 "Wait for response after each messages<br>" + NEWLINE +
48 "<input type=\"radio\" name=\"mode\" id=\"modeAll\" value=\"all\">" +
49 "Send all messages and then wait for all responses<br>" + NEWLINE +
50 "<input type=\"checkbox\" id=\"verifiyResponses\">Verify responded messages<br>" + NEWLINE +
51 "<input type=\"button\" value=\"Start Benchmark\"" + NEWLINE +
52 " onclick=\"startBenchmark()\" />" + NEWLINE +
53 "<h3>Output</h3>" + NEWLINE +
54 "<textarea id=\"output\" style=\"width:500px;height:300px;\"></textarea>" + NEWLINE +
55 "<br>" + NEWLINE +
56 "<input type=\"button\" value=\"Clear\" onclick=\"clearText()\">" + NEWLINE +
57 "</form>" + NEWLINE +
58
59 "<script type=\"text/javascript\">" + NEWLINE +
60 "var benchRunning = false;" + NEWLINE +
61 "var messageSize = 0;" + NEWLINE +
62 "var totalMessages = 0;" + NEWLINE +
63 "var rcvdMessages = 0;" + NEWLINE +
64 "var isBinary = true;" + NEWLINE +
65 "var isSingle = true;" + NEWLINE +
66 "var verifiyResponses = false;" + NEWLINE +
67 "var benchData = null;" + NEWLINE +
68 "var startTime;" + NEWLINE +
69 "var endTime;" + NEWLINE +
70 "var socket;" + NEWLINE +
71 "var output = document.getElementById('output');" + NEWLINE +
72 "var connectionLabel = document.getElementById('connectionLabel');" + NEWLINE +
73 "if (!window.WebSocket) {" + NEWLINE +
74 " window.WebSocket = window.MozWebSocket;" + NEWLINE +
75 '}' + NEWLINE +
76 "if (window.WebSocket) {" + NEWLINE +
77 " socket = new WebSocket(\"" + webSocketLocation + "\");" + NEWLINE +
78 " socket.binaryType = 'arraybuffer';" + NEWLINE +
79 " socket.onmessage = function(event) {" + NEWLINE +
80 " if (verifiyResponses) {" + NEWLINE +
81 " if (isBinary) {" + NEWLINE +
82 " if (!(event.data instanceof ArrayBuffer) || " + NEWLINE +
83 " event.data.byteLength != benchData.byteLength) {" + NEWLINE +
84 " onInvalidResponse(benchData, event.data);" + NEWLINE +
85 " return;" + NEWLINE +
86 " } else {" + NEWLINE +
87 " var v = new Uint8Array(event.data);" + NEWLINE +
88 " for (var j = 0; j < benchData.byteLength; j++) {" + NEWLINE +
89 " if (v[j] != benchData[j]) {" + NEWLINE +
90 " onInvalidResponse(benchData, event.data);" + NEWLINE +
91 " return;" + NEWLINE +
92 " }" + NEWLINE +
93 " }" + NEWLINE +
94 " }" + NEWLINE +
95 " } else {" + NEWLINE +
96 " if (event.data != benchData) {" + NEWLINE +
97 " onInvalidResponse(benchData, event.data);" + NEWLINE +
98 " return;" + NEWLINE +
99 " }" + NEWLINE +
100 " }" + NEWLINE +
101 " }" + NEWLINE +
102 " rcvdMessages++;" + NEWLINE +
103 " if (rcvdMessages == totalMessages) {" + NEWLINE +
104 " onFinished();" + NEWLINE +
105 " } else if (isSingle) {" + NEWLINE +
106 " socket.send(benchData);" + NEWLINE +
107 " }" + NEWLINE +
108 " };" + NEWLINE +
109 " socket.onopen = function(event) {" + NEWLINE +
110 " connectionLabel.innerHTML = \"Connected\";" + NEWLINE +
111 " };" + NEWLINE +
112 " socket.onclose = function(event) {" + NEWLINE +
113 " benchRunning = false;" + NEWLINE +
114 " connectionLabel.innerHTML = \"Disconnected\";" + NEWLINE +
115 " };" + NEWLINE +
116 "} else {" + NEWLINE +
117 " alert(\"Your browser does not support Web Socket.\");" + NEWLINE +
118 '}' + NEWLINE +
119 NEWLINE +
120 "function onInvalidResponse(sent,recvd) {" + NEWLINE +
121 " socket.close();" + NEWLINE +
122 " alert(\"Error: Sent data did not match the received data!\");" + NEWLINE +
123 "}" + NEWLINE +
124 NEWLINE +
125 "function clearText() {" + NEWLINE +
126 " output.value=\"\";" + NEWLINE +
127 "}" + NEWLINE +
128 NEWLINE +
129 "function createBenchData() {" + NEWLINE +
130 " if (isBinary) {" + NEWLINE +
131 " benchData = new Uint8Array(messageSize);" + NEWLINE +
132 " for (var i=0; i < messageSize; i++) {" + NEWLINE +
133 " benchData[i] += Math.floor(Math.random() * 255);" + NEWLINE +
134 " }" + NEWLINE +
135 " } else { " + NEWLINE +
136 " benchData = \"\";" + NEWLINE +
137 " for (var i=0; i < messageSize; i++) {" + NEWLINE +
138 " benchData += String.fromCharCode(Math.floor(Math.random() * (123 - 65) + 65));" + NEWLINE +
139 " }" + NEWLINE +
140 " }" + NEWLINE +
141 "}" + NEWLINE +
142 NEWLINE +
143 "function startBenchmark(message) {" + NEWLINE +
144 " if (!window.WebSocket || benchRunning) { return; }" + NEWLINE +
145 " if (socket.readyState == WebSocket.OPEN) {" + NEWLINE +
146 " isBinary = document.getElementById('typeBinary').checked;" + NEWLINE +
147 " isSingle = document.getElementById('modeSingle').checked;" + NEWLINE +
148 " verifiyResponses = document.getElementById('verifiyResponses').checked;" + NEWLINE +
149 " messageSize = parseInt(document.getElementById('messageSize').value);" + NEWLINE +
150 " totalMessages = parseInt(document.getElementById('nrMessages').value);" + NEWLINE +
151 " if (isNaN(messageSize) || isNaN(totalMessages)) return;" + NEWLINE +
152 " createBenchData();" + NEWLINE +
153 " output.value = output.value + '\\nStarting Benchmark';" + NEWLINE +
154 " rcvdMessages = 0;" + NEWLINE +
155 " benchRunning = true;" + NEWLINE +
156 " startTime = new Date();" + NEWLINE +
157 " if (isSingle) {" + NEWLINE +
158 " socket.send(benchData);" + NEWLINE +
159 " } else {" + NEWLINE +
160 " for (var i = 0; i < totalMessages; i++) socket.send(benchData);" + NEWLINE +
161 " }" + NEWLINE +
162 " } else {" + NEWLINE +
163 " alert(\"The socket is not open.\");" + NEWLINE +
164 " }" + NEWLINE +
165 '}' + NEWLINE +
166 NEWLINE +
167 "function onFinished() {" + NEWLINE +
168 " endTime = new Date();" + NEWLINE +
169 " var duration = (endTime - startTime) / 1000.0;" + NEWLINE +
170 " output.value = output.value + '\\nTest took: ' + duration + 's';" + NEWLINE +
171 " var messagesPerS = totalMessages / duration;" + NEWLINE +
172 " output.value = output.value + '\\nPerformance: ' + messagesPerS + ' Messages/s';" + NEWLINE +
173 " output.value = output.value + ' in each direction';" + NEWLINE +
174 " output.value = output.value + '\\nRound trip: ' + 1000.0/messagesPerS + 'ms';" + NEWLINE +
175 " var throughput = messageSize * totalMessages / duration;" + NEWLINE +
176 " var throughputText;" + NEWLINE +
177 " if (isBinary) throughputText = throughput / (1024*1024) + ' MB/s';" + NEWLINE +
178 " else throughputText = throughput / (1000*1000) + ' MChars/s';" + NEWLINE +
179 " output.value = output.value + '\\nThroughput: ' + throughputText;" + NEWLINE +
180 " output.value = output.value + ' in each direction';" + NEWLINE +
181 " benchRunning = false;" + NEWLINE +
182 "}" + NEWLINE +
183 "</script>" + NEWLINE +
184 "</body>" + NEWLINE +
185 "</html>" + NEWLINE, CharsetUtil.US_ASCII);
186 }
187
188 private WebSocketServerBenchmarkPage() {
189 // Unused
190 }
191 }
以上源码来自： 即时通讯网 - 即时通讯开发者社区！