diff --git a/libs/utils/multithread.js b/libs/utils/multithread.js index 56128b4..f64b369 100644 --- a/libs/utils/multithread.js +++ b/libs/utils/multithread.js @@ -10,27 +10,28 @@ export function initWorkers(threads) { for (const i in workerFarm) { const worker = workerFarm[i] worker.postMessage({ - type: 'setName', + type: 'initWorker', name: `Worker ${parseInt(i) + 1}` }) } } -export function spread(threads, script, data, shouldEval) { - return new Promise((resolve, reject) => { - const worker = new Worker( - script, - { - workerData: data, - eval: shouldEval ?? false - } - ) +let spreadIndex = 0 +export function spread(script, data) { + return new Promise( + (resolve, reject) => { + const worker = workerFarm[spreadIndex] + worker.postMessage({ + type: 'runScript', + script, + data + }) - worker.on('message', resolve) - worker.on('error', reject) - worker.on('exit', (code) => { - if (code !== 0) - reject(new Error(`Process stopped with code: ${code}`)) - }) - }) + worker.once('message', result => resolve(result)) + + ++spreadIndex + if(spreadIndex >= workerFarm.length) + spreadIndex = 0 + } + ) } \ No newline at end of file diff --git a/libs/utils/multithread/fibonacci.js b/libs/utils/multithread/fibonacci.js index c577302..bf93a79 100644 --- a/libs/utils/multithread/fibonacci.js +++ b/libs/utils/multithread/fibonacci.js @@ -1,23 +1,29 @@ -import { workerData, parentPort } from 'worker_threads' +import initLog from '../log.js' -let nums = [1, 1] +const log = initLog('Fibonacci') -for (let i = 0; i < 42; i++) { - nums.push(nums[nums.length - 2] + nums[nums.length - 1]) -} +export function main(data) { + return new Promise( + (resolve, reject) => { + let nums = [1, 1] + + for (let i = 0; i < 44; i++) { + nums.push(nums[nums.length - 2] + nums[nums.length - 1]) + } + + let primes = [] + for (const num of nums) { + let isPrime = true + for (let i = 2; i < num / 2; ++i) { + if (num % i === 0) + isPrime = false + } + + if (isPrime) + primes.push(num) + } -let primes = [] -for (const num of nums) { - let isPrime = true - for (let i = 2; i < num / 2; ++i) { - if (num % i === 0) - isPrime = false - } - - if (isPrime) - primes.push(num) -} - -console.log("DONE!") - -parentPort.postMessage(primes) \ No newline at end of file + resolve(primes) + } + ) +} \ No newline at end of file diff --git a/libs/utils/multithread/genericWorker.js b/libs/utils/multithread/genericWorker.js index 73a4ff4..e146719 100644 --- a/libs/utils/multithread/genericWorker.js +++ b/libs/utils/multithread/genericWorker.js @@ -1,30 +1,37 @@ import { workerData, parentPort } from 'worker_threads' import initLog from '../log.js' -let script = '' +let runScript = '' let myName = '' let log = initLog(myName) parentPort.on( 'message', - message => { + async message => { switch (message.type) { - case 'setName': + case 'initWorker': myName = message.name log = initLog(myName) - log(`Changed name to '${myName}'.`) + process.env.DEBUG && log(`Started new thread -> «${myName}»`) break - case 'setScript': - script = message.script - log = initLog(myName) - log(`Changed name to '${myName}'.`) + case 'runScript': + runScript = message.script + process.env.DEBUG && log(`Running script -> «${runScript}»`) + + import(runScript).then( + script => script.main(message.data).then( + result => { + parentPort.postMessage(result) + process.env.DEBUG && log(`Script done -> «${runScript}»`) + } + ) + ) break - default: - + log(`Invalid message -> ${message.type ?? 'No message type provided!'}`) break } } diff --git a/tests/multihread.test.js b/tests/multihread.test.js index 71c2f1d..4a192b9 100644 --- a/tests/multihread.test.js +++ b/tests/multihread.test.js @@ -1,3 +1,21 @@ import { initWorkers, spread } from '../libs/utils/multithread.js' -initWorkers(8) \ No newline at end of file +initWorkers(10) + +let processes = 20 +let ran = 0 +for (let i = 0; i < processes; ++i) { + spread( + './fibonacci.js', + {} + ).then( + result => { + console.log(i) + + ++ran + if (ran === processes) { + process.exit(0) + } + } + ) +} \ No newline at end of file