fix(worker): make numIdleProcesses work (#135)

livekit · Nov 4, 2024 · d9273f2 · d9273f2
1 parent 70773f9
commit d9273f2
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 11 deletions.
diff --git a/.changeset/afraid-camels-push.md b/.changeset/afraid-camels-push.md
@@ -0,0 +1,5 @@
+---
+"@livekit/agents": patch
+---
+
+make numIdleProcesses work
diff --git a/agents/package.json b/agents/package.json
@@ -21,7 +21,7 @@
     "typescript": "^5.0.0"
   },
   "dependencies": {
-    "@livekit/mutex": "^1.0.0",
+    "@livekit/mutex": "^1.1.0",
     "@livekit/protocol": "^1.21.0",
     "@livekit/rtc-node": "^0.11.0",
     "commander": "^12.0.0",

diff --git a/agents/src/ipc/proc_pool.ts b/agents/src/ipc/proc_pool.ts
@@ -1,7 +1,7 @@
 // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
 //
 // SPDX-License-Identifier: Apache-2.0
-import { Mutex } from '@livekit/mutex';
+import { MultiMutex, Mutex } from '@livekit/mutex';
 import type { RunningJobInfo } from '../job.js';
 import { Queue } from '../utils.js';
 import type { JobExecutor } from './job_executor.js';
@@ -17,18 +17,18 @@ export class ProcPool {
   closed = false;
   controller = new AbortController();
   initMutex = new Mutex();
-  procMutex: Mutex;
+  procMutex: MultiMutex;
   procUnlock?: () => void;
   warmedProcQueue = new Queue<JobExecutor>();
 
   constructor(
     agent: string,
-    _numIdleProcesses: number,
+    numIdleProcesses: number,
     initializeTimeout: number,
     closeTimeout: number,
   ) {
     this.agent = agent;
-    this.procMutex = new Mutex();
+    this.procMutex = new MultiMutex(numIdleProcesses);
     this.initializeTimeout = initializeTimeout;
     this.closeTimeout = closeTimeout;
   }

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml