Estoy usando el siguiente código para realizar un cálculo intensivo en costos usando múltiples subprocesos en NodeJS:
Trabajador:
import { workerData, parentPort } from 'worker_threads'; let digits = workerData as number[]; let digitCount = digits.length; // Worker Code parentPort.on('message', (msg) => { let [start, end] = msg; for(let i = start; i < end; i++) { let res = calculate(digits, digitCount, msg); if(res) { parentPort.postMessage(res); } } parentPort.postMessage(true); process.exit(0); });Principal:
import { Worker } from 'worker_threads'; import { cpus } from 'os'; export class WorkerPool { private workers: Worker[]; private resolve: (value: void | PromiseLike<void>) => void; private done: number; constructor(digits: number[], occurances: Map<number, number | false>, private count: number = (cpus().length - 1)) { if(count < 1) count = 1; this.workers = new Array<Worker>(count); for(let i = 0; i < count; i++) { let worker: Worker = new Worker(require.resolve('./worker'), { workerData: digits, }) as Worker; this.workers[i] = worker; worker.on('message', (val: number | true) => { if(val === true) { if(++this.done >= count) { this.resolve(); } return; } let idx = occurances.get(val); if(idx === undefined) { occurances.set(val, i); } else if(idx !== false) { occurances.set(val, false); } }); } } calculate(maxi: number) { return new Promise<void>((resolve, reject) => { this.resolve = resolve; let d = Math.floor(maxi / this.count); let r = maxi % this.count; let start = 0; for(let i = 0; i < this.count; i++) { let end = start + d + (i < r ? 1 : 0); this.workers[i].postMessage([start, end]); start = end; } }); } }La función de cálculo es solo una función relativamente simple, sin embargo, se ejecuta alrededor de mil millones de veces. Es por eso que traté de usar subprocesos múltiples. Sin embargo, cuando lo hago
await pool.calculate(1000000000)el cómputo no termina y el consumo de Ram pasa a 64GB. Cuando se ejecuta la función de cálculo en el subproceso principal, funciona como se esperaba, así que supongo que es una falla en mi clase WorkerPool.