1 files changed, 33 insertions, 0 deletions
diff --git a/benchmarks/server.js b/benchmarks/server.js
new file mode 100644
index 0000000..e1a32e8
--- /dev/null
+++ b/benchmarks/server.js
@@ -0,0 +1,33 @@
+'use strict'
+
+const { unlinkSync } = require('fs')
+const { createServer } = require('http')
+const os = require('os')
+const path = require('path')
+const cluster = require('cluster')
+
+const socketPath = path.join(os.tmpdir(), 'undici.sock')
+
+const port = process.env.PORT || socketPath
+const timeout = parseInt(process.env.TIMEOUT, 10) || 1
+const workers = parseInt(process.env.WORKERS) || os.cpus().length
+
+if (cluster.isPrimary) {
+  try {
+    unlinkSync(socketPath)
+  } catch (_) {
+    // Do nothing if the socket does not exist
+  }
+
+  for (let i = 0; i < workers; i++) {
+    cluster.fork()
+  }
+} else {
+  const buf = Buffer.alloc(64 * 1024, '_')
+  const server = createServer((req, res) => {
+    setTimeout(function () {
+      res.end(buf)
+    }, timeout)
+  }).listen(port)
+  server.keepAliveTimeout = 600e3
+}