From 810f4caf2bf0b12d5c429d1d21af62b1fe67f965 Mon Sep 17 00:00:00 2001
From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= <jerome.benoit@piment-noir.org>
Date: Thu, 8 Jun 2023 12:37:14 +0200
Subject: [PATCH] refactor: consolidate worker events enum
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---
 src/worker/WorkerSet.ts   |  3 ++-
 src/worker/WorkerTypes.ts | 12 +++++++++++-
 src/worker/index.ts       |  1 +
 3 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/worker/WorkerSet.ts b/src/worker/WorkerSet.ts
index 68160958..9efc88c1 100644
--- a/src/worker/WorkerSet.ts
+++ b/src/worker/WorkerSet.ts
@@ -12,6 +12,7 @@ import {
   WorkerMessageEvents,
   type WorkerOptions,
   type WorkerSetElement,
+  WorkerSetEvents,
 } from './WorkerTypes';
 import { defaultErrorHandler, defaultExitHandler, sleep } from './WorkerUtils';
 
@@ -95,7 +96,7 @@ export class WorkerSet extends WorkerAbstract<WorkerData> {
     );
     worker.on('error', defaultErrorHandler.bind(this) as (err: Error) => void);
     worker.on('error', (error) => {
-      this.emitter.emit('error', error);
+      this.emitter.emit(WorkerSetEvents.error, error);
       this.addWorkerSetElement();
     });
     worker.on('exit', defaultExitHandler.bind(this) as (exitCode: number) => void);
diff --git a/src/worker/WorkerTypes.ts b/src/worker/WorkerTypes.ts
index 6f780bce..fa1c82b0 100644
--- a/src/worker/WorkerTypes.ts
+++ b/src/worker/WorkerTypes.ts
@@ -1,6 +1,6 @@
 import type { Worker } from 'node:worker_threads';
 
-import type { PoolOptions } from 'poolifier';
+import { type PoolEvent, PoolEvents, type PoolOptions } from 'poolifier';
 
 export enum WorkerProcessType {
   workerSet = 'workerSet',
@@ -14,6 +14,16 @@ export type SetInfo = {
   elementsPerWorker: number;
 };
 
+export enum WorkerSetEvents {
+  error = 'error',
+}
+
+export const WorkerEvents = {
+  ...PoolEvents,
+  ...WorkerSetEvents,
+} as const;
+export type WorkerEvents = PoolEvent | WorkerSetEvents;
+
 export type MessageHandler<T> = (this: T, message: unknown) => void;
 
 export type WorkerOptions = {
diff --git a/src/worker/index.ts b/src/worker/index.ts
index 2ad6671b..5c8a2534 100644
--- a/src/worker/index.ts
+++ b/src/worker/index.ts
@@ -4,6 +4,7 @@ export { WorkerFactory } from './WorkerFactory';
 export {
   type MessageHandler,
   type WorkerData,
+  type WorkerEvents,
   type WorkerMessage,
   WorkerMessageEvents,
   WorkerProcessType,
-- 
2.34.1