Function main

benchmark/websocket/simple.js:44–99

(conf)

}

function main(conf) {
  const frame = createFrame(Buffer.alloc(conf.size).fill('.'), conf.useBinary === 'true' ? 2 : 1);
  const server = http.createServer();
  server.on('upgrade', (req, socket) => {
    const key = crypto
      .createHash('sha1')
      .update(req.headers['sec-websocket-key'] + GUID)
      .digest('base64');

    let bytesReceived = 0;
    let roundtrip = 0;

    socket.on('data', function onData(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, () => {
    const ws = new WebSocket(`ws://localhost:${server.address().port}`);
    ws.addEventListener('open', () => {
      bench.start();
    });

    ws.addEventListener('message', (event) => {
      ws.send(event.data);
    });
  });
}