loadtest 8.0.1 → 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/doc/tcp-sockets.md

@@ -1,7 +1,10 @@
-# TCP Sockets Performance
+# Load Testing with TCP Sockets
 
-To improve performance the author has tried out using raw TCP sockets
-using the [net module](https://nodejs.org/api/net.html),
+The `loadtest` module has impressive performance,
+and it has got better during the years as the Node.js core improves.
+Heavily inspired by `autocannon`,
+the author has tried out using raw TCP sockets to improve performance.
+They use the [net module](https://nodejs.org/api/net.html)
 instead of the [HTTP module](https://nodejs.org/api/http.html).
 This is the story of how it went.
 
@@ -113,8 +116,9 @@ Keep-alive cannot be used with `ab` as far as the author knows.
 
 #### Autocannon
 
-The [autocannon](https://www.npmjs.com/package/autocannon) package uses by default
-10 concurrent connections with keep-alive enabled:
+Next we will try out [`autocannon`](https://www.npmjs.com/package/autocannon),
+the package that actually inspired this approach.
+`autocannon` uses by default 10 concurrent connections with keep-alive enabled:
 
 ```console
 $ autocannon --version
@@ -195,10 +199,10 @@ We are around **20 krps**.
 Again quite far from the 57 krps by `autocannon`;
 close to `ab` but it doesn't use keep-alive so the comparison is meaningless.
 
-### Proof of Concept
+### Proof of Concept: Barebones
 
 For the first implementation we want to learn if the bare sockets implementation is worth the time.
-In this naïve implementation we open the socket,
+In this naïve barebones implementation we open the socket,
 send a short canned request without taking into account any parameters or headers:
 
 ```js

package/lib/parser.js

@@ -1,4 +1,3 @@
-import microprofiler from 'microprofiler'
 
 const bodySeparator = '\r\n\r\n'
 const lineSeparator = '\r\n'
@@ -36,13 +35,10 @@ export class Parser {
 			// cannot parse yet
 			return
 		}
-		//const start1 = microprofiler.start()
 		this.packetInfo = new PacketInfo(this.pending.length, division)
 		const messageHeader = this.pending.subarray(0, division)
 		const messageBody = this.pending.subarray(division + 4)
 		this.parseFirstLine(messageHeader)
-		//microprofiler.measureFrom(start1, 'parse first line', 100000)
-		//const start2 = microprofiler.start()
 		const key = this.packetInfo.getKey()
 		const existing = packetInfos.get(key)
 		if (existing && this.packetInfo.isDuplicated(existing)) {
@@ -52,10 +48,7 @@ export class Parser {
 		}
 		packetInfos.set(key, this.packetInfo)
 		this.parseHeaders(messageHeader)
-		//microprofiler.measureFrom(start2, 'parse headers', 100000)
-		const start3 = microprofiler.start()
 		this.parseBody(messageBody)
-		microprofiler.measureFrom(start3, 'parse body', 100000)
 	}
 
 	parseFirstLine(messageHeader) {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "loadtest",
-	"version": "8.0.1",
+	"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",