@vielzeug/familiar 26.10.15 → 26.10.17

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2025 Helmuth Saatkamp
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -1 +1 @@
1
- {"version":3,"file":"_pool-core.cjs","names":[],"sources":["../src/_pool-core.ts"],"sourcesContent":["import { abortError, tapper } from '@vielzeug/arsenal';\nimport { unrefTimer } from './_timers.js';\nimport {\n FamiliarInvalidOptionsError,\n FamiliarQueueFullError,\n type FamiliarRuntimeError,\n FamiliarTerminatedError,\n FamiliarTimeoutError,\n} from './errors.js';\nimport type { DrainOptions, FamiliarTapEvent, WorkerStats, WorkerStatus } from './types.js';\n\nexport const MAX_CONCURRENCY = 512;\nexport const MAX_TIMEOUT = 2_147_483_647;\n\nexport function validatePriority(priority: number): number {\n if (!Number.isFinite(priority)) throw new FamiliarInvalidOptionsError('`priority` must be a finite number');\n return priority;\n}\n\nexport function validateTimeout(timeout: number | undefined, name = 'timeout'): number | undefined {\n if (timeout !== undefined && (!Number.isInteger(timeout) || timeout < 1 || timeout > MAX_TIMEOUT)) {\n throw new FamiliarInvalidOptionsError(`\\`${name}\\` must be an integer from 1 through ${MAX_TIMEOUT}`);\n }\n return timeout;\n}\n\nexport type PoolOptions = {\n defaultTimeout: number | undefined;\n maxQueue: number | undefined;\n onFull: 'reject' | 'wait';\n};\n\nexport type AcquireOptions = {\n priority: number;\n signal?: AbortSignal;\n};\n\n/**\n * A pool slot: tearable on disposal, with an optional channel for reporting\n * unhandled runtime errors. Pool factories wire `notify` to the core's tap so\n * slot owners can emit without holding a pool reference.\n */\nexport type PoolSlot = {\n notify?(error: FamiliarRuntimeError): void;\n terminate(): void;\n};\n\ntype IdleWaiter = {\n reject: (reason: unknown) => void;\n resolve: () => void;\n timer?: ReturnType<typeof setTimeout>;\n};\n\ntype CapacityWaiter = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n resolve(): void;\n sequence: number;\n};\n\ntype SlotWaiter<TSlot> = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n start(slot: TSlot): void;\n sequence: number;\n signal?: AbortSignal;\n};\n\nexport interface PoolCore<TSlot> {\n /**\n * Acquire a free slot and hand it to `fn`, which runs the acquired work.\n * `fn` is invoked synchronously both for an immediately free slot and when\n * a release hands the next waiter its slot, so dispatch never lags a\n * release by a microtask. Honors priority, FIFO order, abort, and `maxQueue`\n * admission; failures surface as a rejected promise. Without `fn`, resolves\n * with the slot instead.\n */\n acquire(options: AcquireOptions): Promise<TSlot>;\n acquire<R>(options: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R>;\n readonly disposalSignal: AbortSignal;\n dispose(): void;\n readonly disposed: boolean;\n drain(options?: DrainOptions): Promise<void>;\n readonly drainPromise: Promise<void> | undefined;\n emitTap(event: FamiliarTapEvent): void;\n /** Return a slot to the pool, handing it to the highest-priority waiter if any. */\n release(slot: TSlot): void;\n settleIdle(): void;\n stats(): WorkerStats;\n status(): WorkerStatus;\n tap(handler: (event: FamiliarTapEvent) => void, options?: { readonly signal?: AbortSignal }): () => void;\n trackActive(delta: number): void;\n trackCompleted(): void;\n trackFailed(): void;\n}\n\n/**\n * The single slot scheduler shared by the result pool and the stream pool:\n * free-slot fast path, priority/FIFO waiter queue, `maxQueue` admission with\n * capacity waiters, idle/drain tracking, stats, disposal, and tap observation.\n * The two pools differ only in what they do with an acquired slot.\n */\nexport function createPoolCore<TSlot extends PoolSlot>(slots: readonly TSlot[], options: PoolOptions): PoolCore<TSlot> {\n const freeSlots = [...slots];\n const waiters: SlotWaiter<TSlot>[] = [];\n const idleWaiters: IdleWaiter[] = [];\n const capacityWaiters: CapacityWaiter[] = [];\n const tappers = tapper<FamiliarTapEvent>();\n const disposalController = new AbortController();\n let active = 0;\n let waiterSequence = 0;\n let capacitySequence = 0;\n let completed = 0;\n let failed = 0;\n let drainPromise: Promise<void> | undefined;\n let terminated = false;\n\n function isIdle(): boolean {\n return freeSlots.length === slots.length && waiters.length === 0;\n }\n\n function settleIdle(): void {\n if (!isIdle()) return;\n\n for (const waiter of idleWaiters.splice(0)) {\n if (waiter.timer) clearTimeout(waiter.timer);\n\n waiter.resolve();\n }\n }\n\n function waitForIdle(drainOptions: DrainOptions): Promise<void> {\n if (isIdle()) return Promise.resolve();\n\n return new Promise<void>((resolve, reject) => {\n const waiter: IdleWaiter = { reject, resolve };\n\n if (drainOptions.timeout !== undefined) {\n waiter.timer = setTimeout(() => {\n const index = idleWaiters.indexOf(waiter);\n\n if (index !== -1) idleWaiters.splice(index, 1);\n\n reject(new FamiliarTimeoutError(drainOptions.timeout!, 'Drain'));\n }, drainOptions.timeout);\n unrefTimer(waiter.timer);\n }\n\n idleWaiters.push(waiter);\n });\n }\n\n function releaseCapacity(): void {\n capacityWaiters.shift()?.resolve();\n }\n\n function rejectCapacity(reason: unknown): void {\n for (const waiter of capacityWaiters.splice(0)) waiter.reject(reason);\n }\n\n function waitForCapacity(signal: AbortSignal | undefined, priority: number): Promise<void> {\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return new Promise<void>((resolve, reject) => {\n let settled = false;\n const waiter: CapacityWaiter = {\n cleanup() {\n signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n\n const index = capacityWaiters.indexOf(waiter);\n\n if (index !== -1) capacityWaiters.splice(index, 1);\n\n reject(reason);\n },\n resolve() {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n resolve();\n },\n sequence: capacitySequence++,\n };\n const onAbort = () => waiter.reject(abortError(signal!));\n\n signal?.addEventListener('abort', onAbort, { once: true });\n capacityWaiters.push(waiter);\n capacityWaiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function acquire<R>(acquireOptions: AcquireOptions, fn?: (slot: TSlot) => R | Promise<R>): Promise<R | TSlot> {\n const { priority, signal } = acquireOptions;\n\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return (async () => {\n for (;;) {\n if (terminated) throw new FamiliarTerminatedError();\n\n if (drainPromise) throw new FamiliarTerminatedError('Worker is draining');\n\n if (options.maxQueue === undefined || waiters.length < options.maxQueue) {\n return admit(acquireOptions, (fn ?? ((slot) => slot)) as (slot: TSlot) => R | Promise<R>);\n }\n\n if (options.onFull === 'reject') throw new FamiliarQueueFullError(options.maxQueue);\n\n await waitForCapacity(signal, priority);\n\n if (signal?.aborted) {\n releaseCapacity();\n throw abortError(signal);\n }\n }\n })();\n }\n\n function admit<R>(acquireOptions: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R> {\n const { priority, signal } = acquireOptions;\n\n const free = freeSlots.pop();\n\n if (free) {\n try {\n return Promise.resolve(fn(free));\n } catch (error) {\n // A synchronous throw must not strand the slot: idle would never settle.\n release(free);\n settleIdle();\n\n return Promise.reject(error);\n }\n }\n\n return new Promise<R>((resolve, reject) => {\n const waiter: SlotWaiter<TSlot> = {\n cleanup() {\n waiter.signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n waiter.cleanup();\n\n const index = waiters.indexOf(waiter);\n\n if (index !== -1) {\n waiters.splice(index, 1);\n releaseCapacity();\n }\n\n reject(reason);\n settleIdle();\n },\n sequence: waiterSequence++,\n signal,\n start(slot) {\n waiter.cleanup();\n\n try {\n resolve(fn(slot));\n } catch (error) {\n release(slot);\n settleIdle();\n reject(error);\n }\n },\n };\n const onAbort = () => waiter.reject(abortError(waiter.signal!));\n\n waiter.signal?.addEventListener('abort', onAbort, { once: true });\n waiters.push(waiter);\n waiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function release(slot: TSlot): void {\n const next = waiters.shift();\n\n if (next) {\n next.start(slot);\n releaseCapacity();\n } else {\n freeSlots.push(slot);\n }\n }\n\n function dispose(): void {\n if (terminated) return;\n\n terminated = true;\n disposalController.abort();\n\n for (const slot of slots) slot.terminate();\n\n for (const waiter of waiters.splice(0)) waiter.reject(new FamiliarTerminatedError());\n\n rejectCapacity(new FamiliarTerminatedError());\n settleIdle();\n tappers.emit({ type: 'dispose' });\n tappers.clear();\n }\n\n function drain(drainOptions: DrainOptions = {}): Promise<void> {\n if (terminated) return Promise.resolve();\n try {\n validateTimeout(drainOptions.timeout, 'drain.timeout');\n } catch (error) {\n return Promise.reject(error);\n }\n\n if (drainPromise) return drainPromise;\n\n rejectCapacity(new FamiliarTerminatedError('Worker is draining'));\n drainPromise = waitForIdle(drainOptions).then(\n () => dispose(),\n (error: unknown) => {\n dispose();\n throw error;\n },\n );\n\n return drainPromise;\n }\n\n return {\n acquire,\n disposalSignal: disposalController.signal,\n dispose,\n get disposed() {\n return terminated;\n },\n drain,\n get drainPromise() {\n return drainPromise;\n },\n emitTap(event) {\n tappers.emit(event);\n },\n release,\n settleIdle,\n stats() {\n return { active, completed, failed, queued: waiters.length + capacityWaiters.length };\n },\n status() {\n if (terminated) return 'terminated';\n\n return active === 0 ? 'idle' : 'running';\n },\n tap(handler, tapOptions) {\n if (terminated) return () => undefined;\n\n return tappers.tap(handler, tapOptions);\n },\n trackActive(delta: number) {\n active += delta;\n },\n trackCompleted() {\n completed += 1;\n },\n trackFailed() {\n failed += 1;\n },\n };\n}\n"],"mappings":"8FAYA,IAAa,EAAc,WAE3B,SAAgB,iBAAiB,EAA0B,CACzD,GAAI,CAAC,OAAO,SAAS,CAAQ,EAAG,MAAM,IAAI,EAAA,4BAA4B,oCAAoC,EAC1G,OAAO,CACT,CAEA,SAAgB,gBAAgB,EAA6B,EAAO,UAA+B,CACjG,GAAI,IAAY,IAAA,KAAc,CAAC,OAAO,UAAU,CAAO,GAAK,EAAU,GAAK,EAAA,YACzE,MAAM,IAAI,EAAA,4BAA4B,KAAK,EAAK,uCAAuC,GAAa,EAEtG,OAAO,CACT,CAgFA,SAAgB,eAAuC,EAAyB,EAAuC,CACrH,IAAM,EAAY,CAAC,GAAG,CAAK,EACrB,EAA+B,CAAC,EAChC,EAA4B,CAAC,EAC7B,EAAoC,CAAC,EACrC,GAAA,EAAU,EAAA,OAAA,CAAyB,EACnC,EAAqB,IAAI,gBAC3B,EAAS,EACT,EAAiB,EACjB,EAAmB,EACnB,EAAY,EACZ,EAAS,EACT,EACA,EAAa,GAEjB,SAAS,QAAkB,CACzB,OAAO,EAAU,SAAW,EAAM,QAAU,EAAQ,SAAW,CACjE,CAEA,SAAS,YAAmB,CACrB,UAAO,EAEZ,IAAK,IAAM,KAAU,EAAY,OAAO,CAAC,EACnC,EAAO,OAAO,aAAa,EAAO,KAAK,EAE3C,EAAO,QAAQ,CAEnB,CAEA,SAAS,YAAY,EAA2C,CAG9D,OAFI,OAAO,EAAU,QAAQ,QAAQ,EAE9B,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAM,EAAqB,CAAE,SAAQ,SAAQ,EAEzC,EAAa,UAAY,IAAA,KAC3B,EAAO,MAAQ,eAAiB,CAC9B,IAAM,EAAQ,EAAY,QAAQ,CAAM,EAEpC,IAAU,IAAI,EAAY,OAAO,EAAO,CAAC,EAE7C,EAAO,IAAI,EAAA,qBAAqB,EAAa,QAAU,OAAO,CAAC,CACjE,EAAG,EAAa,OAAO,EACvB,EAAA,WAAW,EAAO,KAAK,GAGzB,EAAY,KAAK,CAAM,CACzB,CAAC,CACH,CAEA,SAAS,iBAAwB,CAC/B,EAAgB,MAAM,CAAC,EAAE,QAAQ,CACnC,CAEA,SAAS,eAAe,EAAuB,CAC7C,IAAK,IAAM,KAAU,EAAgB,OAAO,CAAC,EAAG,EAAO,OAAO,CAAM,CACtE,CAEA,SAAS,gBAAgB,EAAiC,EAAiC,CAGzF,OAFI,GAAQ,QAAgB,QAAQ,QAAA,EAAO,EAAA,WAAA,CAAW,CAAM,CAAC,EAEtD,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAI,EAAU,GACR,EAAyB,CAC7B,SAAU,CACR,GAAQ,oBAAoB,QAAS,OAAO,CAC9C,EACA,WACA,OAAO,EAAQ,CACb,GAAI,EAAS,OAEb,EAAU,GACV,EAAO,QAAQ,EAEf,IAAM,EAAQ,EAAgB,QAAQ,CAAM,EAExC,IAAU,IAAI,EAAgB,OAAO,EAAO,CAAC,EAEjD,EAAO,CAAM,CACf,EACA,SAAU,CACJ,IAEJ,EAAU,GACV,EAAO,QAAQ,EACf,EAAQ,EACV,EACA,SAAU,GACZ,EACM,YAAgB,EAAO,QAAA,EAAO,EAAA,WAAA,CAAW,CAAO,CAAC,EAEvD,GAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EACzD,EAAgB,KAAK,CAAM,EAC3B,EAAgB,MAAM,EAAG,IAAM,EAAE,SAAW,EAAE,UAAY,EAAE,SAAW,EAAE,QAAQ,CACnF,CAAC,CACH,CAEA,SAAS,QAAW,EAAgC,EAA0D,CAC5G,GAAM,CAAE,WAAU,UAAW,EAI7B,OAFI,GAAQ,QAAgB,QAAQ,QAAA,EAAO,EAAA,WAAA,CAAW,CAAM,CAAC,GAErD,SAAY,CAClB,OAAS,CACP,GAAI,EAAY,MAAM,IAAI,EAAA,wBAE1B,GAAI,EAAc,MAAM,IAAI,EAAA,wBAAwB,oBAAoB,EAExE,GAAI,EAAQ,WAAa,IAAA,IAAa,EAAQ,OAAS,EAAQ,SAC7D,OAAO,MAAM,EAAiB,IAAQ,GAAS,EAAyC,EAG1F,GAAI,EAAQ,SAAW,SAAU,MAAM,IAAI,EAAA,uBAAuB,EAAQ,QAAQ,EAIlF,GAFA,MAAM,gBAAgB,EAAQ,CAAQ,EAElC,GAAQ,QAEV,MADA,gBAAgB,GAChB,EAAM,EAAA,WAAA,CAAW,CAAM,CAE3B,CACF,EAAA,CAAG,CACL,CAEA,SAAS,MAAS,EAAgC,EAAiD,CACjG,GAAM,CAAE,WAAU,UAAW,EAEvB,EAAO,EAAU,IAAI,EAE3B,GAAI,EACF,GAAI,CACF,OAAO,QAAQ,QAAQ,EAAG,CAAI,CAAC,CACjC,OAAS,EAAO,CAKd,OAHA,QAAQ,CAAI,EACZ,WAAW,EAEJ,QAAQ,OAAO,CAAK,CAC7B,CAGF,OAAO,IAAI,SAAY,EAAS,IAAW,CACzC,IAAM,EAA4B,CAChC,SAAU,CACR,EAAO,QAAQ,oBAAoB,QAAS,OAAO,CACrD,EACA,WACA,OAAO,EAAQ,CACb,EAAO,QAAQ,EAEf,IAAM,EAAQ,EAAQ,QAAQ,CAAM,EAEhC,IAAU,KACZ,EAAQ,OAAO,EAAO,CAAC,EACvB,gBAAgB,GAGlB,EAAO,CAAM,EACb,WAAW,CACb,EACA,SAAU,IACV,SACA,MAAM,EAAM,CACV,EAAO,QAAQ,EAEf,GAAI,CACF,EAAQ,EAAG,CAAI,CAAC,CAClB,OAAS,EAAO,CACd,QAAQ,CAAI,EACZ,WAAW,EACX,EAAO,CAAK,CACd,CACF,CACF,EACM,YAAgB,EAAO,QAAA,EAAO,EAAA,WAAA,CAAW,EAAO,MAAO,CAAC,EAE9D,EAAO,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAChE,EAAQ,KAAK,CAAM,EACnB,EAAQ,MAAM,EAAG,IAAM,EAAE,SAAW,EAAE,UAAY,EAAE,SAAW,EAAE,QAAQ,CAC3E,CAAC,CACH,CAEA,SAAS,QAAQ,EAAmB,CAClC,IAAM,EAAO,EAAQ,MAAM,EAEvB,GACF,EAAK,MAAM,CAAI,EACf,gBAAgB,GAEhB,EAAU,KAAK,CAAI,CAEvB,CAEA,SAAS,SAAgB,CACnB,MAGJ,CADA,EAAa,GACb,EAAmB,MAAM,EAEzB,IAAK,IAAM,KAAQ,EAAO,EAAK,UAAU,EAEzC,IAAK,IAAM,KAAU,EAAQ,OAAO,CAAC,EAAG,EAAO,OAAO,IAAI,EAAA,uBAAyB,EAEnF,eAAe,IAAI,EAAA,uBAAyB,EAC5C,WAAW,EACX,EAAQ,KAAK,CAAE,KAAM,SAAU,CAAC,EAChC,EAAQ,MAAM,CATW,CAU3B,CAEA,SAAS,MAAM,EAA6B,CAAC,EAAkB,CAC7D,GAAI,EAAY,OAAO,QAAQ,QAAQ,EACvC,GAAI,CACF,gBAAgB,EAAa,QAAS,eAAe,CACvD,OAAS,EAAO,CACd,OAAO,QAAQ,OAAO,CAAK,CAC7B,CAaA,OAXI,IAEJ,eAAe,IAAI,EAAA,wBAAwB,oBAAoB,CAAC,EAChE,EAAe,YAAY,CAAY,CAAC,CAAC,SACjC,QAAQ,EACb,GAAmB,CAElB,MADA,QAAQ,EACF,CACR,CACF,EAEO,EACT,CAEA,MAAO,CACL,QACA,eAAgB,EAAmB,OACnC,QACA,IAAI,UAAW,CACb,OAAO,CACT,EACA,MACA,IAAI,cAAe,CACjB,OAAO,CACT,EACA,QAAQ,EAAO,CACb,EAAQ,KAAK,CAAK,CACpB,EACA,QACA,WACA,OAAQ,CACN,MAAO,CAAE,SAAQ,YAAW,SAAQ,OAAQ,EAAQ,OAAS,EAAgB,MAAO,CACtF,EACA,QAAS,CAGP,OAFI,EAAmB,aAEhB,IAAW,EAAI,OAAS,SACjC,EACA,IAAI,EAAS,EAAY,CAGvB,OAFI,MAAyB,IAAA,GAEtB,EAAQ,IAAI,EAAS,CAAU,CACxC,EACA,YAAY,EAAe,CACzB,GAAU,CACZ,EACA,gBAAiB,CACf,GAAa,CACf,EACA,aAAc,CACZ,GAAU,CACZ,CACF,CACF"}
1
+ {"version":3,"file":"_pool-core.cjs","names":["FamiliarInvalidOptionsError","tapper","FamiliarTimeoutError","abortError","FamiliarTerminatedError","FamiliarQueueFullError","require__timers","require_errors","_vielzeug_arsenal","MAX_TIMEOUT","priority","timeout","name","freeSlots","slots","waiters","idleWaiters","capacityWaiters","tappers","disposalController","active","waiterSequence","capacitySequence","completed","failed","drainPromise","terminated","waiter","drainOptions","index","reject","reason","signal","settled","resolve","b","a","acquireOptions","options","fn","slot","free","error","next","event","handler","tapOptions","delta"],"sources":["../src/_pool-core.ts"],"sourcesContent":["import { abortError, tapper } from '@vielzeug/arsenal';\nimport { unrefTimer } from './_timers.js';\nimport {\n FamiliarInvalidOptionsError,\n FamiliarQueueFullError,\n type FamiliarRuntimeError,\n FamiliarTerminatedError,\n FamiliarTimeoutError,\n} from './errors.js';\nimport type { DrainOptions, FamiliarTapEvent, WorkerStats, WorkerStatus } from './types.js';\n\nexport const MAX_CONCURRENCY = 512;\nexport const MAX_TIMEOUT = 2_147_483_647;\n\nexport function validatePriority(priority: number): number {\n if (!Number.isFinite(priority)) throw new FamiliarInvalidOptionsError('`priority` must be a finite number');\n return priority;\n}\n\nexport function validateTimeout(timeout: number | undefined, name = 'timeout'): number | undefined {\n if (timeout !== undefined && (!Number.isInteger(timeout) || timeout < 1 || timeout > MAX_TIMEOUT)) {\n throw new FamiliarInvalidOptionsError(`\\`${name}\\` must be an integer from 1 through ${MAX_TIMEOUT}`);\n }\n return timeout;\n}\n\nexport type PoolOptions = {\n defaultTimeout: number | undefined;\n maxQueue: number | undefined;\n onFull: 'reject' | 'wait';\n};\n\nexport type AcquireOptions = {\n priority: number;\n signal?: AbortSignal;\n};\n\n/**\n * A pool slot: tearable on disposal, with an optional channel for reporting\n * unhandled runtime errors. Pool factories wire `notify` to the core's tap so\n * slot owners can emit without holding a pool reference.\n */\nexport type PoolSlot = {\n notify?(error: FamiliarRuntimeError): void;\n terminate(): void;\n};\n\ntype IdleWaiter = {\n reject: (reason: unknown) => void;\n resolve: () => void;\n timer?: ReturnType<typeof setTimeout>;\n};\n\ntype CapacityWaiter = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n resolve(): void;\n sequence: number;\n};\n\ntype SlotWaiter<TSlot> = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n start(slot: TSlot): void;\n sequence: number;\n signal?: AbortSignal;\n};\n\nexport interface PoolCore<TSlot> {\n /**\n * Acquire a free slot and hand it to `fn`, which runs the acquired work.\n * `fn` is invoked synchronously both for an immediately free slot and when\n * a release hands the next waiter its slot, so dispatch never lags a\n * release by a microtask. Honors priority, FIFO order, abort, and `maxQueue`\n * admission; failures surface as a rejected promise. Without `fn`, resolves\n * with the slot instead.\n */\n acquire(options: AcquireOptions): Promise<TSlot>;\n acquire<R>(options: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R>;\n readonly disposalSignal: AbortSignal;\n dispose(): void;\n readonly disposed: boolean;\n drain(options?: DrainOptions): Promise<void>;\n readonly drainPromise: Promise<void> | undefined;\n emitTap(event: FamiliarTapEvent): void;\n /** Return a slot to the pool, handing it to the highest-priority waiter if any. */\n release(slot: TSlot): void;\n settleIdle(): void;\n stats(): WorkerStats;\n status(): WorkerStatus;\n tap(handler: (event: FamiliarTapEvent) => void, options?: { readonly signal?: AbortSignal }): () => void;\n trackActive(delta: number): void;\n trackCompleted(): void;\n trackFailed(): void;\n}\n\n/**\n * The single slot scheduler shared by the result pool and the stream pool:\n * free-slot fast path, priority/FIFO waiter queue, `maxQueue` admission with\n * capacity waiters, idle/drain tracking, stats, disposal, and tap observation.\n * The two pools differ only in what they do with an acquired slot.\n */\nexport function createPoolCore<TSlot extends PoolSlot>(slots: readonly TSlot[], options: PoolOptions): PoolCore<TSlot> {\n const freeSlots = [...slots];\n const waiters: SlotWaiter<TSlot>[] = [];\n const idleWaiters: IdleWaiter[] = [];\n const capacityWaiters: CapacityWaiter[] = [];\n const tappers = tapper<FamiliarTapEvent>();\n const disposalController = new AbortController();\n let active = 0;\n let waiterSequence = 0;\n let capacitySequence = 0;\n let completed = 0;\n let failed = 0;\n let drainPromise: Promise<void> | undefined;\n let terminated = false;\n\n function isIdle(): boolean {\n return freeSlots.length === slots.length && waiters.length === 0;\n }\n\n function settleIdle(): void {\n if (!isIdle()) return;\n\n for (const waiter of idleWaiters.splice(0)) {\n if (waiter.timer) clearTimeout(waiter.timer);\n\n waiter.resolve();\n }\n }\n\n function waitForIdle(drainOptions: DrainOptions): Promise<void> {\n if (isIdle()) return Promise.resolve();\n\n return new Promise<void>((resolve, reject) => {\n const waiter: IdleWaiter = { reject, resolve };\n\n if (drainOptions.timeout !== undefined) {\n waiter.timer = setTimeout(() => {\n const index = idleWaiters.indexOf(waiter);\n\n if (index !== -1) idleWaiters.splice(index, 1);\n\n reject(new FamiliarTimeoutError(drainOptions.timeout!, 'Drain'));\n }, drainOptions.timeout);\n unrefTimer(waiter.timer);\n }\n\n idleWaiters.push(waiter);\n });\n }\n\n function releaseCapacity(): void {\n capacityWaiters.shift()?.resolve();\n }\n\n function rejectCapacity(reason: unknown): void {\n for (const waiter of capacityWaiters.splice(0)) waiter.reject(reason);\n }\n\n function waitForCapacity(signal: AbortSignal | undefined, priority: number): Promise<void> {\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return new Promise<void>((resolve, reject) => {\n let settled = false;\n const waiter: CapacityWaiter = {\n cleanup() {\n signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n\n const index = capacityWaiters.indexOf(waiter);\n\n if (index !== -1) capacityWaiters.splice(index, 1);\n\n reject(reason);\n },\n resolve() {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n resolve();\n },\n sequence: capacitySequence++,\n };\n const onAbort = () => waiter.reject(abortError(signal!));\n\n signal?.addEventListener('abort', onAbort, { once: true });\n capacityWaiters.push(waiter);\n capacityWaiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function acquire<R>(acquireOptions: AcquireOptions, fn?: (slot: TSlot) => R | Promise<R>): Promise<R | TSlot> {\n const { priority, signal } = acquireOptions;\n\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return (async () => {\n for (;;) {\n if (terminated) throw new FamiliarTerminatedError();\n\n if (drainPromise) throw new FamiliarTerminatedError('Worker is draining');\n\n if (options.maxQueue === undefined || waiters.length < options.maxQueue) {\n return admit(acquireOptions, (fn ?? ((slot) => slot)) as (slot: TSlot) => R | Promise<R>);\n }\n\n if (options.onFull === 'reject') throw new FamiliarQueueFullError(options.maxQueue);\n\n await waitForCapacity(signal, priority);\n\n if (signal?.aborted) {\n releaseCapacity();\n throw abortError(signal);\n }\n }\n })();\n }\n\n function admit<R>(acquireOptions: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R> {\n const { priority, signal } = acquireOptions;\n\n const free = freeSlots.pop();\n\n if (free) {\n try {\n return Promise.resolve(fn(free));\n } catch (error) {\n // A synchronous throw must not strand the slot: idle would never settle.\n release(free);\n settleIdle();\n\n return Promise.reject(error);\n }\n }\n\n return new Promise<R>((resolve, reject) => {\n const waiter: SlotWaiter<TSlot> = {\n cleanup() {\n waiter.signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n waiter.cleanup();\n\n const index = waiters.indexOf(waiter);\n\n if (index !== -1) {\n waiters.splice(index, 1);\n releaseCapacity();\n }\n\n reject(reason);\n settleIdle();\n },\n sequence: waiterSequence++,\n signal,\n start(slot) {\n waiter.cleanup();\n\n try {\n resolve(fn(slot));\n } catch (error) {\n release(slot);\n settleIdle();\n reject(error);\n }\n },\n };\n const onAbort = () => waiter.reject(abortError(waiter.signal!));\n\n waiter.signal?.addEventListener('abort', onAbort, { once: true });\n waiters.push(waiter);\n waiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function release(slot: TSlot): void {\n const next = waiters.shift();\n\n if (next) {\n next.start(slot);\n releaseCapacity();\n } else {\n freeSlots.push(slot);\n }\n }\n\n function dispose(): void {\n if (terminated) return;\n\n terminated = true;\n disposalController.abort();\n\n for (const slot of slots) slot.terminate();\n\n for (const waiter of waiters.splice(0)) waiter.reject(new FamiliarTerminatedError());\n\n rejectCapacity(new FamiliarTerminatedError());\n settleIdle();\n tappers.emit({ type: 'dispose' });\n tappers.clear();\n }\n\n function drain(drainOptions: DrainOptions = {}): Promise<void> {\n if (terminated) return Promise.resolve();\n try {\n validateTimeout(drainOptions.timeout, 'drain.timeout');\n } catch (error) {\n return Promise.reject(error);\n }\n\n if (drainPromise) return drainPromise;\n\n rejectCapacity(new FamiliarTerminatedError('Worker is draining'));\n drainPromise = waitForIdle(drainOptions).then(\n () => dispose(),\n (error: unknown) => {\n dispose();\n throw error;\n },\n );\n\n return drainPromise;\n }\n\n return {\n acquire,\n disposalSignal: disposalController.signal,\n dispose,\n get disposed() {\n return terminated;\n },\n drain,\n get drainPromise() {\n return drainPromise;\n },\n emitTap(event) {\n tappers.emit(event);\n },\n release,\n settleIdle,\n stats() {\n return { active, completed, failed, queued: waiters.length + capacityWaiters.length };\n },\n status() {\n if (terminated) return 'terminated';\n\n return active === 0 ? 'idle' : 'running';\n },\n tap(handler, tapOptions) {\n if (terminated) return () => undefined;\n\n return tappers.tap(handler, tapOptions);\n },\n trackActive(delta: number) {\n active += delta;\n },\n trackCompleted() {\n completed += 1;\n },\n trackFailed() {\n failed += 1;\n },\n };\n}\n"],"mappings":"A,8FAYA,IAAaS,EAAc,WAE3B,SAAgB,iBAAiB,EAA0B,CACzD,GAAI,CAAC,OAAO,SAASC,CAAQ,EAAG,MAAM,IAAIV,EAAAA,4BAA4B,oCAAoC,EAC1G,OAAOU,CACT,CAEA,SAAgB,gBAAgB,EAA6B,EAAO,UAA+B,CACjG,GAAIC,IAAY,IAAA,KAAc,CAAC,OAAO,UAAUA,CAAO,GAAKA,EAAU,GAAKA,EAAA,YACzE,MAAM,IAAIX,EAAAA,4BAA4B,KAAKY,EAAK,uCAAuCH,GAAa,EAEtG,OAAOE,CACT,CAgFA,SAAgB,eAAuC,EAAyB,EAAuC,CACrH,IAAME,EAAY,CAAC,GAAGC,CAAK,EACrBC,EAA+B,CAAC,EAChCC,EAA4B,CAAC,EAC7BC,EAAoC,CAAC,EACrCC,GAAA,EAAUjB,EAAAA,OAAAA,CAAyB,EACnCkB,EAAqB,IAAI,gBAC3BC,EAAS,EACTC,EAAiB,EACjBC,EAAmB,EACnBC,EAAY,EACZC,EAAS,EACTC,EACAC,EAAa,GAEjB,SAAS,QAAkB,CACzB,OAAOb,EAAU,SAAWC,EAAM,QAAUC,EAAQ,SAAW,CACjE,CAEA,SAAS,YAAmB,CAC1B,GAAK,OAAO,EAEZ,IAAK,IAAMY,KAAUX,EAAY,OAAO,CAAC,EACnCW,EAAO,OAAO,aAAaA,EAAO,KAAK,EAE3CA,EAAO,QAAQ,CAEnB,CAEA,SAAS,YAAY,EAA2C,CAG9D,OAFI,OAAO,EAAU,QAAQ,QAAQ,EAE9B,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAMA,EAAqB,CAAE,SAAQ,SAAQ,EAEzCC,EAAa,UAAY,IAAA,KAC3B,EAAO,MAAQ,eAAiB,CAC9B,IAAMC,EAAQb,EAAY,QAAQW,CAAM,EAEpCE,IAAU,IAAIb,EAAY,OAAOa,EAAO,CAAC,EAE7CC,EAAO,IAAI5B,EAAAA,qBAAqB0B,EAAa,QAAU,OAAO,CAAC,CACjE,EAAGA,EAAa,OAAO,EACvBtB,EAAA,WAAWqB,EAAO,KAAK,GAGzBX,EAAY,KAAKW,CAAM,CACzB,CAAC,CACH,CAEA,SAAS,iBAAwB,CAC/BV,EAAgB,MAAM,CAAC,EAAE,QAAQ,CACnC,CAEA,SAAS,eAAe,EAAuB,CAC7C,IAAK,IAAMU,KAAUV,EAAgB,OAAO,CAAC,EAAGU,EAAO,OAAOI,CAAM,CACtE,CAEA,SAAS,gBAAgB,EAAiC,EAAiC,CAGzF,OAFIC,GAAQ,QAAgB,QAAQ,QAAA,EAAO7B,EAAAA,WAAAA,CAAW6B,CAAM,CAAC,EAEtD,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAIC,EAAU,GACRN,EAAyB,CAC7B,SAAU,CACRK,GAAQ,oBAAoB,QAAS,OAAO,CAC9C,EACA,WACA,OAAO,EAAQ,CACb,GAAIC,EAAS,OAEb,EAAU,GACVN,EAAO,QAAQ,EAEf,IAAME,EAAQZ,EAAgB,QAAQU,CAAM,EAExCE,IAAU,IAAIZ,EAAgB,OAAOY,EAAO,CAAC,EAEjDC,EAAOC,CAAM,CACf,EACA,SAAU,CACJE,IAEJ,EAAU,GACVN,EAAO,QAAQ,EACfO,EAAQ,EACV,EACA,SAAU,GACZ,EACM,YAAgBP,EAAO,QAAA,EAAOxB,EAAAA,WAAAA,CAAW6B,CAAO,CAAC,EAEvDA,GAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EACzDf,EAAgB,KAAKU,CAAM,EAC3BV,EAAgB,MAAM,EAAG,IAAMkB,EAAE,SAAWC,EAAE,UAAYA,EAAE,SAAWD,EAAE,QAAQ,CACnF,CAAC,CACH,CAEA,SAAS,QAAW,EAAgC,EAA0D,CAC5G,GAAM,CAAE,WAAU,UAAWE,EAI7B,OAFIL,GAAQ,QAAgB,QAAQ,QAAA,EAAO7B,EAAAA,WAAAA,CAAW6B,CAAM,CAAC,GAErD,SAAY,CAClB,OAAS,CACP,GAAIN,EAAY,MAAM,IAAItB,EAAAA,wBAE1B,GAAIqB,EAAc,MAAM,IAAIrB,EAAAA,wBAAwB,oBAAoB,EAExE,GAAIkC,EAAQ,WAAa,IAAA,IAAavB,EAAQ,OAASuB,EAAQ,SAC7D,OAAO,MAAMD,EAAiBE,IAAQ,GAASC,EAAyC,EAG1F,GAAIF,EAAQ,SAAW,SAAU,MAAM,IAAIjC,EAAAA,uBAAuBiC,EAAQ,QAAQ,EAIlF,GAFA,MAAM,gBAAgBN,EAAQtB,CAAQ,EAElCsB,GAAQ,QAEV,MADA,gBAAgB,GAChB,EAAM7B,EAAAA,WAAAA,CAAW6B,CAAM,CAE3B,CACF,EAAA,CAAG,CACL,CAEA,SAAS,MAAS,EAAgC,EAAiD,CACjG,GAAM,CAAE,WAAU,UAAWK,EAEvBI,EAAO5B,EAAU,IAAI,EAE3B,GAAI4B,EACF,GAAI,CACF,OAAO,QAAQ,QAAQF,EAAGE,CAAI,CAAC,CACjC,OAASC,EAAO,CAKd,OAHA,QAAQD,CAAI,EACZ,WAAW,EAEJ,QAAQ,OAAOC,CAAK,CAC7B,CAGF,OAAO,IAAI,SAAY,EAAS,IAAW,CACzC,IAAMf,EAA4B,CAChC,SAAU,CACRA,EAAO,QAAQ,oBAAoB,QAAS,OAAO,CACrD,EACA,WACA,OAAO,EAAQ,CACbA,EAAO,QAAQ,EAEf,IAAME,EAAQd,EAAQ,QAAQY,CAAM,EAEhCE,IAAU,KACZd,EAAQ,OAAOc,EAAO,CAAC,EACvB,gBAAgB,GAGlBC,EAAOC,CAAM,EACb,WAAW,CACb,EACA,SAAU,IACV,SACA,MAAM,EAAM,CACVJ,EAAO,QAAQ,EAEf,GAAI,CACFO,EAAQK,EAAGC,CAAI,CAAC,CAClB,OAASE,EAAO,CACd,QAAQF,CAAI,EACZ,WAAW,EACXV,EAAOY,CAAK,CACd,CACF,CACF,EACM,YAAgBf,EAAO,QAAA,EAAOxB,EAAAA,WAAAA,CAAWwB,EAAO,MAAO,CAAC,EAE9DA,EAAO,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAChEZ,EAAQ,KAAKY,CAAM,EACnBZ,EAAQ,MAAM,EAAG,IAAMoB,EAAE,SAAWC,EAAE,UAAYA,EAAE,SAAWD,EAAE,QAAQ,CAC3E,CAAC,CACH,CAEA,SAAS,QAAQ,EAAmB,CAClC,IAAMQ,EAAO5B,EAAQ,MAAM,EAEvB4B,GACFA,EAAK,MAAMH,CAAI,EACf,gBAAgB,GAEhB3B,EAAU,KAAK2B,CAAI,CAEvB,CAEA,SAAS,SAAgB,CACvB,GAAI,GAEJ,GAAa,GACbrB,EAAmB,MAAM,EAEzB,IAAK,IAAMqB,KAAQ1B,EAAO0B,EAAK,UAAU,EAEzC,IAAK,IAAMb,KAAUZ,EAAQ,OAAO,CAAC,EAAGY,EAAO,OAAO,IAAIvB,EAAAA,uBAAyB,EAEnF,eAAe,IAAIA,EAAAA,uBAAyB,EAC5C,WAAW,EACXc,EAAQ,KAAK,CAAE,KAAM,SAAU,CAAC,EAChCA,EAAQ,MAAM,CAVD,CAWf,CAEA,SAAS,MAAM,EAA6B,CAAC,EAAkB,CAC7D,GAAIQ,EAAY,OAAO,QAAQ,QAAQ,EACvC,GAAI,CACF,gBAAgBE,EAAa,QAAS,eAAe,CACvD,OAASc,EAAO,CACd,OAAO,QAAQ,OAAOA,CAAK,CAC7B,CAaA,OAXIjB,IAEJ,eAAe,IAAIrB,EAAAA,wBAAwB,oBAAoB,CAAC,EAChE,EAAe,YAAYwB,CAAY,CAAC,CAAC,SACjC,QAAQ,EACb,GAAmB,CAElB,MADA,QAAQ,EACFc,CACR,CACF,EAEOjB,EACT,CAEA,MAAO,CACL,QACA,eAAgBN,EAAmB,OACnC,QACA,IAAI,UAAW,CACb,OAAOO,CACT,EACA,MACA,IAAI,cAAe,CACjB,OAAOD,CACT,EACA,QAAQ,EAAO,CACbP,EAAQ,KAAK0B,CAAK,CACpB,EACA,QACA,WACA,OAAQ,CACN,MAAO,CAAE,SAAQ,YAAW,SAAQ,OAAQ7B,EAAQ,OAASE,EAAgB,MAAO,CACtF,EACA,QAAS,CAGP,OAFIS,EAAmB,aAEhBN,IAAW,EAAI,OAAS,SACjC,EACA,IAAI,EAAS,EAAY,CAGvB,OAFIM,MAAyB,IAAA,GAEtBR,EAAQ,IAAI2B,EAASC,CAAU,CACxC,EACA,YAAY,EAAe,CACzB,GAAUC,CACZ,EACA,gBAAiB,CACf,GAAa,CACf,EACA,aAAc,CACZ,GAAU,CACZ,CACF,CACF,C"}
@@ -1,2 +1,172 @@
1
- import{unrefTimer as e}from"./_timers.js";import{FamiliarInvalidOptionsError as t,FamiliarQueueFullError as n,FamiliarTerminatedError as r,FamiliarTimeoutError as i}from"./errors.js";import{abortError as a,tapper as o}from"@vielzeug/arsenal";var s=2147483647;function validatePriority(e){if(!Number.isFinite(e))throw new t("`priority` must be a finite number");return e}function validateTimeout(e,n=`timeout`){if(e!==void 0&&(!Number.isInteger(e)||e<1||e>2147483647))throw new t(`\`${n}\` must be an integer from 1 through ${s}`);return e}function createPoolCore(t,s){let c=[...t],l=[],u=[],d=[],f=o(),p=new AbortController,m=0,h=0,g=0,_=0,v=0,y,b=!1;function isIdle(){return c.length===t.length&&l.length===0}function settleIdle(){if(isIdle())for(let e of u.splice(0))e.timer&&clearTimeout(e.timer),e.resolve()}function waitForIdle(t){return isIdle()?Promise.resolve():new Promise((n,r)=>{let a={reject:r,resolve:n};t.timeout!==void 0&&(a.timer=setTimeout(()=>{let e=u.indexOf(a);e!==-1&&u.splice(e,1),r(new i(t.timeout,`Drain`))},t.timeout),e(a.timer)),u.push(a)})}function releaseCapacity(){d.shift()?.resolve()}function rejectCapacity(e){for(let t of d.splice(0))t.reject(e)}function waitForCapacity(e,t){return e?.aborted?Promise.reject(a(e)):new Promise((n,r)=>{let i=!1,o={cleanup(){e?.removeEventListener(`abort`,onAbort)},priority:t,reject(e){if(i)return;i=!0,o.cleanup();let t=d.indexOf(o);t!==-1&&d.splice(t,1),r(e)},resolve(){i||(i=!0,o.cleanup(),n())},sequence:g++},onAbort=()=>o.reject(a(e));e?.addEventListener(`abort`,onAbort,{once:!0}),d.push(o),d.sort((e,t)=>t.priority-e.priority||e.sequence-t.sequence)})}function acquire(e,t){let{priority:i,signal:o}=e;return o?.aborted?Promise.reject(a(o)):(async()=>{for(;;){if(b)throw new r;if(y)throw new r(`Worker is draining`);if(s.maxQueue===void 0||l.length<s.maxQueue)return admit(e,t??(e=>e));if(s.onFull===`reject`)throw new n(s.maxQueue);if(await waitForCapacity(o,i),o?.aborted)throw releaseCapacity(),a(o)}})()}function admit(e,t){let{priority:n,signal:r}=e,i=c.pop();if(i)try{return Promise.resolve(t(i))}catch(e){return release(i),settleIdle(),Promise.reject(e)}return new Promise((e,i)=>{let o={cleanup(){o.signal?.removeEventListener(`abort`,onAbort)},priority:n,reject(e){o.cleanup();let t=l.indexOf(o);t!==-1&&(l.splice(t,1),releaseCapacity()),i(e),settleIdle()},sequence:h++,signal:r,start(n){o.cleanup();try{e(t(n))}catch(e){release(n),settleIdle(),i(e)}}},onAbort=()=>o.reject(a(o.signal));o.signal?.addEventListener(`abort`,onAbort,{once:!0}),l.push(o),l.sort((e,t)=>t.priority-e.priority||e.sequence-t.sequence)})}function release(e){let t=l.shift();t?(t.start(e),releaseCapacity()):c.push(e)}function dispose(){if(!b){b=!0,p.abort();for(let e of t)e.terminate();for(let e of l.splice(0))e.reject(new r);rejectCapacity(new r),settleIdle(),f.emit({type:`dispose`}),f.clear()}}function drain(e={}){if(b)return Promise.resolve();try{validateTimeout(e.timeout,`drain.timeout`)}catch(e){return Promise.reject(e)}return y||(rejectCapacity(new r(`Worker is draining`)),y=waitForIdle(e).then(()=>dispose(),e=>{throw dispose(),e}),y)}return{acquire,disposalSignal:p.signal,dispose,get disposed(){return b},drain,get drainPromise(){return y},emitTap(e){f.emit(e)},release,settleIdle,stats(){return{active:m,completed:_,failed:v,queued:l.length+d.length}},status(){return b?`terminated`:m===0?`idle`:`running`},tap(e,t){return b?()=>void 0:f.tap(e,t)},trackActive(e){m+=e},trackCompleted(){_+=1},trackFailed(){v+=1}}}export{s as MAX_TIMEOUT,createPoolCore,validatePriority,validateTimeout};
1
+ import { unrefTimer as e } from "./_timers.js";
2
+ import { FamiliarInvalidOptionsError as t, FamiliarQueueFullError as n, FamiliarTerminatedError as r, FamiliarTimeoutError as i } from "./errors.js";
3
+ import { abortError as a, tapper as o } from "@vielzeug/arsenal";
4
+ //#region src/_pool-core.ts
5
+ var s = 2147483647;
6
+ function validatePriority(e) {
7
+ if (!Number.isFinite(e)) throw new t("`priority` must be a finite number");
8
+ return e;
9
+ }
10
+ function validateTimeout(e, n = "timeout") {
11
+ if (e !== void 0 && (!Number.isInteger(e) || e < 1 || e > 2147483647)) throw new t(`\`${n}\` must be an integer from 1 through ${s}`);
12
+ return e;
13
+ }
14
+ function createPoolCore(t, s) {
15
+ let c = [...t], l = [], u = [], d = [], f = o(), p = new AbortController(), m = 0, h = 0, g = 0, _ = 0, v = 0, y, b = !1;
16
+ function isIdle() {
17
+ return c.length === t.length && l.length === 0;
18
+ }
19
+ function settleIdle() {
20
+ if (isIdle()) for (let e of u.splice(0)) e.timer && clearTimeout(e.timer), e.resolve();
21
+ }
22
+ function waitForIdle(t) {
23
+ return isIdle() ? Promise.resolve() : new Promise((n, r) => {
24
+ let a = {
25
+ reject: r,
26
+ resolve: n
27
+ };
28
+ t.timeout !== void 0 && (a.timer = setTimeout(() => {
29
+ let e = u.indexOf(a);
30
+ e !== -1 && u.splice(e, 1), r(new i(t.timeout, "Drain"));
31
+ }, t.timeout), e(a.timer)), u.push(a);
32
+ });
33
+ }
34
+ function releaseCapacity() {
35
+ d.shift()?.resolve();
36
+ }
37
+ function rejectCapacity(e) {
38
+ for (let t of d.splice(0)) t.reject(e);
39
+ }
40
+ function waitForCapacity(e, t) {
41
+ return e?.aborted ? Promise.reject(a(e)) : new Promise((n, r) => {
42
+ let i = !1, o = {
43
+ cleanup() {
44
+ e?.removeEventListener("abort", onAbort);
45
+ },
46
+ priority: t,
47
+ reject(e) {
48
+ if (i) return;
49
+ i = !0, o.cleanup();
50
+ let t = d.indexOf(o);
51
+ t !== -1 && d.splice(t, 1), r(e);
52
+ },
53
+ resolve() {
54
+ i || (i = !0, o.cleanup(), n());
55
+ },
56
+ sequence: g++
57
+ }, onAbort = () => o.reject(a(e));
58
+ e?.addEventListener("abort", onAbort, { once: !0 }), d.push(o), d.sort((e, t) => t.priority - e.priority || e.sequence - t.sequence);
59
+ });
60
+ }
61
+ function acquire(e, t) {
62
+ let { priority: i, signal: o } = e;
63
+ return o?.aborted ? Promise.reject(a(o)) : (async () => {
64
+ for (;;) {
65
+ if (b) throw new r();
66
+ if (y) throw new r("Worker is draining");
67
+ if (s.maxQueue === void 0 || l.length < s.maxQueue) return admit(e, t ?? ((e) => e));
68
+ if (s.onFull === "reject") throw new n(s.maxQueue);
69
+ if (await waitForCapacity(o, i), o?.aborted) throw releaseCapacity(), a(o);
70
+ }
71
+ })();
72
+ }
73
+ function admit(e, t) {
74
+ let { priority: n, signal: r } = e, i = c.pop();
75
+ if (i) try {
76
+ return Promise.resolve(t(i));
77
+ } catch (e) {
78
+ return release(i), settleIdle(), Promise.reject(e);
79
+ }
80
+ return new Promise((e, i) => {
81
+ let o = {
82
+ cleanup() {
83
+ o.signal?.removeEventListener("abort", onAbort);
84
+ },
85
+ priority: n,
86
+ reject(e) {
87
+ o.cleanup();
88
+ let t = l.indexOf(o);
89
+ t !== -1 && (l.splice(t, 1), releaseCapacity()), i(e), settleIdle();
90
+ },
91
+ sequence: h++,
92
+ signal: r,
93
+ start(n) {
94
+ o.cleanup();
95
+ try {
96
+ e(t(n));
97
+ } catch (e) {
98
+ release(n), settleIdle(), i(e);
99
+ }
100
+ }
101
+ }, onAbort = () => o.reject(a(o.signal));
102
+ o.signal?.addEventListener("abort", onAbort, { once: !0 }), l.push(o), l.sort((e, t) => t.priority - e.priority || e.sequence - t.sequence);
103
+ });
104
+ }
105
+ function release(e) {
106
+ let t = l.shift();
107
+ t ? (t.start(e), releaseCapacity()) : c.push(e);
108
+ }
109
+ function dispose() {
110
+ if (!b) {
111
+ b = !0, p.abort();
112
+ for (let e of t) e.terminate();
113
+ for (let e of l.splice(0)) e.reject(new r());
114
+ rejectCapacity(new r()), settleIdle(), f.emit({ type: "dispose" }), f.clear();
115
+ }
116
+ }
117
+ function drain(e = {}) {
118
+ if (b) return Promise.resolve();
119
+ try {
120
+ validateTimeout(e.timeout, "drain.timeout");
121
+ } catch (e) {
122
+ return Promise.reject(e);
123
+ }
124
+ return y || (rejectCapacity(new r("Worker is draining")), y = waitForIdle(e).then(() => dispose(), (e) => {
125
+ throw dispose(), e;
126
+ }), y);
127
+ }
128
+ return {
129
+ acquire,
130
+ disposalSignal: p.signal,
131
+ dispose,
132
+ get disposed() {
133
+ return b;
134
+ },
135
+ drain,
136
+ get drainPromise() {
137
+ return y;
138
+ },
139
+ emitTap(e) {
140
+ f.emit(e);
141
+ },
142
+ release,
143
+ settleIdle,
144
+ stats() {
145
+ return {
146
+ active: m,
147
+ completed: _,
148
+ failed: v,
149
+ queued: l.length + d.length
150
+ };
151
+ },
152
+ status() {
153
+ return b ? "terminated" : m === 0 ? "idle" : "running";
154
+ },
155
+ tap(e, t) {
156
+ return b ? () => void 0 : f.tap(e, t);
157
+ },
158
+ trackActive(e) {
159
+ m += e;
160
+ },
161
+ trackCompleted() {
162
+ _ += 1;
163
+ },
164
+ trackFailed() {
165
+ v += 1;
166
+ }
167
+ };
168
+ }
169
+ //#endregion
170
+ export { s as MAX_TIMEOUT, createPoolCore, validatePriority, validateTimeout };
171
+
2
172
  //# sourceMappingURL=_pool-core.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"_pool-core.js","names":[],"sources":["../src/_pool-core.ts"],"sourcesContent":["import { abortError, tapper } from '@vielzeug/arsenal';\nimport { unrefTimer } from './_timers.js';\nimport {\n FamiliarInvalidOptionsError,\n FamiliarQueueFullError,\n type FamiliarRuntimeError,\n FamiliarTerminatedError,\n FamiliarTimeoutError,\n} from './errors.js';\nimport type { DrainOptions, FamiliarTapEvent, WorkerStats, WorkerStatus } from './types.js';\n\nexport const MAX_CONCURRENCY = 512;\nexport const MAX_TIMEOUT = 2_147_483_647;\n\nexport function validatePriority(priority: number): number {\n if (!Number.isFinite(priority)) throw new FamiliarInvalidOptionsError('`priority` must be a finite number');\n return priority;\n}\n\nexport function validateTimeout(timeout: number | undefined, name = 'timeout'): number | undefined {\n if (timeout !== undefined && (!Number.isInteger(timeout) || timeout < 1 || timeout > MAX_TIMEOUT)) {\n throw new FamiliarInvalidOptionsError(`\\`${name}\\` must be an integer from 1 through ${MAX_TIMEOUT}`);\n }\n return timeout;\n}\n\nexport type PoolOptions = {\n defaultTimeout: number | undefined;\n maxQueue: number | undefined;\n onFull: 'reject' | 'wait';\n};\n\nexport type AcquireOptions = {\n priority: number;\n signal?: AbortSignal;\n};\n\n/**\n * A pool slot: tearable on disposal, with an optional channel for reporting\n * unhandled runtime errors. Pool factories wire `notify` to the core's tap so\n * slot owners can emit without holding a pool reference.\n */\nexport type PoolSlot = {\n notify?(error: FamiliarRuntimeError): void;\n terminate(): void;\n};\n\ntype IdleWaiter = {\n reject: (reason: unknown) => void;\n resolve: () => void;\n timer?: ReturnType<typeof setTimeout>;\n};\n\ntype CapacityWaiter = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n resolve(): void;\n sequence: number;\n};\n\ntype SlotWaiter<TSlot> = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n start(slot: TSlot): void;\n sequence: number;\n signal?: AbortSignal;\n};\n\nexport interface PoolCore<TSlot> {\n /**\n * Acquire a free slot and hand it to `fn`, which runs the acquired work.\n * `fn` is invoked synchronously both for an immediately free slot and when\n * a release hands the next waiter its slot, so dispatch never lags a\n * release by a microtask. Honors priority, FIFO order, abort, and `maxQueue`\n * admission; failures surface as a rejected promise. Without `fn`, resolves\n * with the slot instead.\n */\n acquire(options: AcquireOptions): Promise<TSlot>;\n acquire<R>(options: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R>;\n readonly disposalSignal: AbortSignal;\n dispose(): void;\n readonly disposed: boolean;\n drain(options?: DrainOptions): Promise<void>;\n readonly drainPromise: Promise<void> | undefined;\n emitTap(event: FamiliarTapEvent): void;\n /** Return a slot to the pool, handing it to the highest-priority waiter if any. */\n release(slot: TSlot): void;\n settleIdle(): void;\n stats(): WorkerStats;\n status(): WorkerStatus;\n tap(handler: (event: FamiliarTapEvent) => void, options?: { readonly signal?: AbortSignal }): () => void;\n trackActive(delta: number): void;\n trackCompleted(): void;\n trackFailed(): void;\n}\n\n/**\n * The single slot scheduler shared by the result pool and the stream pool:\n * free-slot fast path, priority/FIFO waiter queue, `maxQueue` admission with\n * capacity waiters, idle/drain tracking, stats, disposal, and tap observation.\n * The two pools differ only in what they do with an acquired slot.\n */\nexport function createPoolCore<TSlot extends PoolSlot>(slots: readonly TSlot[], options: PoolOptions): PoolCore<TSlot> {\n const freeSlots = [...slots];\n const waiters: SlotWaiter<TSlot>[] = [];\n const idleWaiters: IdleWaiter[] = [];\n const capacityWaiters: CapacityWaiter[] = [];\n const tappers = tapper<FamiliarTapEvent>();\n const disposalController = new AbortController();\n let active = 0;\n let waiterSequence = 0;\n let capacitySequence = 0;\n let completed = 0;\n let failed = 0;\n let drainPromise: Promise<void> | undefined;\n let terminated = false;\n\n function isIdle(): boolean {\n return freeSlots.length === slots.length && waiters.length === 0;\n }\n\n function settleIdle(): void {\n if (!isIdle()) return;\n\n for (const waiter of idleWaiters.splice(0)) {\n if (waiter.timer) clearTimeout(waiter.timer);\n\n waiter.resolve();\n }\n }\n\n function waitForIdle(drainOptions: DrainOptions): Promise<void> {\n if (isIdle()) return Promise.resolve();\n\n return new Promise<void>((resolve, reject) => {\n const waiter: IdleWaiter = { reject, resolve };\n\n if (drainOptions.timeout !== undefined) {\n waiter.timer = setTimeout(() => {\n const index = idleWaiters.indexOf(waiter);\n\n if (index !== -1) idleWaiters.splice(index, 1);\n\n reject(new FamiliarTimeoutError(drainOptions.timeout!, 'Drain'));\n }, drainOptions.timeout);\n unrefTimer(waiter.timer);\n }\n\n idleWaiters.push(waiter);\n });\n }\n\n function releaseCapacity(): void {\n capacityWaiters.shift()?.resolve();\n }\n\n function rejectCapacity(reason: unknown): void {\n for (const waiter of capacityWaiters.splice(0)) waiter.reject(reason);\n }\n\n function waitForCapacity(signal: AbortSignal | undefined, priority: number): Promise<void> {\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return new Promise<void>((resolve, reject) => {\n let settled = false;\n const waiter: CapacityWaiter = {\n cleanup() {\n signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n\n const index = capacityWaiters.indexOf(waiter);\n\n if (index !== -1) capacityWaiters.splice(index, 1);\n\n reject(reason);\n },\n resolve() {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n resolve();\n },\n sequence: capacitySequence++,\n };\n const onAbort = () => waiter.reject(abortError(signal!));\n\n signal?.addEventListener('abort', onAbort, { once: true });\n capacityWaiters.push(waiter);\n capacityWaiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function acquire<R>(acquireOptions: AcquireOptions, fn?: (slot: TSlot) => R | Promise<R>): Promise<R | TSlot> {\n const { priority, signal } = acquireOptions;\n\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return (async () => {\n for (;;) {\n if (terminated) throw new FamiliarTerminatedError();\n\n if (drainPromise) throw new FamiliarTerminatedError('Worker is draining');\n\n if (options.maxQueue === undefined || waiters.length < options.maxQueue) {\n return admit(acquireOptions, (fn ?? ((slot) => slot)) as (slot: TSlot) => R | Promise<R>);\n }\n\n if (options.onFull === 'reject') throw new FamiliarQueueFullError(options.maxQueue);\n\n await waitForCapacity(signal, priority);\n\n if (signal?.aborted) {\n releaseCapacity();\n throw abortError(signal);\n }\n }\n })();\n }\n\n function admit<R>(acquireOptions: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R> {\n const { priority, signal } = acquireOptions;\n\n const free = freeSlots.pop();\n\n if (free) {\n try {\n return Promise.resolve(fn(free));\n } catch (error) {\n // A synchronous throw must not strand the slot: idle would never settle.\n release(free);\n settleIdle();\n\n return Promise.reject(error);\n }\n }\n\n return new Promise<R>((resolve, reject) => {\n const waiter: SlotWaiter<TSlot> = {\n cleanup() {\n waiter.signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n waiter.cleanup();\n\n const index = waiters.indexOf(waiter);\n\n if (index !== -1) {\n waiters.splice(index, 1);\n releaseCapacity();\n }\n\n reject(reason);\n settleIdle();\n },\n sequence: waiterSequence++,\n signal,\n start(slot) {\n waiter.cleanup();\n\n try {\n resolve(fn(slot));\n } catch (error) {\n release(slot);\n settleIdle();\n reject(error);\n }\n },\n };\n const onAbort = () => waiter.reject(abortError(waiter.signal!));\n\n waiter.signal?.addEventListener('abort', onAbort, { once: true });\n waiters.push(waiter);\n waiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function release(slot: TSlot): void {\n const next = waiters.shift();\n\n if (next) {\n next.start(slot);\n releaseCapacity();\n } else {\n freeSlots.push(slot);\n }\n }\n\n function dispose(): void {\n if (terminated) return;\n\n terminated = true;\n disposalController.abort();\n\n for (const slot of slots) slot.terminate();\n\n for (const waiter of waiters.splice(0)) waiter.reject(new FamiliarTerminatedError());\n\n rejectCapacity(new FamiliarTerminatedError());\n settleIdle();\n tappers.emit({ type: 'dispose' });\n tappers.clear();\n }\n\n function drain(drainOptions: DrainOptions = {}): Promise<void> {\n if (terminated) return Promise.resolve();\n try {\n validateTimeout(drainOptions.timeout, 'drain.timeout');\n } catch (error) {\n return Promise.reject(error);\n }\n\n if (drainPromise) return drainPromise;\n\n rejectCapacity(new FamiliarTerminatedError('Worker is draining'));\n drainPromise = waitForIdle(drainOptions).then(\n () => dispose(),\n (error: unknown) => {\n dispose();\n throw error;\n },\n );\n\n return drainPromise;\n }\n\n return {\n acquire,\n disposalSignal: disposalController.signal,\n dispose,\n get disposed() {\n return terminated;\n },\n drain,\n get drainPromise() {\n return drainPromise;\n },\n emitTap(event) {\n tappers.emit(event);\n },\n release,\n settleIdle,\n stats() {\n return { active, completed, failed, queued: waiters.length + capacityWaiters.length };\n },\n status() {\n if (terminated) return 'terminated';\n\n return active === 0 ? 'idle' : 'running';\n },\n tap(handler, tapOptions) {\n if (terminated) return () => undefined;\n\n return tappers.tap(handler, tapOptions);\n },\n trackActive(delta: number) {\n active += delta;\n },\n trackCompleted() {\n completed += 1;\n },\n trackFailed() {\n failed += 1;\n },\n };\n}\n"],"mappings":"kPAYA,IAAa,EAAc,WAE3B,SAAgB,iBAAiB,EAA0B,CACzD,GAAI,CAAC,OAAO,SAAS,CAAQ,EAAG,MAAM,IAAI,EAA4B,oCAAoC,EAC1G,OAAO,CACT,CAEA,SAAgB,gBAAgB,EAA6B,EAAO,UAA+B,CACjG,GAAI,IAAY,IAAA,KAAc,CAAC,OAAO,UAAU,CAAO,GAAK,EAAU,GAAK,EAAA,YACzE,MAAM,IAAI,EAA4B,KAAK,EAAK,uCAAuC,GAAa,EAEtG,OAAO,CACT,CAgFA,SAAgB,eAAuC,EAAyB,EAAuC,CACrH,IAAM,EAAY,CAAC,GAAG,CAAK,EACrB,EAA+B,CAAC,EAChC,EAA4B,CAAC,EAC7B,EAAoC,CAAC,EACrC,EAAU,EAAyB,EACnC,EAAqB,IAAI,gBAC3B,EAAS,EACT,EAAiB,EACjB,EAAmB,EACnB,EAAY,EACZ,EAAS,EACT,EACA,EAAa,GAEjB,SAAS,QAAkB,CACzB,OAAO,EAAU,SAAW,EAAM,QAAU,EAAQ,SAAW,CACjE,CAEA,SAAS,YAAmB,CACrB,UAAO,EAEZ,IAAK,IAAM,KAAU,EAAY,OAAO,CAAC,EACnC,EAAO,OAAO,aAAa,EAAO,KAAK,EAE3C,EAAO,QAAQ,CAEnB,CAEA,SAAS,YAAY,EAA2C,CAG9D,OAFI,OAAO,EAAU,QAAQ,QAAQ,EAE9B,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAM,EAAqB,CAAE,SAAQ,SAAQ,EAEzC,EAAa,UAAY,IAAA,KAC3B,EAAO,MAAQ,eAAiB,CAC9B,IAAM,EAAQ,EAAY,QAAQ,CAAM,EAEpC,IAAU,IAAI,EAAY,OAAO,EAAO,CAAC,EAE7C,EAAO,IAAI,EAAqB,EAAa,QAAU,OAAO,CAAC,CACjE,EAAG,EAAa,OAAO,EACvB,EAAW,EAAO,KAAK,GAGzB,EAAY,KAAK,CAAM,CACzB,CAAC,CACH,CAEA,SAAS,iBAAwB,CAC/B,EAAgB,MAAM,CAAC,EAAE,QAAQ,CACnC,CAEA,SAAS,eAAe,EAAuB,CAC7C,IAAK,IAAM,KAAU,EAAgB,OAAO,CAAC,EAAG,EAAO,OAAO,CAAM,CACtE,CAEA,SAAS,gBAAgB,EAAiC,EAAiC,CAGzF,OAFI,GAAQ,QAAgB,QAAQ,OAAO,EAAW,CAAM,CAAC,EAEtD,IAAI,SAAe,EAAS,IAAW,CAC5C,IAAI,EAAU,GACR,EAAyB,CAC7B,SAAU,CACR,GAAQ,oBAAoB,QAAS,OAAO,CAC9C,EACA,WACA,OAAO,EAAQ,CACb,GAAI,EAAS,OAEb,EAAU,GACV,EAAO,QAAQ,EAEf,IAAM,EAAQ,EAAgB,QAAQ,CAAM,EAExC,IAAU,IAAI,EAAgB,OAAO,EAAO,CAAC,EAEjD,EAAO,CAAM,CACf,EACA,SAAU,CACJ,IAEJ,EAAU,GACV,EAAO,QAAQ,EACf,EAAQ,EACV,EACA,SAAU,GACZ,EACM,YAAgB,EAAO,OAAO,EAAW,CAAO,CAAC,EAEvD,GAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EACzD,EAAgB,KAAK,CAAM,EAC3B,EAAgB,MAAM,EAAG,IAAM,EAAE,SAAW,EAAE,UAAY,EAAE,SAAW,EAAE,QAAQ,CACnF,CAAC,CACH,CAEA,SAAS,QAAW,EAAgC,EAA0D,CAC5G,GAAM,CAAE,WAAU,UAAW,EAI7B,OAFI,GAAQ,QAAgB,QAAQ,OAAO,EAAW,CAAM,CAAC,GAErD,SAAY,CAClB,OAAS,CACP,GAAI,EAAY,MAAM,IAAI,EAE1B,GAAI,EAAc,MAAM,IAAI,EAAwB,oBAAoB,EAExE,GAAI,EAAQ,WAAa,IAAA,IAAa,EAAQ,OAAS,EAAQ,SAC7D,OAAO,MAAM,EAAiB,IAAQ,GAAS,EAAyC,EAG1F,GAAI,EAAQ,SAAW,SAAU,MAAM,IAAI,EAAuB,EAAQ,QAAQ,EAIlF,GAFA,MAAM,gBAAgB,EAAQ,CAAQ,EAElC,GAAQ,QAEV,MADA,gBAAgB,EACV,EAAW,CAAM,CAE3B,CACF,EAAA,CAAG,CACL,CAEA,SAAS,MAAS,EAAgC,EAAiD,CACjG,GAAM,CAAE,WAAU,UAAW,EAEvB,EAAO,EAAU,IAAI,EAE3B,GAAI,EACF,GAAI,CACF,OAAO,QAAQ,QAAQ,EAAG,CAAI,CAAC,CACjC,OAAS,EAAO,CAKd,OAHA,QAAQ,CAAI,EACZ,WAAW,EAEJ,QAAQ,OAAO,CAAK,CAC7B,CAGF,OAAO,IAAI,SAAY,EAAS,IAAW,CACzC,IAAM,EAA4B,CAChC,SAAU,CACR,EAAO,QAAQ,oBAAoB,QAAS,OAAO,CACrD,EACA,WACA,OAAO,EAAQ,CACb,EAAO,QAAQ,EAEf,IAAM,EAAQ,EAAQ,QAAQ,CAAM,EAEhC,IAAU,KACZ,EAAQ,OAAO,EAAO,CAAC,EACvB,gBAAgB,GAGlB,EAAO,CAAM,EACb,WAAW,CACb,EACA,SAAU,IACV,SACA,MAAM,EAAM,CACV,EAAO,QAAQ,EAEf,GAAI,CACF,EAAQ,EAAG,CAAI,CAAC,CAClB,OAAS,EAAO,CACd,QAAQ,CAAI,EACZ,WAAW,EACX,EAAO,CAAK,CACd,CACF,CACF,EACM,YAAgB,EAAO,OAAO,EAAW,EAAO,MAAO,CAAC,EAE9D,EAAO,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAChE,EAAQ,KAAK,CAAM,EACnB,EAAQ,MAAM,EAAG,IAAM,EAAE,SAAW,EAAE,UAAY,EAAE,SAAW,EAAE,QAAQ,CAC3E,CAAC,CACH,CAEA,SAAS,QAAQ,EAAmB,CAClC,IAAM,EAAO,EAAQ,MAAM,EAEvB,GACF,EAAK,MAAM,CAAI,EACf,gBAAgB,GAEhB,EAAU,KAAK,CAAI,CAEvB,CAEA,SAAS,SAAgB,CACnB,MAGJ,CADA,EAAa,GACb,EAAmB,MAAM,EAEzB,IAAK,IAAM,KAAQ,EAAO,EAAK,UAAU,EAEzC,IAAK,IAAM,KAAU,EAAQ,OAAO,CAAC,EAAG,EAAO,OAAO,IAAI,CAAyB,EAEnF,eAAe,IAAI,CAAyB,EAC5C,WAAW,EACX,EAAQ,KAAK,CAAE,KAAM,SAAU,CAAC,EAChC,EAAQ,MAAM,CATW,CAU3B,CAEA,SAAS,MAAM,EAA6B,CAAC,EAAkB,CAC7D,GAAI,EAAY,OAAO,QAAQ,QAAQ,EACvC,GAAI,CACF,gBAAgB,EAAa,QAAS,eAAe,CACvD,OAAS,EAAO,CACd,OAAO,QAAQ,OAAO,CAAK,CAC7B,CAaA,OAXI,IAEJ,eAAe,IAAI,EAAwB,oBAAoB,CAAC,EAChE,EAAe,YAAY,CAAY,CAAC,CAAC,SACjC,QAAQ,EACb,GAAmB,CAElB,MADA,QAAQ,EACF,CACR,CACF,EAEO,EACT,CAEA,MAAO,CACL,QACA,eAAgB,EAAmB,OACnC,QACA,IAAI,UAAW,CACb,OAAO,CACT,EACA,MACA,IAAI,cAAe,CACjB,OAAO,CACT,EACA,QAAQ,EAAO,CACb,EAAQ,KAAK,CAAK,CACpB,EACA,QACA,WACA,OAAQ,CACN,MAAO,CAAE,SAAQ,YAAW,SAAQ,OAAQ,EAAQ,OAAS,EAAgB,MAAO,CACtF,EACA,QAAS,CAGP,OAFI,EAAmB,aAEhB,IAAW,EAAI,OAAS,SACjC,EACA,IAAI,EAAS,EAAY,CAGvB,OAFI,MAAyB,IAAA,GAEtB,EAAQ,IAAI,EAAS,CAAU,CACxC,EACA,YAAY,EAAe,CACzB,GAAU,CACZ,EACA,gBAAiB,CACf,GAAa,CACf,EACA,aAAc,CACZ,GAAU,CACZ,CACF,CACF"}
1
+ {"version":3,"file":"_pool-core.js","names":["MAX_TIMEOUT","priority","FamiliarInvalidOptionsError","timeout","name","freeSlots","slots","waiters","idleWaiters","capacityWaiters","tappers","tapper","disposalController","active","waiterSequence","capacitySequence","completed","failed","drainPromise","terminated","waiter","drainOptions","index","reject","FamiliarTimeoutError","unrefTimer","reason","signal","abortError","settled","resolve","b","a","acquireOptions","FamiliarTerminatedError","options","fn","slot","FamiliarQueueFullError","free","error","next","event","handler","tapOptions","delta"],"sources":["../src/_pool-core.ts"],"sourcesContent":["import { abortError, tapper } from '@vielzeug/arsenal';\nimport { unrefTimer } from './_timers.js';\nimport {\n FamiliarInvalidOptionsError,\n FamiliarQueueFullError,\n type FamiliarRuntimeError,\n FamiliarTerminatedError,\n FamiliarTimeoutError,\n} from './errors.js';\nimport type { DrainOptions, FamiliarTapEvent, WorkerStats, WorkerStatus } from './types.js';\n\nexport const MAX_CONCURRENCY = 512;\nexport const MAX_TIMEOUT = 2_147_483_647;\n\nexport function validatePriority(priority: number): number {\n if (!Number.isFinite(priority)) throw new FamiliarInvalidOptionsError('`priority` must be a finite number');\n return priority;\n}\n\nexport function validateTimeout(timeout: number | undefined, name = 'timeout'): number | undefined {\n if (timeout !== undefined && (!Number.isInteger(timeout) || timeout < 1 || timeout > MAX_TIMEOUT)) {\n throw new FamiliarInvalidOptionsError(`\\`${name}\\` must be an integer from 1 through ${MAX_TIMEOUT}`);\n }\n return timeout;\n}\n\nexport type PoolOptions = {\n defaultTimeout: number | undefined;\n maxQueue: number | undefined;\n onFull: 'reject' | 'wait';\n};\n\nexport type AcquireOptions = {\n priority: number;\n signal?: AbortSignal;\n};\n\n/**\n * A pool slot: tearable on disposal, with an optional channel for reporting\n * unhandled runtime errors. Pool factories wire `notify` to the core's tap so\n * slot owners can emit without holding a pool reference.\n */\nexport type PoolSlot = {\n notify?(error: FamiliarRuntimeError): void;\n terminate(): void;\n};\n\ntype IdleWaiter = {\n reject: (reason: unknown) => void;\n resolve: () => void;\n timer?: ReturnType<typeof setTimeout>;\n};\n\ntype CapacityWaiter = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n resolve(): void;\n sequence: number;\n};\n\ntype SlotWaiter<TSlot> = {\n cleanup(): void;\n priority: number;\n reject(reason: unknown): void;\n start(slot: TSlot): void;\n sequence: number;\n signal?: AbortSignal;\n};\n\nexport interface PoolCore<TSlot> {\n /**\n * Acquire a free slot and hand it to `fn`, which runs the acquired work.\n * `fn` is invoked synchronously both for an immediately free slot and when\n * a release hands the next waiter its slot, so dispatch never lags a\n * release by a microtask. Honors priority, FIFO order, abort, and `maxQueue`\n * admission; failures surface as a rejected promise. Without `fn`, resolves\n * with the slot instead.\n */\n acquire(options: AcquireOptions): Promise<TSlot>;\n acquire<R>(options: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R>;\n readonly disposalSignal: AbortSignal;\n dispose(): void;\n readonly disposed: boolean;\n drain(options?: DrainOptions): Promise<void>;\n readonly drainPromise: Promise<void> | undefined;\n emitTap(event: FamiliarTapEvent): void;\n /** Return a slot to the pool, handing it to the highest-priority waiter if any. */\n release(slot: TSlot): void;\n settleIdle(): void;\n stats(): WorkerStats;\n status(): WorkerStatus;\n tap(handler: (event: FamiliarTapEvent) => void, options?: { readonly signal?: AbortSignal }): () => void;\n trackActive(delta: number): void;\n trackCompleted(): void;\n trackFailed(): void;\n}\n\n/**\n * The single slot scheduler shared by the result pool and the stream pool:\n * free-slot fast path, priority/FIFO waiter queue, `maxQueue` admission with\n * capacity waiters, idle/drain tracking, stats, disposal, and tap observation.\n * The two pools differ only in what they do with an acquired slot.\n */\nexport function createPoolCore<TSlot extends PoolSlot>(slots: readonly TSlot[], options: PoolOptions): PoolCore<TSlot> {\n const freeSlots = [...slots];\n const waiters: SlotWaiter<TSlot>[] = [];\n const idleWaiters: IdleWaiter[] = [];\n const capacityWaiters: CapacityWaiter[] = [];\n const tappers = tapper<FamiliarTapEvent>();\n const disposalController = new AbortController();\n let active = 0;\n let waiterSequence = 0;\n let capacitySequence = 0;\n let completed = 0;\n let failed = 0;\n let drainPromise: Promise<void> | undefined;\n let terminated = false;\n\n function isIdle(): boolean {\n return freeSlots.length === slots.length && waiters.length === 0;\n }\n\n function settleIdle(): void {\n if (!isIdle()) return;\n\n for (const waiter of idleWaiters.splice(0)) {\n if (waiter.timer) clearTimeout(waiter.timer);\n\n waiter.resolve();\n }\n }\n\n function waitForIdle(drainOptions: DrainOptions): Promise<void> {\n if (isIdle()) return Promise.resolve();\n\n return new Promise<void>((resolve, reject) => {\n const waiter: IdleWaiter = { reject, resolve };\n\n if (drainOptions.timeout !== undefined) {\n waiter.timer = setTimeout(() => {\n const index = idleWaiters.indexOf(waiter);\n\n if (index !== -1) idleWaiters.splice(index, 1);\n\n reject(new FamiliarTimeoutError(drainOptions.timeout!, 'Drain'));\n }, drainOptions.timeout);\n unrefTimer(waiter.timer);\n }\n\n idleWaiters.push(waiter);\n });\n }\n\n function releaseCapacity(): void {\n capacityWaiters.shift()?.resolve();\n }\n\n function rejectCapacity(reason: unknown): void {\n for (const waiter of capacityWaiters.splice(0)) waiter.reject(reason);\n }\n\n function waitForCapacity(signal: AbortSignal | undefined, priority: number): Promise<void> {\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return new Promise<void>((resolve, reject) => {\n let settled = false;\n const waiter: CapacityWaiter = {\n cleanup() {\n signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n\n const index = capacityWaiters.indexOf(waiter);\n\n if (index !== -1) capacityWaiters.splice(index, 1);\n\n reject(reason);\n },\n resolve() {\n if (settled) return;\n\n settled = true;\n waiter.cleanup();\n resolve();\n },\n sequence: capacitySequence++,\n };\n const onAbort = () => waiter.reject(abortError(signal!));\n\n signal?.addEventListener('abort', onAbort, { once: true });\n capacityWaiters.push(waiter);\n capacityWaiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function acquire<R>(acquireOptions: AcquireOptions, fn?: (slot: TSlot) => R | Promise<R>): Promise<R | TSlot> {\n const { priority, signal } = acquireOptions;\n\n if (signal?.aborted) return Promise.reject(abortError(signal));\n\n return (async () => {\n for (;;) {\n if (terminated) throw new FamiliarTerminatedError();\n\n if (drainPromise) throw new FamiliarTerminatedError('Worker is draining');\n\n if (options.maxQueue === undefined || waiters.length < options.maxQueue) {\n return admit(acquireOptions, (fn ?? ((slot) => slot)) as (slot: TSlot) => R | Promise<R>);\n }\n\n if (options.onFull === 'reject') throw new FamiliarQueueFullError(options.maxQueue);\n\n await waitForCapacity(signal, priority);\n\n if (signal?.aborted) {\n releaseCapacity();\n throw abortError(signal);\n }\n }\n })();\n }\n\n function admit<R>(acquireOptions: AcquireOptions, fn: (slot: TSlot) => R | Promise<R>): Promise<R> {\n const { priority, signal } = acquireOptions;\n\n const free = freeSlots.pop();\n\n if (free) {\n try {\n return Promise.resolve(fn(free));\n } catch (error) {\n // A synchronous throw must not strand the slot: idle would never settle.\n release(free);\n settleIdle();\n\n return Promise.reject(error);\n }\n }\n\n return new Promise<R>((resolve, reject) => {\n const waiter: SlotWaiter<TSlot> = {\n cleanup() {\n waiter.signal?.removeEventListener('abort', onAbort);\n },\n priority,\n reject(reason) {\n waiter.cleanup();\n\n const index = waiters.indexOf(waiter);\n\n if (index !== -1) {\n waiters.splice(index, 1);\n releaseCapacity();\n }\n\n reject(reason);\n settleIdle();\n },\n sequence: waiterSequence++,\n signal,\n start(slot) {\n waiter.cleanup();\n\n try {\n resolve(fn(slot));\n } catch (error) {\n release(slot);\n settleIdle();\n reject(error);\n }\n },\n };\n const onAbort = () => waiter.reject(abortError(waiter.signal!));\n\n waiter.signal?.addEventListener('abort', onAbort, { once: true });\n waiters.push(waiter);\n waiters.sort((a, b) => b.priority - a.priority || a.sequence - b.sequence);\n });\n }\n\n function release(slot: TSlot): void {\n const next = waiters.shift();\n\n if (next) {\n next.start(slot);\n releaseCapacity();\n } else {\n freeSlots.push(slot);\n }\n }\n\n function dispose(): void {\n if (terminated) return;\n\n terminated = true;\n disposalController.abort();\n\n for (const slot of slots) slot.terminate();\n\n for (const waiter of waiters.splice(0)) waiter.reject(new FamiliarTerminatedError());\n\n rejectCapacity(new FamiliarTerminatedError());\n settleIdle();\n tappers.emit({ type: 'dispose' });\n tappers.clear();\n }\n\n function drain(drainOptions: DrainOptions = {}): Promise<void> {\n if (terminated) return Promise.resolve();\n try {\n validateTimeout(drainOptions.timeout, 'drain.timeout');\n } catch (error) {\n return Promise.reject(error);\n }\n\n if (drainPromise) return drainPromise;\n\n rejectCapacity(new FamiliarTerminatedError('Worker is draining'));\n drainPromise = waitForIdle(drainOptions).then(\n () => dispose(),\n (error: unknown) => {\n dispose();\n throw error;\n },\n );\n\n return drainPromise;\n }\n\n return {\n acquire,\n disposalSignal: disposalController.signal,\n dispose,\n get disposed() {\n return terminated;\n },\n drain,\n get drainPromise() {\n return drainPromise;\n },\n emitTap(event) {\n tappers.emit(event);\n },\n release,\n settleIdle,\n stats() {\n return { active, completed, failed, queued: waiters.length + capacityWaiters.length };\n },\n status() {\n if (terminated) return 'terminated';\n\n return active === 0 ? 'idle' : 'running';\n },\n tap(handler, tapOptions) {\n if (terminated) return () => undefined;\n\n return tappers.tap(handler, tapOptions);\n },\n trackActive(delta: number) {\n active += delta;\n },\n trackCompleted() {\n completed += 1;\n },\n trackFailed() {\n failed += 1;\n },\n };\n}\n"],"mappings":";;;;AAYA,IAAaA,IAAc;AAE3B,SAAgB,iBAAiB,GAA0B;CACzD,IAAI,CAAC,OAAO,SAASC,CAAQ,GAAG,MAAM,IAAIC,EAA4B,oCAAoC;CAC1G,OAAOD;AACT;AAEA,SAAgB,gBAAgB,GAA6B,IAAO,WAA+B;CACjG,IAAIE,MAAY,KAAA,MAAc,CAAC,OAAO,UAAUA,CAAO,KAAKA,IAAU,KAAKA,IAAA,aACzE,MAAM,IAAID,EAA4B,KAAKE,EAAK,uCAAuCJ,GAAa;CAEtG,OAAOG;AACT;AAgFA,SAAgB,eAAuC,GAAyB,GAAuC;CACrH,IAAME,IAAY,CAAC,GAAGC,CAAK,GACrBC,IAA+B,CAAC,GAChCC,IAA4B,CAAC,GAC7BC,IAAoC,CAAC,GACrCC,IAAUC,EAAyB,GACnCC,IAAqB,IAAI,gBAAgB,GAC3CC,IAAS,GACTC,IAAiB,GACjBC,IAAmB,GACnBC,IAAY,GACZC,IAAS,GACTC,GACAC,IAAa;CAEjB,SAAS,SAAkB;EACzB,OAAOd,EAAU,WAAWC,EAAM,UAAUC,EAAQ,WAAW;CACjE;CAEA,SAAS,aAAmB;EAC1B,IAAK,OAAO,GAEZ,KAAK,IAAMa,KAAUZ,EAAY,OAAO,CAAC,GAGvC,AAFIY,EAAO,SAAO,aAAaA,EAAO,KAAK,GAE3CA,EAAO,QAAQ;CAEnB;CAEA,SAAS,YAAY,GAA2C;EAG9D,OAFI,OAAO,IAAU,QAAQ,QAAQ,IAE9B,IAAI,SAAe,GAAS,MAAW;GAC5C,IAAMA,IAAqB;IAAE;IAAQ;GAAQ;GAa7C,AAXIC,EAAa,YAAY,KAAA,MAC3B,EAAO,QAAQ,iBAAiB;IAC9B,IAAMC,IAAQd,EAAY,QAAQY,CAAM;IAIxC,AAFIE,MAAU,MAAId,EAAY,OAAOc,GAAO,CAAC,GAE7CC,EAAO,IAAIC,EAAqBH,EAAa,SAAU,OAAO,CAAC;GACjE,GAAGA,EAAa,OAAO,GACvBI,EAAWL,EAAO,KAAK,IAGzBZ,EAAY,KAAKY,CAAM;EACzB,CAAC;CACH;CAEA,SAAS,kBAAwB;EAC/B,EAAgB,MAAM,CAAC,EAAE,QAAQ;CACnC;CAEA,SAAS,eAAe,GAAuB;EAC7C,KAAK,IAAMA,KAAUX,EAAgB,OAAO,CAAC,GAAG,EAAO,OAAOiB,CAAM;CACtE;CAEA,SAAS,gBAAgB,GAAiC,GAAiC;EAGzF,OAFIC,GAAQ,UAAgB,QAAQ,OAAOC,EAAWD,CAAM,CAAC,IAEtD,IAAI,SAAe,GAAS,MAAW;GAC5C,IAAIE,IAAU,IACRT,IAAyB;IAC7B,UAAU;KACR,GAAQ,oBAAoB,SAAS,OAAO;IAC9C;IACA;IACA,OAAO,GAAQ;KACb,IAAIS,GAAS;KAGb,AADA,IAAU,IACVT,EAAO,QAAQ;KAEf,IAAME,IAAQb,EAAgB,QAAQW,CAAM;KAI5C,AAFIE,MAAU,MAAIb,EAAgB,OAAOa,GAAO,CAAC,GAEjDC,EAAOG,CAAM;IACf;IACA,UAAU;KACR,AAAIG,MAEJ,IAAU,IACVT,EAAO,QAAQ,GACfU,EAAQ;IACV;IACA,UAAU;GACZ,GACM,gBAAgBV,EAAO,OAAOQ,EAAWD,CAAO,CAAC;GAIvD,AAFAA,GAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,GAAK,CAAC,GACzDlB,EAAgB,KAAKW,CAAM,GAC3BX,EAAgB,MAAM,GAAG,MAAMsB,EAAE,WAAWC,EAAE,YAAYA,EAAE,WAAWD,EAAE,QAAQ;EACnF,CAAC;CACH;CAEA,SAAS,QAAW,GAAgC,GAA0D;EAC5G,IAAM,EAAE,aAAU,cAAWE;EAI7B,OAFIN,GAAQ,UAAgB,QAAQ,OAAOC,EAAWD,CAAM,CAAC,KAErD,YAAY;GAClB,SAAS;IACP,IAAIR,GAAY,MAAM,IAAIe,EAAwB;IAElD,IAAIhB,GAAc,MAAM,IAAIgB,EAAwB,oBAAoB;IAExE,IAAIC,EAAQ,aAAa,KAAA,KAAa5B,EAAQ,SAAS4B,EAAQ,UAC7D,OAAO,MAAMF,GAAiBG,OAAQ,MAASC,EAAyC;IAG1F,IAAIF,EAAQ,WAAW,UAAU,MAAM,IAAIG,EAAuBH,EAAQ,QAAQ;IAIlF,IAFA,MAAM,gBAAgBR,GAAQ1B,CAAQ,GAElC0B,GAAQ,SAEV,MADA,gBAAgB,GACVC,EAAWD,CAAM;GAE3B;EACF,EAAA,CAAG;CACL;CAEA,SAAS,MAAS,GAAgC,GAAiD;EACjG,IAAM,EAAE,aAAU,cAAWM,GAEvBM,IAAOlC,EAAU,IAAI;EAE3B,IAAIkC,GACF,IAAI;GACF,OAAO,QAAQ,QAAQH,EAAGG,CAAI,CAAC;EACjC,SAASC,GAAO;GAKd,OAHA,QAAQD,CAAI,GACZ,WAAW,GAEJ,QAAQ,OAAOC,CAAK;EAC7B;EAGF,OAAO,IAAI,SAAY,GAAS,MAAW;GACzC,IAAMpB,IAA4B;IAChC,UAAU;KACR,EAAO,QAAQ,oBAAoB,SAAS,OAAO;IACrD;IACA;IACA,OAAO,GAAQ;KACb,EAAO,QAAQ;KAEf,IAAME,IAAQf,EAAQ,QAAQa,CAAM;KAQpC,AANIE,MAAU,OACZf,EAAQ,OAAOe,GAAO,CAAC,GACvB,gBAAgB,IAGlBC,EAAOG,CAAM,GACb,WAAW;IACb;IACA,UAAU;IACV;IACA,MAAM,GAAM;KACV,EAAO,QAAQ;KAEf,IAAI;MACF,EAAQU,EAAGC,CAAI,CAAC;KAClB,SAASG,GAAO;MAGd,AAFA,QAAQH,CAAI,GACZ,WAAW,GACXd,EAAOiB,CAAK;KACd;IACF;GACF,GACM,gBAAgBpB,EAAO,OAAOQ,EAAWR,EAAO,MAAO,CAAC;GAI9D,AAFAA,EAAO,QAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,GAAK,CAAC,GAChEb,EAAQ,KAAKa,CAAM,GACnBb,EAAQ,MAAM,GAAG,MAAMwB,EAAE,WAAWC,EAAE,YAAYA,EAAE,WAAWD,EAAE,QAAQ;EAC3E,CAAC;CACH;CAEA,SAAS,QAAQ,GAAmB;EAClC,IAAMU,IAAOlC,EAAQ,MAAM;EAE3B,AAAIkC,KACFA,EAAK,MAAMJ,CAAI,GACf,gBAAgB,KAEhBhC,EAAU,KAAKgC,CAAI;CAEvB;CAEA,SAAS,UAAgB;EACvB,IAAI,IAEJ;GACA,AADA,IAAa,IACbzB,EAAmB,MAAM;GAEzB,KAAK,IAAMyB,KAAQ/B,GAAO,EAAK,UAAU;GAEzC,KAAK,IAAMc,KAAUb,EAAQ,OAAO,CAAC,GAAG,EAAO,OAAO,IAAI2B,EAAwB,CAAC;GAKnF,AAHA,eAAe,IAAIA,EAAwB,CAAC,GAC5C,WAAW,GACXxB,EAAQ,KAAK,EAAE,MAAM,UAAU,CAAC,GAChCA,EAAQ,MAAM;EAVD;CAWf;CAEA,SAAS,MAAM,IAA6B,CAAC,GAAkB;EAC7D,IAAIS,GAAY,OAAO,QAAQ,QAAQ;EACvC,IAAI;GACF,gBAAgBE,EAAa,SAAS,eAAe;EACvD,SAASmB,GAAO;GACd,OAAO,QAAQ,OAAOA,CAAK;EAC7B;EAaA,OAXItB,MAEJ,eAAe,IAAIgB,EAAwB,oBAAoB,CAAC,GAChE,IAAe,YAAYb,CAAY,CAAC,CAAC,WACjC,QAAQ,IACb,MAAmB;GAElB,MADA,QAAQ,GACFmB;EACR,CACF,GAEOtB;CACT;CAEA,OAAO;EACL;EACA,gBAAgBN,EAAmB;EACnC;EACA,IAAI,WAAW;GACb,OAAOO;EACT;EACA;EACA,IAAI,eAAe;GACjB,OAAOD;EACT;EACA,QAAQ,GAAO;GACb,EAAQ,KAAKwB,CAAK;EACpB;EACA;EACA;EACA,QAAQ;GACN,OAAO;IAAE;IAAQ;IAAW;IAAQ,QAAQnC,EAAQ,SAASE,EAAgB;GAAO;EACtF;EACA,SAAS;GAGP,OAFIU,IAAmB,eAEhBN,MAAW,IAAI,SAAS;EACjC;EACA,IAAI,GAAS,GAAY;GAGvB,OAFIM,UAAyB,KAAA,IAEtBT,EAAQ,IAAIiC,GAASC,CAAU;EACxC;EACA,YAAY,GAAe;GACzB,KAAUC;EACZ;EACA,iBAAiB;GACf,KAAa;EACf;EACA,cAAc;GACZ,KAAU;EACZ;CACF;AACF"}
@@ -1 +1 @@
1
- {"version":3,"file":"_pool.cjs","names":[],"sources":["../src/_pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, SlotStrategy, WorkerPool } from './types.js';\n\nexport type { PoolOptions } from './_pool-core.js';\n\nexport function createPool<TInput, TOutput>(\n slots: readonly SlotStrategy<TInput, TOutput>[],\n options: PoolOptions,\n): WorkerPool<TInput, TOutput> {\n const core: PoolCore<SlotStrategy<TInput, TOutput>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function run(input: TInput, runOptions: RunOptions = {}): Promise<TOutput> {\n let priority: number;\n let timeout: number | undefined;\n let transferables: Transferable[];\n\n try {\n priority = validatePriority(runOptions.priority ?? 0);\n timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n transferables = [...(runOptions.transferables ?? [])];\n } catch (error) {\n return Promise.reject(error);\n }\n\n return core.acquire({ priority, signal: runOptions.signal }, (slot) => {\n core.trackActive(1);\n\n const onAbort = () => slot.cancel(abortError(runOptions.signal!));\n runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n const finish = async (): Promise<TOutput> => {\n try {\n const value = await slot.run(input, transferables, timeout);\n\n core.trackCompleted();\n\n return value;\n } catch (error) {\n if (!runOptions.signal?.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n };\n\n return finish();\n });\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n run,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n\nexport type BatchOptions<TInput> = Pick<RunOptions, 'priority' | 'signal' | 'timeout'> & {\n getTransferables?: (input: TInput, index: number) => readonly Transferable[];\n};\n\n/** Run related tasks concurrently and yield their results in input order. */\nexport function runBatch<TInput, TOutput>(\n pool: WorkerPool<TInput, TOutput>,\n inputs: readonly TInput[],\n options: BatchOptions<TInput> = {},\n): AsyncIterable<TOutput> {\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TOutput> {\n const tasks: Promise<TOutput>[] = [];\n const onAbort = () => controller.abort(options.signal?.reason);\n\n if (options.signal?.aborted) controller.abort(options.signal.reason);\n else options.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n for (const [index, input] of inputs.entries()) {\n const task = pool\n .run(input, {\n priority: options.priority,\n signal: controller.signal,\n timeout: options.timeout,\n transferables: options.getTransferables?.(input, index),\n })\n .catch((error: unknown) => {\n controller.abort(error);\n throw error;\n });\n\n void task.catch(() => undefined);\n tasks.push(task);\n }\n\n if (tasks.length === 0 && controller.signal.aborted) throw abortError(controller.signal);\n for (const task of tasks) {\n if (controller.signal.aborted) throw abortError(controller.signal);\n const value = await task;\n if (controller.signal.aborted) throw abortError(controller.signal);\n yield value;\n }\n } catch (error) {\n controller.abort(error);\n throw error;\n } finally {\n options.signal?.removeEventListener('abort', onAbort);\n controller.abort(new DOMException('Batch consumer stopped', 'AbortError'));\n await Promise.allSettled(tasks);\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TOutput> | undefined;\n const start = (): AsyncGenerator<TOutput> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker batches can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TOutput };\n controller.abort(new FamiliarTerminatedError('Batch consumer stopped'));\n return iterator?.return(value as TOutput) ?? { done: true, value: value as TOutput };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker batches can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n}\n"],"mappings":"iGAOA,SAAgB,WACd,EACA,EAC6B,CAC7B,IAAM,EAAgD,EAAA,eAAe,EAAO,CAAO,EAEnF,IAAK,IAAM,KAAQ,EAAO,EAAK,OAAU,GAAU,EAAK,QAAQ,CAAE,QAAO,KAAM,cAAe,CAAC,EAE/F,SAAS,IAAI,EAAe,EAAyB,CAAC,EAAqB,CACzE,IAAI,EACA,EACA,EAEJ,GAAI,CACF,EAAW,EAAA,iBAAiB,EAAW,UAAY,CAAC,EACpD,EAAU,EAAA,gBAAgB,EAAW,OAAO,GAAK,EAAQ,eACzD,EAAgB,CAAC,GAAI,EAAW,eAAiB,CAAC,CAAE,CACtD,OAAS,EAAO,CACd,OAAO,QAAQ,OAAO,CAAK,CAC7B,CAEA,OAAO,EAAK,QAAQ,CAAE,WAAU,OAAQ,EAAW,MAAO,EAAI,GAAS,CACrE,EAAK,YAAY,CAAC,EAElB,IAAM,YAAgB,EAAK,QAAA,EAAO,EAAA,WAAA,CAAW,EAAW,MAAO,CAAC,EAChE,EAAW,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEpE,IAAM,OAAS,SAA8B,CAC3C,GAAI,CACF,IAAM,EAAQ,MAAM,EAAK,IAAI,EAAO,EAAe,CAAO,EAI1D,OAFA,EAAK,eAAe,EAEb,CACT,OAAS,EAAO,CAGd,KAFI,CAAC,EAAW,QAAQ,SAAW,EAAE,aAAiB,EAAA,0BAA0B,EAAK,YAAY,EAE3F,CACR,QAAU,CACR,EAAW,QAAQ,oBAAoB,QAAS,OAAO,EACvD,EAAK,YAAY,EAAE,EACnB,EAAK,QAAQ,CAAI,EACjB,EAAK,WAAW,CAClB,CACF,EAEA,OAAO,OAAO,CAChB,CAAC,CACH,CAEA,MAAO,CACL,IAAI,gBAAiB,CACnB,OAAO,EAAK,cACd,EACA,QAAS,EAAK,QACd,IAAI,UAAW,CACb,OAAO,EAAK,QACd,EACA,MAAO,EAAK,MACZ,IACA,IAAI,OAAQ,CACV,OAAO,EAAK,MAAM,CACpB,EACA,IAAI,QAAS,CACX,OAAO,EAAK,OAAO,CACrB,EACA,IAAK,EAAK,KACT,OAAO,kBAAqB,EAAK,MAAM,CAAC,CAAC,GACzC,OAAO,SAAU,EAAK,OACzB,CACF,CAOA,SAAgB,SACd,EACA,EACA,EAAgC,CAAC,EACT,CACxB,IAAI,EAEE,QAAU,gBAAiB,EAAsD,CACrF,IAAM,EAA4B,CAAC,EAC7B,YAAgB,EAAW,MAAM,EAAQ,QAAQ,MAAM,EAEzD,EAAQ,QAAQ,QAAS,EAAW,MAAM,EAAQ,OAAO,MAAM,EAC9D,EAAQ,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEtE,GAAI,CACF,IAAK,GAAM,CAAC,EAAO,KAAU,EAAO,QAAQ,EAAG,CAC7C,IAAM,EAAO,EACV,IAAI,EAAO,CACV,SAAU,EAAQ,SAClB,OAAQ,EAAW,OACnB,QAAS,EAAQ,QACjB,cAAe,EAAQ,mBAAmB,EAAO,CAAK,CACxD,CAAC,CAAC,CACD,MAAO,GAAmB,CAEzB,MADA,EAAW,MAAM,CAAK,EAChB,CACR,CAAC,EAEH,EAAU,UAAY,IAAA,EAAS,EAC/B,EAAM,KAAK,CAAI,CACjB,CAEA,GAAI,EAAM,SAAW,GAAK,EAAW,OAAO,QAAS,MAAA,EAAM,EAAA,WAAA,CAAW,EAAW,MAAM,EACvF,IAAK,IAAM,KAAQ,EAAO,CACxB,GAAI,EAAW,OAAO,QAAS,MAAA,EAAM,EAAA,WAAA,CAAW,EAAW,MAAM,EACjE,IAAM,EAAQ,MAAM,EACpB,GAAI,EAAW,OAAO,QAAS,MAAA,EAAM,EAAA,WAAA,CAAW,EAAW,MAAM,EACjE,MAAM,CACR,CACF,OAAS,EAAO,CAEd,MADA,EAAW,MAAM,CAAK,EAChB,CACR,QAAU,CACR,EAAQ,QAAQ,oBAAoB,QAAS,OAAO,EACpD,EAAW,MAAM,IAAI,aAAa,yBAA0B,YAAY,CAAC,EACzE,MAAM,QAAQ,WAAW,CAAK,CAChC,CACF,EAEA,MAAO,CACL,CAAC,OAAO,gBAAiB,CACvB,IAAM,EAAQ,CAAC,EACT,EAAa,IAAI,gBACnB,EACE,UAAmD,CACvD,OAAU,EACN,IAAU,EAEd,MADA,KAAa,QAAQ,CAAU,EACxB,CACT,EAEA,MAAO,CACL,SACE,MAAM,CAAC,EAAE,KAAK,GAAK,QAAQ,OAAO,IAAI,EAAA,qBAAqB,0CAA0C,CAAC,EACxG,OAAQ,KAAO,IACT,CAAC,GAAS,IAAU,EAAc,CAAE,KAAM,GAAa,OAAiB,GAC5E,EAAW,MAAM,IAAI,EAAA,wBAAwB,wBAAwB,CAAC,EAC/D,GAAU,OAAO,CAAgB,GAAK,CAAE,KAAM,GAAa,OAAiB,GAErF,MAAO,KAAO,IAAoB,CAChC,IAAM,EAAS,MAAM,EACrB,GAAI,CAAC,EAAQ,MAAM,IAAI,EAAA,qBAAqB,0CAA0C,EAEtF,OADA,EAAW,MAAM,CAAK,EACf,EAAO,MAAM,CAAK,CAC3B,CACF,CACF,CACF,CACF"}
1
+ {"version":3,"file":"_pool.cjs","names":["createPoolCore","validatePriority","validateTimeout","abortError","FamiliarTerminatedError","FamiliarRuntimeError","require_errors","require__pool_core","_vielzeug_arsenal","core","slots","options","slot","priority","timeout","transferables","runOptions","error","value","input","owner","tasks","controller","index","inputs","task","pool","token","iterator","active"],"sources":["../src/_pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, SlotStrategy, WorkerPool } from './types.js';\n\nexport type { PoolOptions } from './_pool-core.js';\n\nexport function createPool<TInput, TOutput>(\n slots: readonly SlotStrategy<TInput, TOutput>[],\n options: PoolOptions,\n): WorkerPool<TInput, TOutput> {\n const core: PoolCore<SlotStrategy<TInput, TOutput>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function run(input: TInput, runOptions: RunOptions = {}): Promise<TOutput> {\n let priority: number;\n let timeout: number | undefined;\n let transferables: Transferable[];\n\n try {\n priority = validatePriority(runOptions.priority ?? 0);\n timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n transferables = [...(runOptions.transferables ?? [])];\n } catch (error) {\n return Promise.reject(error);\n }\n\n return core.acquire({ priority, signal: runOptions.signal }, (slot) => {\n core.trackActive(1);\n\n const onAbort = () => slot.cancel(abortError(runOptions.signal!));\n runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n const finish = async (): Promise<TOutput> => {\n try {\n const value = await slot.run(input, transferables, timeout);\n\n core.trackCompleted();\n\n return value;\n } catch (error) {\n if (!runOptions.signal?.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n };\n\n return finish();\n });\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n run,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n\nexport type BatchOptions<TInput> = Pick<RunOptions, 'priority' | 'signal' | 'timeout'> & {\n getTransferables?: (input: TInput, index: number) => readonly Transferable[];\n};\n\n/** Run related tasks concurrently and yield their results in input order. */\nexport function runBatch<TInput, TOutput>(\n pool: WorkerPool<TInput, TOutput>,\n inputs: readonly TInput[],\n options: BatchOptions<TInput> = {},\n): AsyncIterable<TOutput> {\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TOutput> {\n const tasks: Promise<TOutput>[] = [];\n const onAbort = () => controller.abort(options.signal?.reason);\n\n if (options.signal?.aborted) controller.abort(options.signal.reason);\n else options.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n for (const [index, input] of inputs.entries()) {\n const task = pool\n .run(input, {\n priority: options.priority,\n signal: controller.signal,\n timeout: options.timeout,\n transferables: options.getTransferables?.(input, index),\n })\n .catch((error: unknown) => {\n controller.abort(error);\n throw error;\n });\n\n void task.catch(() => undefined);\n tasks.push(task);\n }\n\n if (tasks.length === 0 && controller.signal.aborted) throw abortError(controller.signal);\n for (const task of tasks) {\n if (controller.signal.aborted) throw abortError(controller.signal);\n const value = await task;\n if (controller.signal.aborted) throw abortError(controller.signal);\n yield value;\n }\n } catch (error) {\n controller.abort(error);\n throw error;\n } finally {\n options.signal?.removeEventListener('abort', onAbort);\n controller.abort(new DOMException('Batch consumer stopped', 'AbortError'));\n await Promise.allSettled(tasks);\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TOutput> | undefined;\n const start = (): AsyncGenerator<TOutput> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker batches can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TOutput };\n controller.abort(new FamiliarTerminatedError('Batch consumer stopped'));\n return iterator?.return(value as TOutput) ?? { done: true, value: value as TOutput };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker batches can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n}\n"],"mappings":"A,iGAOA,SAAgB,WACd,EACA,EAC6B,CAC7B,IAAMS,EAAgDT,EAAAA,eAAeU,EAAOC,CAAO,EAEnF,IAAK,IAAMC,KAAQF,EAAO,EAAK,OAAU,GAAUD,EAAK,QAAQ,CAAE,QAAO,KAAM,cAAe,CAAC,EAE/F,SAAS,IAAI,EAAe,EAAyB,CAAC,EAAqB,CACzE,IAAII,EACAC,EACAC,EAEJ,GAAI,CACF,EAAWd,EAAAA,iBAAiBe,EAAW,UAAY,CAAC,EACpD,EAAUd,EAAAA,gBAAgBc,EAAW,OAAO,GAAKL,EAAQ,eACzD,EAAgB,CAAC,GAAIK,EAAW,eAAiB,CAAC,CAAE,CACtD,OAASC,EAAO,CACd,OAAO,QAAQ,OAAOA,CAAK,CAC7B,CAEA,OAAOR,EAAK,QAAQ,CAAE,WAAU,OAAQO,EAAW,MAAO,EAAI,GAAS,CACrEP,EAAK,YAAY,CAAC,EAElB,IAAM,YAAgBG,EAAK,QAAA,EAAOT,EAAAA,WAAAA,CAAWa,EAAW,MAAO,CAAC,EAChEA,EAAW,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEpE,IAAM,OAAS,SAA8B,CAC3C,GAAI,CACF,IAAME,EAAQ,MAAMN,EAAK,IAAIO,EAAOJ,EAAeD,CAAO,EAI1D,OAFAL,EAAK,eAAe,EAEbS,CACT,OAASD,EAAO,CAGd,KAFI,CAACD,EAAW,QAAQ,SAAW,EAAEC,aAAiBb,EAAAA,0BAA0BK,EAAK,YAAY,EAE3FQ,CACR,QAAU,CACRD,EAAW,QAAQ,oBAAoB,QAAS,OAAO,EACvDP,EAAK,YAAY,EAAE,EACnBA,EAAK,QAAQG,CAAI,EACjBH,EAAK,WAAW,CAClB,CACF,EAEA,OAAO,OAAO,CAChB,CAAC,CACH,CAEA,MAAO,CACL,IAAI,gBAAiB,CACnB,OAAOA,EAAK,cACd,EACA,QAASA,EAAK,QACd,IAAI,UAAW,CACb,OAAOA,EAAK,QACd,EACA,MAAOA,EAAK,MACZ,IACA,IAAI,OAAQ,CACV,OAAOA,EAAK,MAAM,CACpB,EACA,IAAI,QAAS,CACX,OAAOA,EAAK,OAAO,CACrB,EACA,IAAKA,EAAK,KACT,OAAO,kBAAqBA,EAAK,MAAM,CAAC,CAAC,GACzC,OAAO,SAAUA,EAAK,OACzB,CACF,CAOA,SAAgB,SACd,EACA,EACA,EAAgC,CAAC,EACT,CACxB,IAAIW,EAEE,QAAU,gBAAiB,EAAsD,CACrF,IAAMC,EAA4B,CAAC,EAC7B,YAAgBC,EAAW,MAAMX,EAAQ,QAAQ,MAAM,EAEzDA,EAAQ,QAAQ,QAASW,EAAW,MAAMX,EAAQ,OAAO,MAAM,EAC9DA,EAAQ,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEtE,GAAI,CACF,IAAK,GAAM,CAACY,EAAOJ,KAAUK,EAAO,QAAQ,EAAG,CAC7C,IAAMC,EAAOC,EACV,IAAIP,EAAO,CACV,SAAUR,EAAQ,SAClB,OAAQW,EAAW,OACnB,QAASX,EAAQ,QACjB,cAAeA,EAAQ,mBAAmBQ,EAAOI,CAAK,CACxD,CAAC,CAAC,CACD,MAAO,GAAmB,CAEzB,MADAD,EAAW,MAAML,CAAK,EAChBA,CACR,CAAC,EAEHQ,EAAU,UAAY,IAAA,EAAS,EAC/BJ,EAAM,KAAKI,CAAI,CACjB,CAEA,GAAIJ,EAAM,SAAW,GAAKC,EAAW,OAAO,QAAS,MAAA,EAAMnB,EAAAA,WAAAA,CAAWmB,EAAW,MAAM,EACvF,IAAK,IAAMG,KAAQJ,EAAO,CACxB,GAAIC,EAAW,OAAO,QAAS,MAAA,EAAMnB,EAAAA,WAAAA,CAAWmB,EAAW,MAAM,EACjE,IAAMJ,EAAQ,MAAMO,EACpB,GAAIH,EAAW,OAAO,QAAS,MAAA,EAAMnB,EAAAA,WAAAA,CAAWmB,EAAW,MAAM,EACjE,MAAMJ,CACR,CACF,OAASD,EAAO,CAEd,MADAK,EAAW,MAAML,CAAK,EAChBA,CACR,QAAU,CACRN,EAAQ,QAAQ,oBAAoB,QAAS,OAAO,EACpDW,EAAW,MAAM,IAAI,aAAa,yBAA0B,YAAY,CAAC,EACzE,MAAM,QAAQ,WAAWD,CAAK,CAChC,CACF,EAEA,MAAO,CACL,CAAC,OAAO,gBAAiB,CACvB,IAAMM,EAAQ,CAAC,EACTL,EAAa,IAAI,gBACnBM,EACE,UAAmD,CAEvD,GADA,IAAUD,EACNP,IAAUO,EAEd,MADA,KAAa,QAAQL,CAAU,EACxBM,CACT,EAEA,MAAO,CACL,SACE,MAAM,CAAC,EAAE,KAAK,GAAK,QAAQ,OAAO,IAAIvB,EAAAA,qBAAqB,0CAA0C,CAAC,EACxG,OAAQ,KAAO,IACT,CAACe,GAASA,IAAUO,EAAc,CAAE,KAAM,GAAa,OAAiB,GAC5EL,EAAW,MAAM,IAAIlB,EAAAA,wBAAwB,wBAAwB,CAAC,EAC/DwB,GAAU,OAAOV,CAAgB,GAAK,CAAE,KAAM,GAAa,OAAiB,GAErF,MAAO,KAAO,IAAoB,CAChC,IAAMW,EAAS,MAAM,EACrB,GAAI,CAACA,EAAQ,MAAM,IAAIxB,EAAAA,qBAAqB,0CAA0C,EAEtF,OADAiB,EAAW,MAAML,CAAK,EACfY,EAAO,MAAMZ,CAAK,CAC3B,CACF,CACF,CACF,CACF,C"}
package/dist/_pool.js CHANGED
@@ -1,2 +1,112 @@
1
- import{FamiliarRuntimeError as e,FamiliarTerminatedError as t}from"./errors.js";import{createPoolCore as n,validatePriority as r,validateTimeout as i}from"./_pool-core.js";import{abortError as a}from"@vielzeug/arsenal";function createPool(e,o){let s=n(e,o);for(let t of e)t.notify=e=>s.emitTap({error:e,type:`worker-error`});function run(e,n={}){let c,l,u;try{c=r(n.priority??0),l=i(n.timeout)??o.defaultTimeout,u=[...n.transferables??[]]}catch(e){return Promise.reject(e)}return s.acquire({priority:c,signal:n.signal},r=>{s.trackActive(1);let onAbort=()=>r.cancel(a(n.signal));n.signal?.addEventListener(`abort`,onAbort,{once:!0});let finish=async()=>{try{let t=await r.run(e,u,l);return s.trackCompleted(),t}catch(e){throw!n.signal?.aborted&&!(e instanceof t)&&s.trackFailed(),e}finally{n.signal?.removeEventListener(`abort`,onAbort),s.trackActive(-1),s.release(r),s.settleIdle()}};return finish()})}return{get disposalSignal(){return s.disposalSignal},dispose:s.dispose,get disposed(){return s.disposed},drain:s.drain,run,get stats(){return s.stats()},get status(){return s.status()},tap:s.tap,[Symbol.asyncDispose]:()=>s.drain({}),[Symbol.dispose]:s.dispose}}function runBatch(n,r,i={}){let o,iterate=async function*(e){let t=[],onAbort=()=>e.abort(i.signal?.reason);i.signal?.aborted?e.abort(i.signal.reason):i.signal?.addEventListener(`abort`,onAbort,{once:!0});try{for(let[a,o]of r.entries()){let r=n.run(o,{priority:i.priority,signal:e.signal,timeout:i.timeout,transferables:i.getTransferables?.(o,a)}).catch(t=>{throw e.abort(t),t});r.catch(()=>void 0),t.push(r)}if(t.length===0&&e.signal.aborted)throw a(e.signal);for(let n of t){if(e.signal.aborted)throw a(e.signal);let t=await n;if(e.signal.aborted)throw a(e.signal);yield t}}catch(t){throw e.abort(t),t}finally{i.signal?.removeEventListener(`abort`,onAbort),e.abort(new DOMException(`Batch consumer stopped`,`AbortError`)),await Promise.allSettled(t)}};return{[Symbol.asyncIterator](){let n={},r=new AbortController,i,start=()=>{if(o??=n,o===n)return i??=iterate(r),i};return{next:()=>start()?.next()??Promise.reject(new e(`Worker batches can only be consumed once`)),return:async e=>!o||o!==n?{done:!0,value:e}:(r.abort(new t(`Batch consumer stopped`)),i?.return(e)??{done:!0,value:e}),throw:async t=>{let n=start();if(!n)throw new e(`Worker batches can only be consumed once`);return r.abort(t),n.throw(t)}}}}}export{createPool,runBatch};
1
+ import { FamiliarRuntimeError as e, FamiliarTerminatedError as t } from "./errors.js";
2
+ import { createPoolCore as n, validatePriority as r, validateTimeout as i } from "./_pool-core.js";
3
+ import { abortError as a } from "@vielzeug/arsenal";
4
+ //#region src/_pool.ts
5
+ function createPool(e, o) {
6
+ let s = n(e, o);
7
+ for (let t of e) t.notify = (e) => s.emitTap({
8
+ error: e,
9
+ type: "worker-error"
10
+ });
11
+ function run(e, n = {}) {
12
+ let c, l, u;
13
+ try {
14
+ c = r(n.priority ?? 0), l = i(n.timeout) ?? o.defaultTimeout, u = [...n.transferables ?? []];
15
+ } catch (e) {
16
+ return Promise.reject(e);
17
+ }
18
+ return s.acquire({
19
+ priority: c,
20
+ signal: n.signal
21
+ }, (r) => {
22
+ s.trackActive(1);
23
+ let onAbort = () => r.cancel(a(n.signal));
24
+ n.signal?.addEventListener("abort", onAbort, { once: !0 });
25
+ let finish = async () => {
26
+ try {
27
+ let t = await r.run(e, u, l);
28
+ return s.trackCompleted(), t;
29
+ } catch (e) {
30
+ throw !n.signal?.aborted && !(e instanceof t) && s.trackFailed(), e;
31
+ } finally {
32
+ n.signal?.removeEventListener("abort", onAbort), s.trackActive(-1), s.release(r), s.settleIdle();
33
+ }
34
+ };
35
+ return finish();
36
+ });
37
+ }
38
+ return {
39
+ get disposalSignal() {
40
+ return s.disposalSignal;
41
+ },
42
+ dispose: s.dispose,
43
+ get disposed() {
44
+ return s.disposed;
45
+ },
46
+ drain: s.drain,
47
+ run,
48
+ get stats() {
49
+ return s.stats();
50
+ },
51
+ get status() {
52
+ return s.status();
53
+ },
54
+ tap: s.tap,
55
+ [Symbol.asyncDispose]: () => s.drain({}),
56
+ [Symbol.dispose]: s.dispose
57
+ };
58
+ }
59
+ function runBatch(n, r, i = {}) {
60
+ let o, iterate = async function* (e) {
61
+ let t = [], onAbort = () => e.abort(i.signal?.reason);
62
+ i.signal?.aborted ? e.abort(i.signal.reason) : i.signal?.addEventListener("abort", onAbort, { once: !0 });
63
+ try {
64
+ for (let [a, o] of r.entries()) {
65
+ let r = n.run(o, {
66
+ priority: i.priority,
67
+ signal: e.signal,
68
+ timeout: i.timeout,
69
+ transferables: i.getTransferables?.(o, a)
70
+ }).catch((t) => {
71
+ throw e.abort(t), t;
72
+ });
73
+ r.catch(() => void 0), t.push(r);
74
+ }
75
+ if (t.length === 0 && e.signal.aborted) throw a(e.signal);
76
+ for (let n of t) {
77
+ if (e.signal.aborted) throw a(e.signal);
78
+ let t = await n;
79
+ if (e.signal.aborted) throw a(e.signal);
80
+ yield t;
81
+ }
82
+ } catch (t) {
83
+ throw e.abort(t), t;
84
+ } finally {
85
+ i.signal?.removeEventListener("abort", onAbort), e.abort(new DOMException("Batch consumer stopped", "AbortError")), await Promise.allSettled(t);
86
+ }
87
+ };
88
+ return { [Symbol.asyncIterator]() {
89
+ let n = {}, r = new AbortController(), i, start = () => {
90
+ if (o ??= n, o === n) return i ??= iterate(r), i;
91
+ };
92
+ return {
93
+ next: () => start()?.next() ?? Promise.reject(new e("Worker batches can only be consumed once")),
94
+ return: async (e) => !o || o !== n ? {
95
+ done: !0,
96
+ value: e
97
+ } : (r.abort(new t("Batch consumer stopped")), i?.return(e) ?? {
98
+ done: !0,
99
+ value: e
100
+ }),
101
+ throw: async (t) => {
102
+ let n = start();
103
+ if (!n) throw new e("Worker batches can only be consumed once");
104
+ return r.abort(t), n.throw(t);
105
+ }
106
+ };
107
+ } };
108
+ }
109
+ //#endregion
110
+ export { createPool, runBatch };
111
+
2
112
  //# sourceMappingURL=_pool.js.map
package/dist/_pool.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"_pool.js","names":[],"sources":["../src/_pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, SlotStrategy, WorkerPool } from './types.js';\n\nexport type { PoolOptions } from './_pool-core.js';\n\nexport function createPool<TInput, TOutput>(\n slots: readonly SlotStrategy<TInput, TOutput>[],\n options: PoolOptions,\n): WorkerPool<TInput, TOutput> {\n const core: PoolCore<SlotStrategy<TInput, TOutput>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function run(input: TInput, runOptions: RunOptions = {}): Promise<TOutput> {\n let priority: number;\n let timeout: number | undefined;\n let transferables: Transferable[];\n\n try {\n priority = validatePriority(runOptions.priority ?? 0);\n timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n transferables = [...(runOptions.transferables ?? [])];\n } catch (error) {\n return Promise.reject(error);\n }\n\n return core.acquire({ priority, signal: runOptions.signal }, (slot) => {\n core.trackActive(1);\n\n const onAbort = () => slot.cancel(abortError(runOptions.signal!));\n runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n const finish = async (): Promise<TOutput> => {\n try {\n const value = await slot.run(input, transferables, timeout);\n\n core.trackCompleted();\n\n return value;\n } catch (error) {\n if (!runOptions.signal?.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n };\n\n return finish();\n });\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n run,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n\nexport type BatchOptions<TInput> = Pick<RunOptions, 'priority' | 'signal' | 'timeout'> & {\n getTransferables?: (input: TInput, index: number) => readonly Transferable[];\n};\n\n/** Run related tasks concurrently and yield their results in input order. */\nexport function runBatch<TInput, TOutput>(\n pool: WorkerPool<TInput, TOutput>,\n inputs: readonly TInput[],\n options: BatchOptions<TInput> = {},\n): AsyncIterable<TOutput> {\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TOutput> {\n const tasks: Promise<TOutput>[] = [];\n const onAbort = () => controller.abort(options.signal?.reason);\n\n if (options.signal?.aborted) controller.abort(options.signal.reason);\n else options.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n for (const [index, input] of inputs.entries()) {\n const task = pool\n .run(input, {\n priority: options.priority,\n signal: controller.signal,\n timeout: options.timeout,\n transferables: options.getTransferables?.(input, index),\n })\n .catch((error: unknown) => {\n controller.abort(error);\n throw error;\n });\n\n void task.catch(() => undefined);\n tasks.push(task);\n }\n\n if (tasks.length === 0 && controller.signal.aborted) throw abortError(controller.signal);\n for (const task of tasks) {\n if (controller.signal.aborted) throw abortError(controller.signal);\n const value = await task;\n if (controller.signal.aborted) throw abortError(controller.signal);\n yield value;\n }\n } catch (error) {\n controller.abort(error);\n throw error;\n } finally {\n options.signal?.removeEventListener('abort', onAbort);\n controller.abort(new DOMException('Batch consumer stopped', 'AbortError'));\n await Promise.allSettled(tasks);\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TOutput> | undefined;\n const start = (): AsyncGenerator<TOutput> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker batches can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TOutput };\n controller.abort(new FamiliarTerminatedError('Batch consumer stopped'));\n return iterator?.return(value as TOutput) ?? { done: true, value: value as TOutput };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker batches can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n}\n"],"mappings":"2NAOA,SAAgB,WACd,EACA,EAC6B,CAC7B,IAAM,EAAgD,EAAe,EAAO,CAAO,EAEnF,IAAK,IAAM,KAAQ,EAAO,EAAK,OAAU,GAAU,EAAK,QAAQ,CAAE,QAAO,KAAM,cAAe,CAAC,EAE/F,SAAS,IAAI,EAAe,EAAyB,CAAC,EAAqB,CACzE,IAAI,EACA,EACA,EAEJ,GAAI,CACF,EAAW,EAAiB,EAAW,UAAY,CAAC,EACpD,EAAU,EAAgB,EAAW,OAAO,GAAK,EAAQ,eACzD,EAAgB,CAAC,GAAI,EAAW,eAAiB,CAAC,CAAE,CACtD,OAAS,EAAO,CACd,OAAO,QAAQ,OAAO,CAAK,CAC7B,CAEA,OAAO,EAAK,QAAQ,CAAE,WAAU,OAAQ,EAAW,MAAO,EAAI,GAAS,CACrE,EAAK,YAAY,CAAC,EAElB,IAAM,YAAgB,EAAK,OAAO,EAAW,EAAW,MAAO,CAAC,EAChE,EAAW,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEpE,IAAM,OAAS,SAA8B,CAC3C,GAAI,CACF,IAAM,EAAQ,MAAM,EAAK,IAAI,EAAO,EAAe,CAAO,EAI1D,OAFA,EAAK,eAAe,EAEb,CACT,OAAS,EAAO,CAGd,KAFI,CAAC,EAAW,QAAQ,SAAW,EAAE,aAAiB,IAA0B,EAAK,YAAY,EAE3F,CACR,QAAU,CACR,EAAW,QAAQ,oBAAoB,QAAS,OAAO,EACvD,EAAK,YAAY,EAAE,EACnB,EAAK,QAAQ,CAAI,EACjB,EAAK,WAAW,CAClB,CACF,EAEA,OAAO,OAAO,CAChB,CAAC,CACH,CAEA,MAAO,CACL,IAAI,gBAAiB,CACnB,OAAO,EAAK,cACd,EACA,QAAS,EAAK,QACd,IAAI,UAAW,CACb,OAAO,EAAK,QACd,EACA,MAAO,EAAK,MACZ,IACA,IAAI,OAAQ,CACV,OAAO,EAAK,MAAM,CACpB,EACA,IAAI,QAAS,CACX,OAAO,EAAK,OAAO,CACrB,EACA,IAAK,EAAK,KACT,OAAO,kBAAqB,EAAK,MAAM,CAAC,CAAC,GACzC,OAAO,SAAU,EAAK,OACzB,CACF,CAOA,SAAgB,SACd,EACA,EACA,EAAgC,CAAC,EACT,CACxB,IAAI,EAEE,QAAU,gBAAiB,EAAsD,CACrF,IAAM,EAA4B,CAAC,EAC7B,YAAgB,EAAW,MAAM,EAAQ,QAAQ,MAAM,EAEzD,EAAQ,QAAQ,QAAS,EAAW,MAAM,EAAQ,OAAO,MAAM,EAC9D,EAAQ,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEtE,GAAI,CACF,IAAK,GAAM,CAAC,EAAO,KAAU,EAAO,QAAQ,EAAG,CAC7C,IAAM,EAAO,EACV,IAAI,EAAO,CACV,SAAU,EAAQ,SAClB,OAAQ,EAAW,OACnB,QAAS,EAAQ,QACjB,cAAe,EAAQ,mBAAmB,EAAO,CAAK,CACxD,CAAC,CAAC,CACD,MAAO,GAAmB,CAEzB,MADA,EAAW,MAAM,CAAK,EAChB,CACR,CAAC,EAEH,EAAU,UAAY,IAAA,EAAS,EAC/B,EAAM,KAAK,CAAI,CACjB,CAEA,GAAI,EAAM,SAAW,GAAK,EAAW,OAAO,QAAS,MAAM,EAAW,EAAW,MAAM,EACvF,IAAK,IAAM,KAAQ,EAAO,CACxB,GAAI,EAAW,OAAO,QAAS,MAAM,EAAW,EAAW,MAAM,EACjE,IAAM,EAAQ,MAAM,EACpB,GAAI,EAAW,OAAO,QAAS,MAAM,EAAW,EAAW,MAAM,EACjE,MAAM,CACR,CACF,OAAS,EAAO,CAEd,MADA,EAAW,MAAM,CAAK,EAChB,CACR,QAAU,CACR,EAAQ,QAAQ,oBAAoB,QAAS,OAAO,EACpD,EAAW,MAAM,IAAI,aAAa,yBAA0B,YAAY,CAAC,EACzE,MAAM,QAAQ,WAAW,CAAK,CAChC,CACF,EAEA,MAAO,CACL,CAAC,OAAO,gBAAiB,CACvB,IAAM,EAAQ,CAAC,EACT,EAAa,IAAI,gBACnB,EACE,UAAmD,CACvD,OAAU,EACN,IAAU,EAEd,MADA,KAAa,QAAQ,CAAU,EACxB,CACT,EAEA,MAAO,CACL,SACE,MAAM,CAAC,EAAE,KAAK,GAAK,QAAQ,OAAO,IAAI,EAAqB,0CAA0C,CAAC,EACxG,OAAQ,KAAO,IACT,CAAC,GAAS,IAAU,EAAc,CAAE,KAAM,GAAa,OAAiB,GAC5E,EAAW,MAAM,IAAI,EAAwB,wBAAwB,CAAC,EAC/D,GAAU,OAAO,CAAgB,GAAK,CAAE,KAAM,GAAa,OAAiB,GAErF,MAAO,KAAO,IAAoB,CAChC,IAAM,EAAS,MAAM,EACrB,GAAI,CAAC,EAAQ,MAAM,IAAI,EAAqB,0CAA0C,EAEtF,OADA,EAAW,MAAM,CAAK,EACf,EAAO,MAAM,CAAK,CAC3B,CACF,CACF,CACF,CACF"}
1
+ {"version":3,"file":"_pool.js","names":["core","createPoolCore","slots","options","slot","priority","timeout","transferables","validatePriority","runOptions","validateTimeout","error","abortError","value","input","FamiliarTerminatedError","owner","tasks","controller","index","inputs","task","pool","token","iterator","FamiliarRuntimeError","active"],"sources":["../src/_pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, SlotStrategy, WorkerPool } from './types.js';\n\nexport type { PoolOptions } from './_pool-core.js';\n\nexport function createPool<TInput, TOutput>(\n slots: readonly SlotStrategy<TInput, TOutput>[],\n options: PoolOptions,\n): WorkerPool<TInput, TOutput> {\n const core: PoolCore<SlotStrategy<TInput, TOutput>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function run(input: TInput, runOptions: RunOptions = {}): Promise<TOutput> {\n let priority: number;\n let timeout: number | undefined;\n let transferables: Transferable[];\n\n try {\n priority = validatePriority(runOptions.priority ?? 0);\n timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n transferables = [...(runOptions.transferables ?? [])];\n } catch (error) {\n return Promise.reject(error);\n }\n\n return core.acquire({ priority, signal: runOptions.signal }, (slot) => {\n core.trackActive(1);\n\n const onAbort = () => slot.cancel(abortError(runOptions.signal!));\n runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n const finish = async (): Promise<TOutput> => {\n try {\n const value = await slot.run(input, transferables, timeout);\n\n core.trackCompleted();\n\n return value;\n } catch (error) {\n if (!runOptions.signal?.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n };\n\n return finish();\n });\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n run,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n\nexport type BatchOptions<TInput> = Pick<RunOptions, 'priority' | 'signal' | 'timeout'> & {\n getTransferables?: (input: TInput, index: number) => readonly Transferable[];\n};\n\n/** Run related tasks concurrently and yield their results in input order. */\nexport function runBatch<TInput, TOutput>(\n pool: WorkerPool<TInput, TOutput>,\n inputs: readonly TInput[],\n options: BatchOptions<TInput> = {},\n): AsyncIterable<TOutput> {\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TOutput> {\n const tasks: Promise<TOutput>[] = [];\n const onAbort = () => controller.abort(options.signal?.reason);\n\n if (options.signal?.aborted) controller.abort(options.signal.reason);\n else options.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n for (const [index, input] of inputs.entries()) {\n const task = pool\n .run(input, {\n priority: options.priority,\n signal: controller.signal,\n timeout: options.timeout,\n transferables: options.getTransferables?.(input, index),\n })\n .catch((error: unknown) => {\n controller.abort(error);\n throw error;\n });\n\n void task.catch(() => undefined);\n tasks.push(task);\n }\n\n if (tasks.length === 0 && controller.signal.aborted) throw abortError(controller.signal);\n for (const task of tasks) {\n if (controller.signal.aborted) throw abortError(controller.signal);\n const value = await task;\n if (controller.signal.aborted) throw abortError(controller.signal);\n yield value;\n }\n } catch (error) {\n controller.abort(error);\n throw error;\n } finally {\n options.signal?.removeEventListener('abort', onAbort);\n controller.abort(new DOMException('Batch consumer stopped', 'AbortError'));\n await Promise.allSettled(tasks);\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TOutput> | undefined;\n const start = (): AsyncGenerator<TOutput> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker batches can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TOutput };\n controller.abort(new FamiliarTerminatedError('Batch consumer stopped'));\n return iterator?.return(value as TOutput) ?? { done: true, value: value as TOutput };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker batches can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n}\n"],"mappings":";;;;AAOA,SAAgB,WACd,GACA,GAC6B;CAC7B,IAAMA,IAAgDC,EAAeC,GAAOC,CAAO;CAEnF,KAAK,IAAMC,KAAQF,GAAO,EAAK,UAAU,MAAUF,EAAK,QAAQ;EAAE;EAAO,MAAM;CAAe,CAAC;CAE/F,SAAS,IAAI,GAAe,IAAyB,CAAC,GAAqB;EACzE,IAAIK,GACAC,GACAC;EAEJ,IAAI;GAGF,AAFA,IAAWC,EAAiBC,EAAW,YAAY,CAAC,GACpD,IAAUC,EAAgBD,EAAW,OAAO,KAAKN,EAAQ,gBACzD,IAAgB,CAAC,GAAIM,EAAW,iBAAiB,CAAC,CAAE;EACtD,SAASE,GAAO;GACd,OAAO,QAAQ,OAAOA,CAAK;EAC7B;EAEA,OAAOX,EAAK,QAAQ;GAAE;GAAU,QAAQS,EAAW;EAAO,IAAI,MAAS;GACrE,EAAK,YAAY,CAAC;GAElB,IAAM,gBAAgBL,EAAK,OAAOQ,EAAWH,EAAW,MAAO,CAAC;GAChE,EAAW,QAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,GAAK,CAAC;GAEpE,IAAM,SAAS,YAA8B;IAC3C,IAAI;KACF,IAAMI,IAAQ,MAAMT,EAAK,IAAIU,GAAOP,GAAeD,CAAO;KAI1D,OAFAN,EAAK,eAAe,GAEba;IACT,SAASF,GAAO;KAGd,MAFI,CAACF,EAAW,QAAQ,WAAW,EAAEE,aAAiBI,MAA0Bf,EAAK,YAAY,GAE3FW;IACR,UAAU;KAIR,AAHAF,EAAW,QAAQ,oBAAoB,SAAS,OAAO,GACvDT,EAAK,YAAY,EAAE,GACnBA,EAAK,QAAQI,CAAI,GACjBJ,EAAK,WAAW;IAClB;GACF;GAEA,OAAO,OAAO;EAChB,CAAC;CACH;CAEA,OAAO;EACL,IAAI,iBAAiB;GACnB,OAAOA,EAAK;EACd;EACA,SAASA,EAAK;EACd,IAAI,WAAW;GACb,OAAOA,EAAK;EACd;EACA,OAAOA,EAAK;EACZ;EACA,IAAI,QAAQ;GACV,OAAOA,EAAK,MAAM;EACpB;EACA,IAAI,SAAS;GACX,OAAOA,EAAK,OAAO;EACrB;EACA,KAAKA,EAAK;GACT,OAAO,qBAAqBA,EAAK,MAAM,CAAC,CAAC;GACzC,OAAO,UAAUA,EAAK;CACzB;AACF;AAOA,SAAgB,SACd,GACA,GACA,IAAgC,CAAC,GACT;CACxB,IAAIgB,GAEE,UAAU,iBAAiB,GAAsD;EACrF,IAAMC,IAA4B,CAAC,GAC7B,gBAAgBC,EAAW,MAAMf,EAAQ,QAAQ,MAAM;EAE7D,AAAIA,EAAQ,QAAQ,UAASe,EAAW,MAAMf,EAAQ,OAAO,MAAM,IAC9DA,EAAQ,QAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,GAAK,CAAC;EAEtE,IAAI;GACF,KAAK,IAAM,CAACgB,GAAOL,MAAUM,EAAO,QAAQ,GAAG;IAC7C,IAAMC,IAAOC,EACV,IAAIR,GAAO;KACV,UAAUX,EAAQ;KAClB,QAAQe,EAAW;KACnB,SAASf,EAAQ;KACjB,eAAeA,EAAQ,mBAAmBW,GAAOK,CAAK;IACxD,CAAC,CAAC,CACD,OAAO,MAAmB;KAEzB,MADAD,EAAW,MAAMP,CAAK,GAChBA;IACR,CAAC;IAGH,AADAU,EAAU,YAAY,KAAA,CAAS,GAC/BJ,EAAM,KAAKI,CAAI;GACjB;GAEA,IAAIJ,EAAM,WAAW,KAAKC,EAAW,OAAO,SAAS,MAAMN,EAAWM,EAAW,MAAM;GACvF,KAAK,IAAMG,KAAQJ,GAAO;IACxB,IAAIC,EAAW,OAAO,SAAS,MAAMN,EAAWM,EAAW,MAAM;IACjE,IAAML,IAAQ,MAAMQ;IACpB,IAAIH,EAAW,OAAO,SAAS,MAAMN,EAAWM,EAAW,MAAM;IACjE,MAAML;GACR;EACF,SAASF,GAAO;GAEd,MADAO,EAAW,MAAMP,CAAK,GAChBA;EACR,UAAU;GAGR,AAFAR,EAAQ,QAAQ,oBAAoB,SAAS,OAAO,GACpDe,EAAW,MAAM,IAAI,aAAa,0BAA0B,YAAY,CAAC,GACzE,MAAM,QAAQ,WAAWD,CAAK;EAChC;CACF;CAEA,OAAO,EACL,CAAC,OAAO,iBAAiB;EACvB,IAAMM,IAAQ,CAAC,GACTL,IAAa,IAAI,gBAAgB,GACnCM,GACE,cAAmD;GAEvD,IADA,MAAUD,GACNP,MAAUO,GAEd,OADA,MAAa,QAAQL,CAAU,GACxBM;EACT;EAEA,OAAO;GACL,YACE,MAAM,CAAC,EAAE,KAAK,KAAK,QAAQ,OAAO,IAAIC,EAAqB,0CAA0C,CAAC;GACxG,QAAQ,OAAO,MACT,CAACT,KAASA,MAAUO,IAAc;IAAE,MAAM;IAAa;GAAiB,KAC5EL,EAAW,MAAM,IAAIH,EAAwB,wBAAwB,CAAC,GAC/DS,GAAU,OAAOX,CAAgB,KAAK;IAAE,MAAM;IAAa;GAAiB;GAErF,OAAO,OAAO,MAAoB;IAChC,IAAMa,IAAS,MAAM;IACrB,IAAI,CAACA,GAAQ,MAAM,IAAID,EAAqB,0CAA0C;IAEtF,OADAP,EAAW,MAAMP,CAAK,GACfe,EAAO,MAAMf,CAAK;GAC3B;EACF;CACF,EACF;AACF"}
@@ -1 +1 @@
1
- {"version":3,"file":"_stream-pool.cjs","names":[],"sources":["../src/_stream-pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, StreamWorkerPool } from './types.js';\n\nexport type StreamSlot<TInput, TChunk> = {\n cancel(reason: unknown): void;\n /** Wired by the pool factory to report unhandled worker runtime errors through `tap()`. */\n notify?(error: FamiliarRuntimeError): void;\n stream(input: TInput, options: RunOptions): { done: Promise<void>; iterable: AsyncIterable<TChunk> };\n terminate(): void;\n};\n\nexport function createStreamPool<TInput, TChunk>(\n slots: readonly StreamSlot<TInput, TChunk>[],\n options: PoolOptions,\n): StreamWorkerPool<TInput, TChunk> {\n const core: PoolCore<StreamSlot<TInput, TChunk>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function runStream(input: TInput, runOptions: RunOptions = {}): AsyncIterable<TChunk> {\n const priority = validatePriority(runOptions.priority ?? 0);\n const timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n const transferables = [...(runOptions.transferables ?? [])];\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TChunk> {\n const onAbort = () => controller.abort(runOptions.signal?.reason);\n let slot: StreamSlot<TInput, TChunk> | undefined;\n let onCancel: (() => void) | undefined;\n\n if (runOptions.signal?.aborted) controller.abort(runOptions.signal.reason);\n else runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n const running = await core.acquire({ priority, signal: controller.signal }, (acquired) => {\n slot = acquired;\n core.trackActive(1);\n onCancel = () => slot?.cancel(abortError(controller.signal));\n controller.signal.addEventListener('abort', onCancel, { once: true });\n\n return acquired.stream(input, { priority, signal: controller.signal, timeout, transferables });\n });\n\n for await (const value of running.iterable) yield value;\n await running.done;\n core.trackCompleted();\n } catch (error) {\n if (slot && !controller.signal.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n\n if (slot) {\n if (onCancel) controller.signal.removeEventListener('abort', onCancel);\n\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TChunk> | undefined;\n const start = (): AsyncGenerator<TChunk> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker streams can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TChunk };\n controller.abort(new FamiliarTerminatedError('Stream consumer stopped'));\n return iterator?.return(value as TChunk) ?? { done: true, value: value as TChunk };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker streams can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n runStream,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n"],"mappings":"iGAaA,SAAgB,iBACd,EACA,EACkC,CAClC,IAAM,EAA6C,EAAA,eAAe,EAAO,CAAO,EAEhF,IAAK,IAAM,KAAQ,EAAO,EAAK,OAAU,GAAU,EAAK,QAAQ,CAAE,QAAO,KAAM,cAAe,CAAC,EAE/F,SAAS,UAAU,EAAe,EAAyB,CAAC,EAA0B,CACpF,IAAM,EAAW,EAAA,iBAAiB,EAAW,UAAY,CAAC,EACpD,EAAU,EAAA,gBAAgB,EAAW,OAAO,GAAK,EAAQ,eACzD,EAAgB,CAAC,GAAI,EAAW,eAAiB,CAAC,CAAE,EACtD,EAEE,QAAU,gBAAiB,EAAqD,CACpF,IAAM,YAAgB,EAAW,MAAM,EAAW,QAAQ,MAAM,EAC5D,EACA,SAEA,EAAW,QAAQ,QAAS,EAAW,MAAM,EAAW,OAAO,MAAM,EACpE,EAAW,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEzE,GAAI,CACF,IAAM,EAAU,MAAM,EAAK,QAAQ,CAAE,WAAU,OAAQ,EAAW,MAAO,EAAI,IAC3E,EAAO,EACP,EAAK,YAAY,CAAC,EAClB,aAAiB,GAAM,QAAA,EAAO,EAAA,WAAA,CAAW,EAAW,MAAM,CAAC,EAC3D,EAAW,OAAO,iBAAiB,QAAS,SAAU,CAAE,KAAM,EAAK,CAAC,EAE7D,EAAS,OAAO,EAAO,CAAE,WAAU,OAAQ,EAAW,OAAQ,UAAS,eAAc,CAAC,EAC9F,EAED,UAAW,IAAM,KAAS,EAAQ,SAAU,MAAM,EAClD,MAAM,EAAQ,KACd,EAAK,eAAe,CACtB,OAAS,EAAO,CAEd,MADI,GAAQ,CAAC,EAAW,OAAO,SAAW,EAAE,aAAiB,EAAA,0BAA0B,EAAK,YAAY,EAClG,CACR,QAAU,CACR,EAAW,QAAQ,oBAAoB,QAAS,OAAO,EAEnD,IACE,UAAU,EAAW,OAAO,oBAAoB,QAAS,QAAQ,EAErE,EAAK,YAAY,EAAE,EACnB,EAAK,QAAQ,CAAI,EACjB,EAAK,WAAW,EAEpB,CACF,EAEA,MAAO,CACL,CAAC,OAAO,gBAAiB,CACvB,IAAM,EAAQ,CAAC,EACT,EAAa,IAAI,gBACnB,EACE,UAAkD,CACtD,OAAU,EACN,IAAU,EAEd,MADA,KAAa,QAAQ,CAAU,EACxB,CACT,EAEA,MAAO,CACL,SACE,MAAM,CAAC,EAAE,KAAK,GAAK,QAAQ,OAAO,IAAI,EAAA,qBAAqB,0CAA0C,CAAC,EACxG,OAAQ,KAAO,IACT,CAAC,GAAS,IAAU,EAAc,CAAE,KAAM,GAAa,OAAgB,GAC3E,EAAW,MAAM,IAAI,EAAA,wBAAwB,yBAAyB,CAAC,EAChE,GAAU,OAAO,CAAe,GAAK,CAAE,KAAM,GAAa,OAAgB,GAEnF,MAAO,KAAO,IAAoB,CAChC,IAAM,EAAS,MAAM,EACrB,GAAI,CAAC,EAAQ,MAAM,IAAI,EAAA,qBAAqB,0CAA0C,EAEtF,OADA,EAAW,MAAM,CAAK,EACf,EAAO,MAAM,CAAK,CAC3B,CACF,CACF,CACF,CACF,CAEA,MAAO,CACL,IAAI,gBAAiB,CACnB,OAAO,EAAK,cACd,EACA,QAAS,EAAK,QACd,IAAI,UAAW,CACb,OAAO,EAAK,QACd,EACA,MAAO,EAAK,MACZ,UACA,IAAI,OAAQ,CACV,OAAO,EAAK,MAAM,CACpB,EACA,IAAI,QAAS,CACX,OAAO,EAAK,OAAO,CACrB,EACA,IAAK,EAAK,KACT,OAAO,kBAAqB,EAAK,MAAM,CAAC,CAAC,GACzC,OAAO,SAAU,EAAK,OACzB,CACF"}
1
+ {"version":3,"file":"_stream-pool.cjs","names":["createPoolCore","validatePriority","validateTimeout","abortError","FamiliarTerminatedError","FamiliarRuntimeError","require_errors","require__pool_core","_vielzeug_arsenal","core","slots","options","slot","priority","runOptions","timeout","transferables","owner","controller","running","acquired","input","value","error","token","iterator","active"],"sources":["../src/_stream-pool.ts"],"sourcesContent":["import { abortError } from '@vielzeug/arsenal';\nimport { createPoolCore, type PoolCore, type PoolOptions, validatePriority, validateTimeout } from './_pool-core.js';\nimport { FamiliarRuntimeError, FamiliarTerminatedError } from './errors.js';\nimport type { RunOptions, StreamWorkerPool } from './types.js';\n\nexport type StreamSlot<TInput, TChunk> = {\n cancel(reason: unknown): void;\n /** Wired by the pool factory to report unhandled worker runtime errors through `tap()`. */\n notify?(error: FamiliarRuntimeError): void;\n stream(input: TInput, options: RunOptions): { done: Promise<void>; iterable: AsyncIterable<TChunk> };\n terminate(): void;\n};\n\nexport function createStreamPool<TInput, TChunk>(\n slots: readonly StreamSlot<TInput, TChunk>[],\n options: PoolOptions,\n): StreamWorkerPool<TInput, TChunk> {\n const core: PoolCore<StreamSlot<TInput, TChunk>> = createPoolCore(slots, options);\n\n for (const slot of slots) slot.notify = (error) => core.emitTap({ error, type: 'worker-error' });\n\n function runStream(input: TInput, runOptions: RunOptions = {}): AsyncIterable<TChunk> {\n const priority = validatePriority(runOptions.priority ?? 0);\n const timeout = validateTimeout(runOptions.timeout) ?? options.defaultTimeout;\n const transferables = [...(runOptions.transferables ?? [])];\n let owner: object | undefined;\n\n const iterate = async function* (controller: AbortController): AsyncGenerator<TChunk> {\n const onAbort = () => controller.abort(runOptions.signal?.reason);\n let slot: StreamSlot<TInput, TChunk> | undefined;\n let onCancel: (() => void) | undefined;\n\n if (runOptions.signal?.aborted) controller.abort(runOptions.signal.reason);\n else runOptions.signal?.addEventListener('abort', onAbort, { once: true });\n\n try {\n const running = await core.acquire({ priority, signal: controller.signal }, (acquired) => {\n slot = acquired;\n core.trackActive(1);\n onCancel = () => slot?.cancel(abortError(controller.signal));\n controller.signal.addEventListener('abort', onCancel, { once: true });\n\n return acquired.stream(input, { priority, signal: controller.signal, timeout, transferables });\n });\n\n for await (const value of running.iterable) yield value;\n await running.done;\n core.trackCompleted();\n } catch (error) {\n if (slot && !controller.signal.aborted && !(error instanceof FamiliarTerminatedError)) core.trackFailed();\n throw error;\n } finally {\n runOptions.signal?.removeEventListener('abort', onAbort);\n\n if (slot) {\n if (onCancel) controller.signal.removeEventListener('abort', onCancel);\n\n core.trackActive(-1);\n core.release(slot);\n core.settleIdle();\n }\n }\n };\n\n return {\n [Symbol.asyncIterator]() {\n const token = {};\n const controller = new AbortController();\n let iterator: AsyncGenerator<TChunk> | undefined;\n const start = (): AsyncGenerator<TChunk> | undefined => {\n owner ??= token;\n if (owner !== token) return undefined;\n iterator ??= iterate(controller);\n return iterator;\n };\n\n return {\n next: () =>\n start()?.next() ?? Promise.reject(new FamiliarRuntimeError('Worker streams can only be consumed once')),\n return: async (value?: unknown) => {\n if (!owner || owner !== token) return { done: true, value: value as TChunk };\n controller.abort(new FamiliarTerminatedError('Stream consumer stopped'));\n return iterator?.return(value as TChunk) ?? { done: true, value: value as TChunk };\n },\n throw: async (error?: unknown) => {\n const active = start();\n if (!active) throw new FamiliarRuntimeError('Worker streams can only be consumed once');\n controller.abort(error);\n return active.throw(error);\n },\n };\n },\n };\n }\n\n return {\n get disposalSignal() {\n return core.disposalSignal;\n },\n dispose: core.dispose,\n get disposed() {\n return core.disposed;\n },\n drain: core.drain,\n runStream,\n get stats() {\n return core.stats();\n },\n get status() {\n return core.status();\n },\n tap: core.tap,\n [Symbol.asyncDispose]: () => core.drain({}),\n [Symbol.dispose]: core.dispose,\n };\n}\n"],"mappings":"A,iGAaA,SAAgB,iBACd,EACA,EACkC,CAClC,IAAMS,EAA6CT,EAAAA,eAAeU,EAAOC,CAAO,EAEhF,IAAK,IAAMC,KAAQF,EAAO,EAAK,OAAU,GAAUD,EAAK,QAAQ,CAAE,QAAO,KAAM,cAAe,CAAC,EAE/F,SAAS,UAAU,EAAe,EAAyB,CAAC,EAA0B,CACpF,IAAMI,EAAWZ,EAAAA,iBAAiBa,EAAW,UAAY,CAAC,EACpDC,EAAUb,EAAAA,gBAAgBY,EAAW,OAAO,GAAKH,EAAQ,eACzDK,EAAgB,CAAC,GAAIF,EAAW,eAAiB,CAAC,CAAE,EACtDG,EAEE,QAAU,gBAAiB,EAAqD,CACpF,IAAM,YAAgBC,EAAW,MAAMJ,EAAW,QAAQ,MAAM,EAC5DF,EACA,SAEAE,EAAW,QAAQ,QAASI,EAAW,MAAMJ,EAAW,OAAO,MAAM,EACpEA,EAAW,QAAQ,iBAAiB,QAAS,QAAS,CAAE,KAAM,EAAK,CAAC,EAEzE,GAAI,CACF,IAAMK,EAAU,MAAMV,EAAK,QAAQ,CAAE,WAAU,OAAQS,EAAW,MAAO,EAAI,IAC3E,EAAOE,EACPX,EAAK,YAAY,CAAC,EAClB,aAAiBG,GAAM,QAAA,EAAOT,EAAAA,WAAAA,CAAWe,EAAW,MAAM,CAAC,EAC3DA,EAAW,OAAO,iBAAiB,QAAS,SAAU,CAAE,KAAM,EAAK,CAAC,EAE7DE,EAAS,OAAOC,EAAO,CAAE,WAAU,OAAQH,EAAW,OAAQ,UAAS,eAAc,CAAC,EAC9F,EAED,UAAW,IAAMI,KAASH,EAAQ,SAAU,MAAMG,EAClD,MAAMH,EAAQ,KACdV,EAAK,eAAe,CACtB,OAASc,EAAO,CAEd,MADIX,GAAQ,CAACM,EAAW,OAAO,SAAW,EAAEK,aAAiBnB,EAAAA,0BAA0BK,EAAK,YAAY,EAClGc,CACR,QAAU,CACRT,EAAW,QAAQ,oBAAoB,QAAS,OAAO,EAEnDF,IACE,UAAUM,EAAW,OAAO,oBAAoB,QAAS,QAAQ,EAErET,EAAK,YAAY,EAAE,EACnBA,EAAK,QAAQG,CAAI,EACjBH,EAAK,WAAW,EAEpB,CACF,EAEA,MAAO,CACL,CAAC,OAAO,gBAAiB,CACvB,IAAMe,EAAQ,CAAC,EACTN,EAAa,IAAI,gBACnBO,EACE,UAAkD,CAEtD,GADA,IAAUD,EACNP,IAAUO,EAEd,MADA,KAAa,QAAQN,CAAU,EACxBO,CACT,EAEA,MAAO,CACL,SACE,MAAM,CAAC,EAAE,KAAK,GAAK,QAAQ,OAAO,IAAIpB,EAAAA,qBAAqB,0CAA0C,CAAC,EACxG,OAAQ,KAAO,IACT,CAACY,GAASA,IAAUO,EAAc,CAAE,KAAM,GAAa,OAAgB,GAC3EN,EAAW,MAAM,IAAId,EAAAA,wBAAwB,yBAAyB,CAAC,EAChEqB,GAAU,OAAOH,CAAe,GAAK,CAAE,KAAM,GAAa,OAAgB,GAEnF,MAAO,KAAO,IAAoB,CAChC,IAAMI,EAAS,MAAM,EACrB,GAAI,CAACA,EAAQ,MAAM,IAAIrB,EAAAA,qBAAqB,0CAA0C,EAEtF,OADAa,EAAW,MAAMK,CAAK,EACfG,EAAO,MAAMH,CAAK,CAC3B,CACF,CACF,CACF,CACF,CAEA,MAAO,CACL,IAAI,gBAAiB,CACnB,OAAOd,EAAK,cACd,EACA,QAASA,EAAK,QACd,IAAI,UAAW,CACb,OAAOA,EAAK,QACd,EACA,MAAOA,EAAK,MACZ,UACA,IAAI,OAAQ,CACV,OAAOA,EAAK,MAAM,CACpB,EACA,IAAI,QAAS,CACX,OAAOA,EAAK,OAAO,CACrB,EACA,IAAKA,EAAK,KACT,OAAO,kBAAqBA,EAAK,MAAM,CAAC,CAAC,GACzC,OAAO,SAAUA,EAAK,OACzB,CACF,C"}