[Nodejs - worker-threads]: Is there an option for the main thread to wait till all worker threads are done?

Is there an option by which we can make the main thread wait until all worker threads are done?
In the below code, mainThread.js simply creates workers.
The worker takes long time to execute.
In this situation, how do we instruct the main thread to wait till all threads are done with execution?
Something similar to thread.join() in java.

mainThread.js:

const {Worker, parentPort, workerData} = require('worker_threads');
const path = require('path');

const startWorker = function(path, workerData, callback) {
    //Create worker thread
    let worker = new Worker(path, {workerData});

    //Listen to events of the worker thread
    worker.on('message', (msg) => {
        callback(null, msg);
    });
    worker.on('error', callback);
    worker.on('exit', (code) => {
        if(code !== 0) {
            console.error(`Error: Non-zero exit code ${code}`)
        }
    });

    //Return the worker thread
    return worker;
};

console.log('This is Main Thread');

//Create and start worker threads
let thread1 = startWorker(__dirname + path.sep + 'workerThread.js', "user1", (error, result) => {
    if(error) return console.error(error);
    console.log('Result: ', result);
});

let thread2 = startWorker(__dirname + path.sep + 'workerThread.js', "user2", (error, result) => {
    if(error) return console.error(error);
    console.log('Result: ', result);
});

workerThread.js:

const {workerData, parentPort} = require('worker_threads');
setTimeout(() => {
    console.log(workerData);
    parentPort.postMessage(workerData + "done");
}, 2000)

Execution: node --experimental-worker mainThread.js

PS: We can try to make it work by running it as an http server instance or by having a setTimeout with a huge value. But, is there an option by nodejs itself to achieve it is the question.

Recommended topics

Hot tags