benchmark: add undici websocket benchmark

Refs: nodejs/performance#114
PR-URL: #50586
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
Reviewed-By: Matthew Aitken <maitken033380023@gmail.com>

Author: Chenyu Yang

benchmark/websocket/simple.js

@@ -0,0 +1,100 @@
+'use strict';
+
+constcommon=require('../common.js');
+constcrypto=require('crypto');
+consthttp=require('http');
+const{ WebSocket }=require('../../deps/undici/undici');
+
+constGUID='258EAFA5-E914-47DA-95CA-C5AB0DC85B11';
+
+constconfigs={
+size: [64,16*1024,128*1024,1024*1024],
+useBinary: ['true','false'],
+roundtrips: [5000,1000,100],
+};
+
+constbench=common.createBenchmark(main,configs);
+
+functioncreateFrame(data,opcode){
+letinfoLength=2;
+letpayloadLength=data.length;
+
+if(payloadLength>=65536){
+infoLength+=8;
+payloadLength=127;
+}elseif(payloadLength>125){
+infoLength+=2;
+payloadLength=126;
+}
+
+constinfo=Buffer.alloc(infoLength);
+
+info[0]=opcode|0x80;
+info[1]=payloadLength;
+
+if(payloadLength===126){
+info.writeUInt16BE(data.length,2);
+}elseif(payloadLength===127){
+info[2]=info[3]=0;
+info.writeUIntBE(data.length,4,6);
+}
+
+returnBuffer.concat([info,data]);
+}
+
+functionmain(conf){
+constframe=createFrame(Buffer.alloc(conf.size).fill('.'),conf.useBinary==='true' ? 2 : 1);
+constserver=http.createServer();
+server.on('upgrade',(req,socket)=>{
+constkey=crypto
+.createHash('sha1')
+.update(req.headers['sec-websocket-key']+GUID)
+.digest('base64');
+
+letbytesReceived=0;
+letroundtrip=0;
+
+socket.on('data',functiononData(chunk){
+bytesReceived+=chunk.length;
+
+if(bytesReceived===frame.length+4){// +4 for the mask.
+// Message completely received.
+bytesReceived=0;
+
+if(++roundtrip===conf.roundtrips){
+socket.removeListener('data',onData);
+socket.resume();
+socket.end();
+server.close();
+
+bench.end(conf.roundtrips);
+}else{
+socket.write(frame);
+}
+}
+});
+
+socket.write(
+[
+'HTTP/1.1 101 Switching Protocols',
+'Upgrade: websocket',
+'Connection: Upgrade',
+`Sec-WebSocket-Accept: ${key}`,
+'\r\n',
+].join('\r\n'),
+);
+
+socket.write(frame);
+});
+
+server.listen(0,()=>{
+constws=newWebSocket(`ws://localhost:${server.address().port}`);
+ws.addEventListener('open',()=>{
+bench.start();
+});
+
+ws.addEventListener('message',(event)=>{
+ws.send(event.data);
+});
+});
+}

test/benchmark/test-benchmark-websocket.js

@@ -0,0 +1,9 @@
+'use strict';
+
+constcommon=require('../common');
+if(!common.enoughTestMem)
+common.skip('Insufficient memory for Websocket benchmark test');
+
+construnBenchmark=require('../common/benchmark');
+
+runBenchmark('websocket',{NODEJS_BENCHMARK_ZERO_ALLOWED: 1});