-import { DynamicThreadPool, DynamicThreadPoolOptions } from 'poolifier';
+import { DynamicThreadPool, PoolOptions } from 'poolifier';
import Constants from '../utils/Constants';
import Utils from '../utils/Utils';
+import { Worker } from 'worker_threads';
import { WorkerData } from '../types/Worker';
import Wrk from './Wrk';
-export default class WorkerDynamicPool extends Wrk {
+export default class WorkerDynamicPool<T> extends Wrk {
private pool: DynamicPool;
/**
* @return {Promise<void>}
* @public
*/
- public async addElement(elementData: WorkerData): Promise<void> {
+ public async stop(): Promise<void> {
+ return this.pool.destroy();
+ }
+
+ /**
+ *
+ * @return {Promise<void>}
+ * @public
+ */
+ public async addElement(elementData: T): Promise<void> {
await this.pool.execute(elementData);
// Start worker sequentially to optimize memory at startup
await Utils.sleep(Constants.START_WORKER_DELAY);
class DynamicPool extends DynamicThreadPool<WorkerData> {
private static instance: DynamicPool;
- private constructor(min: number, max: number, filename: string, opts?: DynamicThreadPoolOptions) {
- super(min, max, filename, opts);
+ private constructor(min: number, max: number, workerScript: string, opts?: PoolOptions<Worker>) {
+ super(min, max, workerScript, opts);
}
- public static getInstance(min: number, max: number, filename: string): DynamicPool {
+ public static getInstance(min: number, max: number, workerScript: string): DynamicPool {
if (!DynamicPool.instance) {
- DynamicPool.instance = new DynamicPool(min, max, filename,
+ DynamicPool.instance = new DynamicPool(min, max, workerScript,
{
exitHandler: (code) => {
if (code !== 0) {