'use strict'
-const {
- isMainThread, parentPort
-} = require('worker_threads')
+const { isMainThread, parentPort } = require('worker_threads')
const { AsyncResource } = require('async_hooks')
/**
constructor (fn, opts) {
super('worker-thread-pool:pioardi')
this.opts = opts || {}
- this.maxInactiveTime = this.opts.maxInactiveTime || (1000 * 60)
+ this.maxInactiveTime = this.opts.maxInactiveTime || 1000 * 60
+ this.async = !!this.opts.async
this.lastTask = Date.now()
if (!fn) throw new Error('Fn parameter is mandatory')
// keep the worker active
if (!isMainThread) {
- this.interval = setInterval(this._checkAlive.bind(this), this.maxInactiveTime / 2)
+ this.interval = setInterval(
+ this._checkAlive.bind(this),
+ this.maxInactiveTime / 2
+ )
this._checkAlive.bind(this)()
}
- parentPort.on('message', (value) => {
+ parentPort.on('message', value => {
if (value && value.data && value._id) {
// here you will receive messages
// console.log('This is the main thread ' + isMainThread)
- this.runInAsyncScope(this._run.bind(this), this, fn, value)
+ if (this.async) {
+ this.runInAsyncScope(this._runAsync.bind(this), this, fn, value)
+ } else {
+ this.runInAsyncScope(this._run.bind(this), this, fn, value)
+ }
} else if (value.parent) {
// save the port to communicate with the main thread
// this will be received once
}
_checkAlive () {
- if ((Date.now() - this.lastTask) > this.maxInactiveTime) {
+ if (Date.now() - this.lastTask > this.maxInactiveTime) {
this.parent.postMessage({ kill: 1 })
}
}
this.lastTask = Date.now()
}
}
+
+ _runAsync (fn, value) {
+ fn(value.data)
+ .then(res => {
+ this.parent.postMessage({ data: res, _id: value._id })
+ this.lastTask = Date.now()
+ })
+ .catch(e => {
+ this.parent.postMessage({ error: e, _id: value._id })
+ this.lastTask = Date.now()
+ })
+ }
}
module.exports.ThreadWorker = ThreadWorker