const DynamicThreadPool = require('../lib/dynamic')
const Pool = require('worker-threads-pool')
const tasks = 1000
-const size = 10
+const size = 16
// pools
const externalPool = new Pool({ max: size })
-const fixedPool = new FixedThreadPool(size, './yourWorker.js', { maxTasks: 10000 })
-const dynamicPool = new DynamicThreadPool(size, size * 2, './yourWorker.js', { maxTasks: 10000 })
+const fixedPool = new FixedThreadPool(size, './yourWorker.js', {
+ maxTasks: 10000
+})
+const dynamicPool = new DynamicThreadPool(
+ size / 2,
+ size * 3,
+ './yourWorker.js',
+ { maxTasks: 10000 }
+)
// data
const workerData = { proof: 'ok' }
async function fixedTest () {
let executions = 0
const time = Date.now()
- for (let i = 0; i < tasks; i++) {
- await fixedPool.execute(workerData)
- executions++
+ for (let i = 0; i <= tasks; i++) {
+ fixedPool.execute(workerData).then(res => {
+ executions++
+ if (executions === tasks) {
+ console.log(
+ `Fixed pool take ${Date.now() - time} to work on ${executions} tasks`
+ )
+ }
+ })
}
- console.log(`Fixed pool take ${Date.now() - time} to work on ${executions} tasks`)
}
async function dynamicTest () {
let executions = 0
const time = Date.now()
- for (let i = 0; i < tasks; i++) {
- await dynamicPool.execute(workerData)
- executions++
+ for (let i = 0; i <= tasks; i++) {
+ dynamicPool.execute(workerData).then(res => {
+ executions++
+ if (executions === tasks) {
+ console.log(
+ `Dynamic pool take ${Date.now() -
+ time} to work on ${executions} tasks`
+ )
+ }
+ })
}
- console.log(`Dynamic pool take ${Date.now() - time} to work on ${executions} tasks`)
}
async function externalPoolTest () {
let executions = 0
const time = Date.now()
- for (let i = 0; i < tasks; i++) {
- await new Promise((resolve, reject) => {
- externalPool.acquire('./externalWorker.js', { workerData: workerData }, (err, worker) => {
- if (err) {
- return reject(err)
+ for (let i = 0; i <= tasks; i++) {
+ new Promise((resolve, reject) => {
+ externalPool.acquire(
+ './externalWorker.js',
+ { workerData: workerData },
+ (err, worker) => {
+ if (err) {
+ return reject(err)
+ }
+ worker.on('error', reject)
+ worker.on('message', res => {
+ executions++
+ resolve(res)
+ })
}
- worker.on('error', reject)
- worker.on('message', res => {
- executions++
- resolve(res)
- })
- })
+ )
+ }).then(res => {
+ if (tasks === executions) {
+ console.log(
+ `External pool take ${Date.now() -
+ time} to work on ${executions} tasks`
+ )
+ }
})
}
- console.log(`External pool take ${Date.now() - time} to work on ${executions} tasks`)
}
async function test () {
- await fixedTest()
- await dynamicTest()
- await externalPoolTest()
+ fixedTest()
+ dynamicTest()
+ externalPoolTest()
}
test()