De-duplicate code for workers (#154)

[poolifier.git] / src / worker / thread-worker.ts

diff --git a/src/worker/thread-worker.ts b/src/worker/thread-worker.ts
index b6fbdae257399ac7fa9f8d9fca9e37f6cdfc08d7..615e1d8b7f0b1dc5f92cc2d6417813ef60d83240 100644 (file)
--- a/src/worker/thread-worker.ts
+++ b/src/worker/thread-worker.ts
@@ -1,3 +1,4 @@
+import type { MessagePort } from 'worker_threads'
 import { isMainThread, parentPort } from 'worker_threads'
 import type { JSONValue, MessageValue } from '../utility-types'
 import { AbstractWorker } from './abstract-worker'
@@ -22,11 +23,6 @@ export class ThreadWorker<
   Data extends JSONValue = JSONValue,
   Response extends JSONValue = JSONValue
 > extends AbstractWorker<MessagePort, Data, Response> {
-  /**
-   * Reference to main thread.
-   */
-  protected parent?: MessagePort
-
   /**
    * Constructs a new poolifier thread worker.
    *
@@ -34,34 +30,7 @@ export class ThreadWorker<
    * @param opts Options for the worker.
    */
   public constructor (fn: (data: Data) => Response, opts: WorkerOptions = {}) {
-    super('worker-thread-pool:pioardi', isMainThread, fn, opts)
-
-    parentPort?.on('message', (value: MessageValue<Data>) => {
-      if (value?.data && value.id) {
-        // here you will receive messages
-        // console.log('This is the main worker ' + isMainThread)
-        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
-        this.parent = value.parent
-      } else if (value.kill) {
-        // here is time to kill this worker, just clearing the interval
-        if (this.interval) clearInterval(this.interval)
-        this.emitDestroy()
-      }
-    })
-  }
-
-  protected getMainWorker (): MessagePort {
-    if (!this.parent) {
-      throw new Error('Parent was not set')
-    }
-    return this.parent
+    super('worker-thread-pool:pioardi', isMainThread, fn, parentPort, opts)
   }
 
   protected sendToMainWorker (message: MessageValue<Response>): void {