fixup: avoid socket alloc

websockets · Jan 5, 2022 · e9a4ca2 · e9a4ca2
1 parent 16a7034
commit e9a4ca2
Showing 1 changed file with 4 additions and 19 deletions.
diff --git a/bench/bench.mjs b/bench/bench.mjs
@@ -4,9 +4,9 @@ import cronometro from 'cronometro';
 const server = net
   .createServer((socket) => {
     socket.resume();
-    socket.on('error', () => {});
   })
   .listen(0);
+const socket = net.connect(server.address().port, '0.0.0.0');
 
 function toBuffer(data) {
   toBuffer.readOnly = true;
@@ -53,37 +53,22 @@ function toBuffer2(data) {
 cronometro(
   {
     string(cb) {
-      const socket = net.connect(server.address().port, '0.0.0.0');
       for (let n = 0; n < 16; ++n) {
         socket.write(Buffer.allocUnsafe(4096).toString());
       }
-      socket.end();
-      socket.on('finish', () => {
-        socket.on('error', () => {}).destroy();
-        cb();
-      });
+      socket.write('eof', cb);
     },
     toBuffer(cb) {
-      const socket = net.connect(server.address().port, '0.0.0.0');
       for (let n = 0; n < 16; ++n) {
         socket.write(toBuffer(Buffer.allocUnsafe(4096).toString()));
       }
-      socket.end();
-      socket.on('finish', () => {
-        socket.on('error', () => {}).destroy();
-        cb();
-      });
+      socket.write('eof', cb);
     },
     toBuffer2(cb) {
-      const socket = net.connect(server.address().port, '0.0.0.0');
       for (let n = 0; n < 16; ++n) {
         socket.write(toBuffer2(Buffer.allocUnsafe(4096).toString()));
       }
-      socket.end();
-      socket.on('finish', () => {
-        socket.on('error', () => {}).destroy();
-        cb();
-      });
+      socket.write('eof', cb);
     }
   },
   {