worker-threads-toolkit/worker-function/pool-factory.js

Worker threads toolkit

'use strict';

const { Worker } = require('worker_threads');
const { createPool } = require('generic-pool');
const WORKER_TEMPLATE = require('./worker-template');

/**
 * @param {Object} options
 * @param {Object} workerOptions
 * @returns {*}
 */
module.exports = function (options, workerOptions) {
    return createPool({
        create() {
            return new Promise(function (resolve, reject) {
                const worker = new Worker(WORKER_TEMPLATE, workerOptions);

                worker.on('online', function () {
                    worker.removeAllListeners();
                    resolve(worker);
                });

                worker.on('error', reject);
            });
        },

        destroy(worker) {
            worker.terminate();
        },

        validate(worker) {
            return worker && -1 !== worker.threadId;
        }
    }, options);
};