1 files changed, 49 insertions, 0 deletions
diff --git a/benchmarks/server-http2.js b/benchmarks/server-http2.js
new file mode 100644
index 0000000..0be99cd
--- /dev/null
+++ b/benchmarks/server-http2.js
@@ -0,0 +1,49 @@
+'use strict'
+
+const { unlinkSync, readFileSync } = require('fs')
+const { createSecureServer } = require('http2')
+const os = require('os')
+const path = require('path')
+const cluster = require('cluster')
+
+const key = readFileSync(path.join(__dirname, '..', 'test', 'fixtures', 'key.pem'), 'utf8')
+const cert = readFileSync(path.join(__dirname, '..', 'test', 'fixtures', 'cert.pem'), 'utf8')
+
+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
+
+const sessionTimeout = 600e3 // 10 minutes
+
+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 = createSecureServer(
+    {
+      key,
+      cert,
+      allowHTTP1: true,
+      sessionTimeout
+    },
+    (req, res) => {
+      setTimeout(() => {
+        res.end(buf)
+      }, timeout)
+    }
+  )
+
+  server.keepAliveTimeout = 600e3
+
+  server.listen(port)
+}