fix: use internal workerId for child_process runtime (tinylibs#81)

AriPerkkio · Jan 17, 2024 · 0eb5e0c · 0eb5e0c
1 parent ac2e839
commit 0eb5e0c
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/src/common.ts b/src/common.ts
@@ -16,7 +16,7 @@ export interface TinypoolWorker {
     argv?: string[]
     execArgv?: string[]
     resourceLimits?: any
-    workerData?: TinypoolData
+    workerData: TinypoolData
     trackUnmanagedFds?: boolean
   }): void
   terminate(): Promise<any>

diff --git a/src/entry/process.ts b/src/entry/process.ts
@@ -20,7 +20,7 @@ process.__tinypool_state__ = {
   isChildProcess: true,
   isTinypoolWorker: true,
   workerData: null,
-  workerId: process.pid,
+  workerId: Number(process.env.TINYPOOL_WORKER_ID),
 }
 
 process.on('message', (message: IncomingMessage) => {

diff --git a/src/runtime/process-worker.ts b/src/runtime/process-worker.ts
@@ -23,7 +23,13 @@ export default class ProcessWorker implements TinypoolWorker {
     this.process = fork(
       fileURLToPath(import.meta.url + '/../entry/process.js'),
       options.argv,
-      options
+      {
+        ...options,
+        env: {
+          ...options.env,
+          TINYPOOL_WORKER_ID: options.workerData[0].workerId.toString(),
+        },
+      }
     )
     this.threadId = this.process.pid!
 

diff --git a/test/runtime.test.ts b/test/runtime.test.ts
@@ -85,6 +85,12 @@ describe('child_process', () => {
     expect(result).toBe(threadId)
   })
 
+  test('child process workerId should be internal tinypool workerId', async () => {
+    const pool = createPool({ runtime: 'child_process' })
+    const workerId = await pool.run('process.__tinypool_state__.workerId')
+    expect(workerId).toBe(1)
+  })
+
   test('errors are serialized', async () => {
     const pool = createPool({ runtime: 'child_process' })