loadtest 8.0.2 → 8.0.3

package/bin/tcp-performance.js

@@ -1,21 +1,56 @@
 import {loadTest, startServer} from '../index.js'
+const cluster = await import('cluster')
+import {getHalfCores} from '../lib/cluster.js'
 
-const port = 7359;
+const port = 7360;
 const serverOptions = {port}
 
+if (cluster.isPrimary) {
+	await runTcpPerformanceTest()
+} else {
+	await runServer()
+}
 
 async function runTcpPerformanceTest() {
-	const server = await startServer(serverOptions)
+	const workers = await startWorkers()
 	const options = {
 		url: `http://localhost:${port}`,
 		method: 'GET',
 		tcp: true,
+		concurrency: 10,
 	};
 	const result = await loadTest(options)
-	await server.close()
-	console.log(`Requests received: ${server.totalRequests}`)
 	result.show()
+	console.log(`Test finished; closing server on ${workers.length} cores`)
+	await stopServer(workers)
+}
+
+function startWorkers() {
+	return new Promise(resolve => {
+		const cores = getHalfCores()
+		const workers = []
+		for (let i = 0; i < cores; i++) {
+			const worker = cluster.fork()
+			worker.on('message', async () => {
+				workers.push(worker)
+				if (workers.length != cores) {
+					return
+				}
+				console.log(`Server started on ${workers.length} cores`)
+				return resolve(workers)
+			})
+		}
+	})
 }
 
-await runTcpPerformanceTest()
+function stopServer(workers) {
+	for (const worker of workers) {
+		worker.kill('SIGTERM')
+	}
+}
+
+async function runServer() {
+	await startServer(serverOptions)
+	process.send('server ready')
+}
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "loadtest",
-	"version": "8.0.2",
+	"version": "8.0.3",
 	"type": "module",
 	"description": "Run load tests for your web application. Mostly ab-compatible interface, with an option to force requests per second. Includes an API for automated load testing.",
 	"homepage": "https://github.com/alexfernandez/loadtest",