mirror of https://github.com/nodejs/node.git
70 lines
1.5 KiB
JavaScript
70 lines
1.5 KiB
JavaScript
'use strict';
|
|
|
|
const common = require('../common.js');
|
|
const { Worker } = require('worker_threads');
|
|
const path = require('path');
|
|
const bench = common.createBenchmark(main, {
|
|
workers: [1],
|
|
payload: ['string', 'object'],
|
|
sendsPerBroadcast: [1, 10],
|
|
n: [1e5],
|
|
});
|
|
|
|
const workerPath = path.resolve(__dirname, '..', 'fixtures', 'echo.worker.js');
|
|
|
|
function main({ n, workers, sendsPerBroadcast: sends, payload: payloadType }) {
|
|
const expectedPerBroadcast = sends * workers;
|
|
let payload;
|
|
let readies = 0;
|
|
let broadcasts = 0;
|
|
let msgCount = 0;
|
|
|
|
switch (payloadType) {
|
|
case 'string':
|
|
payload = 'hello world!';
|
|
break;
|
|
case 'object':
|
|
payload = { action: 'pewpewpew', powerLevel: 9001 };
|
|
break;
|
|
default:
|
|
throw new Error('Unsupported payload type');
|
|
}
|
|
|
|
const workerObjs = [];
|
|
|
|
for (let i = 0; i < workers; ++i) {
|
|
const worker = new Worker(workerPath);
|
|
workerObjs.push(worker);
|
|
worker.on('online', onOnline);
|
|
worker.on('message', onMessage);
|
|
}
|
|
|
|
function onOnline() {
|
|
if (++readies === workers) {
|
|
bench.start();
|
|
broadcast();
|
|
}
|
|
}
|
|
|
|
function broadcast() {
|
|
if (broadcasts++ === n) {
|
|
bench.end(n);
|
|
for (const worker of workerObjs) {
|
|
worker.unref();
|
|
}
|
|
return;
|
|
}
|
|
for (const worker of workerObjs) {
|
|
for (let i = 0; i < sends; ++i)
|
|
worker.postMessage(payload);
|
|
}
|
|
}
|
|
|
|
function onMessage() {
|
|
if (++msgCount === expectedPerBroadcast) {
|
|
msgCount = 0;
|
|
broadcast();
|
|
}
|
|
}
|
|
}
|