graphql 17.0.0-alpha.8 → 17.0.0-beta.0

package/execution/incremental/Queue.js

@@ -0,0 +1,330 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Queue = void 0;
+const invariant_js_1 = require("../../jsutils/invariant.js");
+const isPromise_js_1 = require("../../jsutils/isPromise.js");
+const promiseWithResolvers_js_1 = require("../../jsutils/promiseWithResolvers.js");
+const withConcurrentAbruptClose_js_1 = require("../withConcurrentAbruptClose.js");
+/**
+ * A Queue is a lightweight async-generator primitive inspired by Brian Kim's
+ * Repeater (https://repeater.js.org, https://github.com/repeaterjs/repeater).
+ * The ergonomics are similar, but this implementation favors clarity over
+ * performance and gives producers flexibility to remain lazy, become eager, or
+ * live somewhere in between.
+ *
+ * The constructor takes an executor function and an optional `initialCapacity`.
+ * Executors receive `{ push, stop, onStop, started }` and may return `void` or
+ * a promise if they perform asynchronous setup. They call `push` whenever
+ * another item is ready, call `stop` when no more values will be produced
+ * (optionally supplying an error), register stop-time cleanup via `onStop`,
+ * and await `started` when setup should run only after iteration begins.
+ * Because `push`, `stop`, and `onStop` are plain functions, executors can
+ * hoist them into outside scopes or pass them to helpers. If the executor
+ * throws or its returned promise rejects, the queue treats it as `stop(error)`
+ * and propagates the failure.
+ *
+ * The `initialCapacity` argument (default `1`) governs backpressure. Capacity
+ * is the maximum number of buffered items allowed before a push must wait.
+ * When the backlog reaches capacity, `push` returns a promise that settles
+ * once consumption releases space; otherwise it returns `undefined`. Setting
+ * capacity to `1` yields a fully lazy queue (every push waits unless a prior
+ * item has been consumed); higher capacities buffer that many items eagerly.
+ * Capacity can be changed later via `setCapacity` and observed via
+ * `getCapacity`.
+ *
+ * `subscribe(reducer)` returns an async generator whose batches feed a generator
+ * of settled values into the reducer; whatever the reducer returns (other than
+ * `undefined`) becomes the yielded value for that batch. Calling `return()` on
+ * the subscription settles pending `next` calls thanks to `withConcurrent`,
+ * providing direct abort semantics rather than leaving `next()` suspended.
+ *
+ * 'forEachBatch(reducer)` is a convenience method that subscribes with the
+ * given reducer and runs it for each batch until the queue stops.
+ *
+ * Producers can stay lazy by awaiting `started`, using zero capacity, and
+ * awaiting each `push`. Skipping those waits while raising capacity makes the
+ * queue eager up to its configured limit. The `isStopped()` helper exposes
+ * whether the queue has fully stopped, which can be useful when the reducer
+ * function actually performs external work and wants to bail early without
+ * awaiting another `next`.
+ *
+ * @internal
+ */
+class Queue {
+    constructor(executor, initialCapacity = 1) {
+        this._backlog = 0;
+        this._waiters = [];
+        this._entries = [];
+        this._isStopped = false;
+        this._stopRequested = false;
+        this._stopCleanupCallbacks = [];
+        this._batchRequests = new Set();
+        this._capacity = this._normalizeCapacity(initialCapacity);
+        const { promise: started, resolve: resolveStarted } = 
+        // eslint-disable-next-line @typescript-eslint/no-invalid-void-type
+        (0, promiseWithResolvers_js_1.promiseWithResolvers)();
+        this._resolveStarted = resolveStarted;
+        try {
+            const result = executor({
+                push: this._push.bind(this),
+                stop: this._stop.bind(this),
+                onStop: this._onStop.bind(this),
+                started,
+            });
+            if ((0, isPromise_js_1.isPromise)(result)) {
+                result.catch((error) => this._stop(error));
+            }
+        }
+        catch (error) {
+            const stopped = this._stop(error);
+            if ((0, isPromise_js_1.isPromise)(stopped)) {
+                stopped.catch(() => undefined);
+            }
+        }
+    }
+    subscribe(reducer = (generator) => Array.from(generator)) {
+        const generator = this._iteratorLoop(reducer);
+        return (0, withConcurrentAbruptClose_js_1.withConcurrentAbruptClose)(generator, () => this.cancel(), (error) => this.abort(error));
+    }
+    cancel() {
+        if (this._stopRequested) {
+            return this._stopCompletion;
+        }
+        return this._terminate(undefined, () => {
+            this._isStopped = true;
+            this._batchRequests.forEach((request) => request.resolve(undefined));
+            this._batchRequests.clear();
+        });
+    }
+    abort(reason) {
+        if (this._stopRequested) {
+            return this._stopCompletion;
+        }
+        return this._terminate(reason, () => {
+            this._isStopped = true;
+            if (this._batchRequests.size) {
+                this._batchRequests.forEach((request) => request.reject(reason));
+                this._batchRequests.clear();
+                return;
+            }
+            // save rejection for later batch requests
+            this._entries.push({
+                kind: 'item',
+                settled: { status: 'rejected', reason },
+            });
+        });
+    }
+    async forEachBatch(reducer) {
+        const sub = this.subscribe(async (generator) => {
+            const { promise: drained, resolve } = 
+            // eslint-disable-next-line @typescript-eslint/no-invalid-void-type
+            (0, promiseWithResolvers_js_1.promiseWithResolvers)();
+            const wrappedBatch = (function* wrapper() {
+                yield* generator;
+                resolve();
+            })();
+            await Promise.all([reducer(wrappedBatch), drained]);
+        });
+        for await (const _ of sub /* c8 ignore start */) {
+            // intentionally empty
+        } /* c8 ignore stop */
+    }
+    setCapacity(nextCapacity) {
+        this._capacity = this._normalizeCapacity(nextCapacity);
+        this._flush();
+    }
+    getCapacity() {
+        return this._capacity;
+    }
+    isStopped() {
+        return this._isStopped;
+    }
+    _normalizeCapacity(capacity) {
+        return Math.max(1, Math.floor(capacity));
+    }
+    _flush() {
+        while (this._waiters.length > 0 && this._backlog < this._capacity) {
+            this._waiters.shift()?.();
+        }
+    }
+    _reserve() {
+        this._backlog += 1;
+        if (this._backlog < this._capacity) {
+            return undefined;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-invalid-void-type
+        const { promise, resolve } = (0, promiseWithResolvers_js_1.promiseWithResolvers)();
+        this._waiters.push(resolve);
+        return promise;
+    }
+    _release() {
+        if (this._backlog > 0) {
+            this._backlog -= 1;
+        }
+        this._flush();
+    }
+    _onStop(cleanup) {
+        if (this._stopRequested) {
+            throw new Error('Cannot register onStop cleanup after stop has been requested.');
+        }
+        this._stopCleanupCallbacks.push(cleanup);
+    }
+    _runStopCleanup(reason, afterCleanup) {
+        this._stopRequested = true;
+        const cleanupPromises = this._stopCleanupCallbacks.flatMap((cleanupCallback) => {
+            try {
+                const result = cleanupCallback(reason);
+                return (0, isPromise_js_1.isPromise)(result) ? [result] : [];
+            } /* c8 ignore start */
+            catch {
+                // ignore errors
+                return [];
+            } /* c8 ignore stop */
+        });
+        const cleanup = cleanupPromises.length > 0
+            ? Promise.allSettled(cleanupPromises).then(() => undefined)
+            : undefined;
+        if ((0, isPromise_js_1.isPromise)(cleanup)) {
+            this._stopCompletion = cleanup
+                .then(afterCleanup, afterCleanup)
+                .then(() => undefined);
+            return this._stopCompletion;
+        }
+        afterCleanup();
+    }
+    async *_iteratorLoop(reducer) {
+        this._resolveStarted();
+        let nextBatch;
+        // eslint-disable-next-line no-await-in-loop
+        while ((nextBatch = await this._waitForNextBatch())) {
+            let reduced = reducer(nextBatch);
+            if ((0, isPromise_js_1.isPromise)(reduced)) {
+                // eslint-disable-next-line no-await-in-loop
+                reduced = await reduced;
+            }
+            if (reduced === undefined) {
+                continue;
+            }
+            yield reduced;
+        }
+    }
+    _waitForNextBatch() {
+        const { promise, resolve, reject } = (0, promiseWithResolvers_js_1.promiseWithResolvers)();
+        this._batchRequests.add({ resolve, reject });
+        this._deliverBatchIfReady();
+        return promise;
+    }
+    _push(item) {
+        if (this._stopRequested) {
+            return;
+        }
+        const maybePushPromise = this._reserve();
+        if ((0, isPromise_js_1.isPromise)(item)) {
+            const entry = { kind: 'item' };
+            this._entries.push(entry);
+            item.then((resolved) => {
+                entry.settled = { status: 'fulfilled', value: resolved };
+                this._deliverBatchIfReady();
+            }, (reason) => {
+                entry.settled = { status: 'rejected', reason };
+                this._deliverBatchIfReady();
+            });
+        }
+        else {
+            this._entries.push({
+                kind: 'item',
+                settled: { status: 'fulfilled', value: item },
+            });
+            this._deliverBatchIfReady();
+        }
+        return maybePushPromise;
+    }
+    _terminate(reason, afterCleanup) {
+        for (const entry of this._entries) {
+            if (entry.kind === 'item') {
+                this._release();
+            }
+        }
+        this._entries.length = 0;
+        return this._runStopCleanup(reason, afterCleanup);
+    }
+    _stop(reason) {
+        if (this._stopRequested) {
+            return this._stopCompletion;
+        }
+        const stopCompletion = this._runStopCleanup(reason, () => {
+            if (reason === undefined) {
+                if (this._entries.length === 0) {
+                    this._isStopped = true;
+                    this._deliverBatchIfReady();
+                    return;
+                }
+                this._entries.push({ kind: 'stop' });
+                this._deliverBatchIfReady();
+                return;
+            }
+            this._entries.push({
+                kind: 'item',
+                settled: { status: 'rejected', reason },
+            });
+            this._entries.push({ kind: 'stop' });
+            this._deliverBatchIfReady();
+        });
+        if ((0, isPromise_js_1.isPromise)(stopCompletion)) {
+            stopCompletion.catch(() => undefined);
+        }
+        return stopCompletion;
+    }
+    _deliverBatchIfReady() {
+        if (!this._batchRequests.size) {
+            return;
+        }
+        const headEntry = this._entries[0];
+        const requests = this._batchRequests;
+        if (headEntry !== undefined) {
+            if (!(headEntry.kind !== 'stop'))
+                (0, 
+                // stop sentinel always follows other work
+                invariant_js_1.invariant)(false);
+            const settled = headEntry.settled;
+            if (settled !== undefined) {
+                if (settled.status === 'fulfilled') {
+                    this._batchRequests = new Set();
+                    requests.forEach((request) => request.resolve(this._drainBatch()));
+                    return;
+                }
+                this._entries.shift();
+                this._release();
+                this._isStopped = true;
+                this._batchRequests = new Set();
+                requests.forEach((request) => request.reject(settled.reason));
+            }
+        }
+        else if (this._isStopped) {
+            this._batchRequests = new Set();
+            requests.forEach((request) => request.resolve(undefined));
+        }
+    }
+    *_drainBatch() {
+        while (true) {
+            const entry = this._entries[0];
+            if (entry === undefined) {
+                return;
+            }
+            if (entry.kind === 'stop') {
+                this._isStopped = true;
+                this._entries.shift();
+                return;
+            }
+            const settled = entry.settled;
+            if (settled === undefined || settled.status === 'rejected') {
+                return;
+            }
+            this._entries.shift();
+            this._release();
+            yield settled.value;
+        }
+    }
+}
+exports.Queue = Queue;
+//# sourceMappingURL=Queue.js.map

package/execution/incremental/Queue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Queue.js","sourceRoot":"","sources":["../../../src/execution/incremental/Queue.ts"],"names":[],"mappings":";;;AAAA,6DAAuD;AACvD,6DAAuD;AAEvD,mFAA6E;AAE7E,kFAA4E;AA6B5E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4CG;AACH,MAAa,KAAK;IAehB,YACE,QAKmD,EACnD,eAAe,GAAG,CAAC;QApBb,aAAQ,GAAG,CAAC,CAAC;QACb,aAAQ,GAAsB,EAAE,CAAC;QACjC,aAAQ,GAAoB,EAAE,CAAC;QAC/B,eAAU,GAAG,KAAK,CAAC;QACnB,mBAAc,GAAG,KAAK,CAAC;QACvB,0BAAqB,GAEzB,EAAE,CAAC;QAEC,mBAAc,GAAG,IAAI,GAAG,EAAmB,CAAC;QAalD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC;QAE1D,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE;QACjD,mEAAmE;QACnE,IAAA,8CAAoB,GAAQ,CAAC;QAE/B,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;QAEtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC3B,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC3B,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC/B,OAAO;aACR,CAAC,CAAC;YACH,IAAI,IAAA,wBAAS,EAAC,MAAM,CAAC,EAAE,CAAC;gBACtB,MAAM,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,IAAA,wBAAS,EAAC,OAAO,CAAC,EAAE,CAAC;gBACvB,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,CACP,UAEqC,CAAC,SAAS,EAAE,EAAE,CACjD,KAAK,CAAC,IAAI,CAAC,SAAS,CAAM;QAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,OAAO,IAAA,wDAAyB,EAC9B,SAAS,EACT,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,EACnB,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED,MAAM;QACJ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,GAAG,EAAE;YACrC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;YACrE,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAgB;QACpB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,EAAE;YAClC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;gBAC7B,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;gBACjE,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;gBAC5B,OAAO;YACT,CAAC;YACD,0CAA0C;YAC1C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;aACxC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAsE;QAEtE,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAC7C,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE;YACjC,mEAAmE;YACnE,IAAA,8CAAoB,GAAQ,CAAC;YAE/B,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,CAAC,OAAO;gBACrC,KAAK,CAAC,CAAC,SAAS,CAAC;gBACjB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,EAAE,CAAC;YAEL,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,GAAG,CAAC,qBAAqB,EAAE,CAAC;YAChD,sBAAsB;QACxB,CAAC,CAAC,oBAAoB;IACxB,CAAC;IAED,WAAW,CAAC,YAAoB;QAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAEO,kBAAkB,CAAC,QAAgB;QACzC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC3C,CAAC;IAEO,MAAM;QACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAClE,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;QACnB,IAAI,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YACnC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,mEAAmE;QACnE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,IAAA,8CAAoB,GAAQ,CAAC;QAC1D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,QAAQ;QACd,IAAI,IAAI,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;QACrB,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAEO,OAAO,CAAC,OAAmD;QACjE,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,+DAA+D,CAChE,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEO,eAAe,CACrB,MAAe,EACf,YAAwB;QAExB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CACxD,CAAC,eAAe,EAA2B,EAAE;YAC3C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;gBACvC,OAAO,IAAA,wBAAS,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,CAAC,CAAC,qBAAqB;YAAC,MAAM,CAAC;gBAC7B,gBAAgB;gBAChB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,oBAAoB;QACxB,CAAC,CACF,CAAC;QACF,MAAM,OAAO,GACX,eAAe,CAAC,MAAM,GAAG,CAAC;YACxB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YAC3D,CAAC,CAAC,SAAS,CAAC;QAChB,IAAI,IAAA,wBAAS,EAAC,OAAO,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,eAAe,GAAG,OAAO;iBAC3B,IAAI,CAAC,YAAY,EAAE,YAAY,CAAC;iBAChC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACzB,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,YAAY,EAAE,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa,CAC1B,OAEkC;QAElC,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,SAAmC,CAAC;QACxC,4CAA4C;QAC5C,OAAO,CAAC,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC,EAAE,CAAC;YACpD,IAAI,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;YACjC,IAAI,IAAA,wBAAS,EAAC,OAAO,CAAC,EAAE,CAAC;gBACvB,4CAA4C;gBAC5C,OAAO,GAAG,MAAM,OAAO,CAAC;YAC1B,CAAC;YACD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,OAAO,CAAC;QAChB,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAA,8CAAoB,GAEtD,CAAC;QACJ,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5B,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,IAAuB;QACnC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QACzC,IAAI,IAAA,wBAAS,EAAC,IAAI,CAAC,EAAE,CAAC;YACpB,MAAM,KAAK,GAAiB,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAC7C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,CAAC,IAAI,CACP,CAAC,QAAQ,EAAE,EAAE;gBACX,KAAK,CAAC,OAAO,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACzD,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,CAAC,EACD,CAAC,MAAe,EAAE,EAAE;gBAClB,KAAK,CAAC,OAAO,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBAC/C,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,CAAC,CACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE;aAC9C,CAAC,CAAC;YACH,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAEO,UAAU,CAChB,MAAe,EACf,YAAwB;QAExB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACzB,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,MAAgB;QAC5B,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,GAAG,EAAE;YACvD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC/B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;oBACvB,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBACrC,IAAI,CAAC,oBAAoB,EAAE,CAAC;gBAC5B,OAAO;YACT,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;aACxC,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YACrC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAI,IAAA,wBAAS,EAAC,cAAc,CAAC,EAAE,CAAC;YAC9B,cAAc,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IAEO,oBAAoB;QAC1B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC9B,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC;QACrC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;kBAElB,SAAS,CAAC,IAAI,KAAK,MAAM;;gBADnC,0CAA0C;gBAC1C,wBAAS;YAET,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;YAClC,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACnC,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;oBAChC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;oBACnE,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAChB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;gBACvB,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;gBAChC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC3B,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;YAChC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,CAAC,WAAW;QAClB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC/B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,OAAO;YACT,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;gBACvB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YAC9B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBAC3D,OAAO;YACT,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChB,MAAM,OAAO,CAAC,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAzVD,sBAyVC","sourcesContent":["import { invariant } from '../../jsutils/invariant.js';\nimport { isPromise } from '../../jsutils/isPromise.js';\nimport type { PromiseOrValue } from '../../jsutils/PromiseOrValue.js';\nimport { promiseWithResolvers } from '../../jsutils/promiseWithResolvers.js';\n\nimport { withConcurrentAbruptClose } from '../withConcurrentAbruptClose.js';\n\ntype Settled<T> =\n  | { status: 'fulfilled'; value: T }\n  | { status: 'rejected'; reason: unknown };\n\ninterface ItemEntry<T> {\n  kind: 'item';\n  settled?: Settled<T>;\n}\n\ninterface StopEntry {\n  kind: 'stop';\n}\n\ntype Entry<T> = ItemEntry<T> | StopEntry;\n\ninterface BatchRequest<T> {\n  resolve: (generator: Generator<T> | undefined) => void;\n  reject: (reason: unknown) => void;\n}\n\ninterface QueueExecutorOptions<T> {\n  push: (item: PromiseOrValue<T>) => PromiseOrValue<void>;\n  stop: (reason?: unknown) => PromiseOrValue<void>;\n  onStop: (cleanup: (reason?: unknown) => PromiseOrValue<void>) => void;\n  started: Promise<void>;\n}\n\n/**\n * A Queue is a lightweight async-generator primitive inspired by Brian Kim's\n * Repeater (https://repeater.js.org, https://github.com/repeaterjs/repeater).\n * The ergonomics are similar, but this implementation favors clarity over\n * performance and gives producers flexibility to remain lazy, become eager, or\n * live somewhere in between.\n *\n * The constructor takes an executor function and an optional `initialCapacity`.\n * Executors receive `{ push, stop, onStop, started }` and may return `void` or\n * a promise if they perform asynchronous setup. They call `push` whenever\n * another item is ready, call `stop` when no more values will be produced\n * (optionally supplying an error), register stop-time cleanup via `onStop`,\n * and await `started` when setup should run only after iteration begins.\n * Because `push`, `stop`, and `onStop` are plain functions, executors can\n * hoist them into outside scopes or pass them to helpers. If the executor\n * throws or its returned promise rejects, the queue treats it as `stop(error)`\n * and propagates the failure.\n *\n * The `initialCapacity` argument (default `1`) governs backpressure. Capacity\n * is the maximum number of buffered items allowed before a push must wait.\n * When the backlog reaches capacity, `push` returns a promise that settles\n * once consumption releases space; otherwise it returns `undefined`. Setting\n * capacity to `1` yields a fully lazy queue (every push waits unless a prior\n * item has been consumed); higher capacities buffer that many items eagerly.\n * Capacity can be changed later via `setCapacity` and observed via\n * `getCapacity`.\n *\n * `subscribe(reducer)` returns an async generator whose batches feed a generator\n * of settled values into the reducer; whatever the reducer returns (other than\n * `undefined`) becomes the yielded value for that batch. Calling `return()` on\n * the subscription settles pending `next` calls thanks to `withConcurrent`,\n * providing direct abort semantics rather than leaving `next()` suspended.\n *\n * 'forEachBatch(reducer)` is a convenience method that subscribes with the\n * given reducer and runs it for each batch until the queue stops.\n *\n * Producers can stay lazy by awaiting `started`, using zero capacity, and\n * awaiting each `push`. Skipping those waits while raising capacity makes the\n * queue eager up to its configured limit. The `isStopped()` helper exposes\n * whether the queue has fully stopped, which can be useful when the reducer\n * function actually performs external work and wants to bail early without\n * awaiting another `next`.\n *\n * @internal\n */\nexport class Queue<T> {\n  private _capacity: number;\n  private _backlog = 0;\n  private _waiters: Array<() => void> = [];\n  private _entries: Array<Entry<T>> = [];\n  private _isStopped = false;\n  private _stopRequested = false;\n  private _stopCleanupCallbacks: Array<\n    (reason?: unknown) => PromiseOrValue<void>\n  > = [];\n  private _stopCompletion: Promise<void> | undefined;\n  private _batchRequests = new Set<BatchRequest<T>>();\n\n  private _resolveStarted: () => void;\n\n  constructor(\n    executor: ({\n      push,\n      stop,\n      onStop,\n      started,\n    }: QueueExecutorOptions<T>) => PromiseOrValue<void>,\n    initialCapacity = 1,\n  ) {\n    this._capacity = this._normalizeCapacity(initialCapacity);\n\n    const { promise: started, resolve: resolveStarted } =\n      // eslint-disable-next-line @typescript-eslint/no-invalid-void-type\n      promiseWithResolvers<void>();\n\n    this._resolveStarted = resolveStarted;\n\n    try {\n      const result = executor({\n        push: this._push.bind(this),\n        stop: this._stop.bind(this),\n        onStop: this._onStop.bind(this),\n        started,\n      });\n      if (isPromise(result)) {\n        result.catch((error: unknown) => this._stop(error));\n      }\n    } catch (error) {\n      const stopped = this._stop(error);\n      if (isPromise(stopped)) {\n        stopped.catch(() => undefined);\n      }\n    }\n  }\n\n  subscribe<U>(\n    reducer: (\n      generator: Generator<T, void, void>,\n    ) => PromiseOrValue<U | undefined> = (generator) =>\n      Array.from(generator) as U,\n  ): AsyncGenerator<U, void, void> {\n    const generator = this._iteratorLoop(reducer);\n    return withConcurrentAbruptClose(\n      generator,\n      () => this.cancel(),\n      (error) => this.abort(error),\n    );\n  }\n\n  cancel(): PromiseOrValue<void> {\n    if (this._stopRequested) {\n      return this._stopCompletion;\n    }\n    return this._terminate(undefined, () => {\n      this._isStopped = true;\n      this._batchRequests.forEach((request) => request.resolve(undefined));\n      this._batchRequests.clear();\n    });\n  }\n\n  abort(reason?: unknown): PromiseOrValue<void> {\n    if (this._stopRequested) {\n      return this._stopCompletion;\n    }\n    return this._terminate(reason, () => {\n      this._isStopped = true;\n      if (this._batchRequests.size) {\n        this._batchRequests.forEach((request) => request.reject(reason));\n        this._batchRequests.clear();\n        return;\n      }\n      // save rejection for later batch requests\n      this._entries.push({\n        kind: 'item',\n        settled: { status: 'rejected', reason },\n      });\n    });\n  }\n\n  async forEachBatch(\n    reducer: (generator: Generator<T, void, void>) => PromiseOrValue<void>,\n  ): Promise<void> {\n    const sub = this.subscribe(async (generator) => {\n      const { promise: drained, resolve } =\n        // eslint-disable-next-line @typescript-eslint/no-invalid-void-type\n        promiseWithResolvers<void>();\n\n      const wrappedBatch = (function* wrapper(): Generator<T> {\n        yield* generator;\n        resolve();\n      })();\n\n      await Promise.all([reducer(wrappedBatch), drained]);\n    });\n\n    for await (const _ of sub /* c8 ignore start */) {\n      // intentionally empty\n    } /* c8 ignore stop */\n  }\n\n  setCapacity(nextCapacity: number): void {\n    this._capacity = this._normalizeCapacity(nextCapacity);\n    this._flush();\n  }\n\n  getCapacity(): number {\n    return this._capacity;\n  }\n\n  isStopped(): boolean {\n    return this._isStopped;\n  }\n\n  private _normalizeCapacity(capacity: number): number {\n    return Math.max(1, Math.floor(capacity));\n  }\n\n  private _flush(): void {\n    while (this._waiters.length > 0 && this._backlog < this._capacity) {\n      this._waiters.shift()?.();\n    }\n  }\n\n  private _reserve(): PromiseOrValue<void> {\n    this._backlog += 1;\n    if (this._backlog < this._capacity) {\n      return undefined;\n    }\n    // eslint-disable-next-line @typescript-eslint/no-invalid-void-type\n    const { promise, resolve } = promiseWithResolvers<void>();\n    this._waiters.push(resolve);\n    return promise;\n  }\n\n  private _release(): void {\n    if (this._backlog > 0) {\n      this._backlog -= 1;\n    }\n    this._flush();\n  }\n\n  private _onStop(cleanup: (reason?: unknown) => PromiseOrValue<void>): void {\n    if (this._stopRequested) {\n      throw new Error(\n        'Cannot register onStop cleanup after stop has been requested.',\n      );\n    }\n    this._stopCleanupCallbacks.push(cleanup);\n  }\n\n  private _runStopCleanup(\n    reason: unknown,\n    afterCleanup: () => void,\n  ): PromiseOrValue<void> {\n    this._stopRequested = true;\n    const cleanupPromises = this._stopCleanupCallbacks.flatMap(\n      (cleanupCallback): Array<Promise<unknown>> => {\n        try {\n          const result = cleanupCallback(reason);\n          return isPromise(result) ? [result] : [];\n        } /* c8 ignore start */ catch {\n          // ignore errors\n          return [];\n        } /* c8 ignore stop */\n      },\n    );\n    const cleanup =\n      cleanupPromises.length > 0\n        ? Promise.allSettled(cleanupPromises).then(() => undefined)\n        : undefined;\n    if (isPromise(cleanup)) {\n      this._stopCompletion = cleanup\n        .then(afterCleanup, afterCleanup)\n        .then(() => undefined);\n      return this._stopCompletion;\n    }\n    afterCleanup();\n  }\n\n  private async *_iteratorLoop<U>(\n    reducer: (\n      generator: Generator<T, void, void>,\n    ) => PromiseOrValue<U | undefined>,\n  ): AsyncGenerator<U, void, void> {\n    this._resolveStarted();\n    let nextBatch: Generator<T> | undefined;\n    // eslint-disable-next-line no-await-in-loop\n    while ((nextBatch = await this._waitForNextBatch())) {\n      let reduced = reducer(nextBatch);\n      if (isPromise(reduced)) {\n        // eslint-disable-next-line no-await-in-loop\n        reduced = await reduced;\n      }\n      if (reduced === undefined) {\n        continue;\n      }\n      yield reduced;\n    }\n  }\n\n  private _waitForNextBatch(): Promise<Generator<T> | undefined> {\n    const { promise, resolve, reject } = promiseWithResolvers<\n      Generator<T> | undefined\n    >();\n    this._batchRequests.add({ resolve, reject });\n    this._deliverBatchIfReady();\n    return promise;\n  }\n\n  private _push(item: PromiseOrValue<T>): PromiseOrValue<void> {\n    if (this._stopRequested) {\n      return;\n    }\n    const maybePushPromise = this._reserve();\n    if (isPromise(item)) {\n      const entry: ItemEntry<T> = { kind: 'item' };\n      this._entries.push(entry);\n      item.then(\n        (resolved) => {\n          entry.settled = { status: 'fulfilled', value: resolved };\n          this._deliverBatchIfReady();\n        },\n        (reason: unknown) => {\n          entry.settled = { status: 'rejected', reason };\n          this._deliverBatchIfReady();\n        },\n      );\n    } else {\n      this._entries.push({\n        kind: 'item',\n        settled: { status: 'fulfilled', value: item },\n      });\n      this._deliverBatchIfReady();\n    }\n    return maybePushPromise;\n  }\n\n  private _terminate(\n    reason: unknown,\n    afterCleanup: () => void,\n  ): PromiseOrValue<void> {\n    for (const entry of this._entries) {\n      if (entry.kind === 'item') {\n        this._release();\n      }\n    }\n    this._entries.length = 0;\n    return this._runStopCleanup(reason, afterCleanup);\n  }\n\n  private _stop(reason?: unknown): PromiseOrValue<void> {\n    if (this._stopRequested) {\n      return this._stopCompletion;\n    }\n    const stopCompletion = this._runStopCleanup(reason, () => {\n      if (reason === undefined) {\n        if (this._entries.length === 0) {\n          this._isStopped = true;\n          this._deliverBatchIfReady();\n          return;\n        }\n\n        this._entries.push({ kind: 'stop' });\n        this._deliverBatchIfReady();\n        return;\n      }\n\n      this._entries.push({\n        kind: 'item',\n        settled: { status: 'rejected', reason },\n      });\n      this._entries.push({ kind: 'stop' });\n      this._deliverBatchIfReady();\n    });\n\n    if (isPromise(stopCompletion)) {\n      stopCompletion.catch(() => undefined);\n    }\n    return stopCompletion;\n  }\n\n  private _deliverBatchIfReady(): void {\n    if (!this._batchRequests.size) {\n      return;\n    }\n    const headEntry = this._entries[0];\n    const requests = this._batchRequests;\n    if (headEntry !== undefined) {\n      // stop sentinel always follows other work\n      invariant(headEntry.kind !== 'stop');\n\n      const settled = headEntry.settled;\n      if (settled !== undefined) {\n        if (settled.status === 'fulfilled') {\n          this._batchRequests = new Set();\n          requests.forEach((request) => request.resolve(this._drainBatch()));\n          return;\n        }\n        this._entries.shift();\n        this._release();\n        this._isStopped = true;\n        this._batchRequests = new Set();\n        requests.forEach((request) => request.reject(settled.reason));\n      }\n    } else if (this._isStopped) {\n      this._batchRequests = new Set();\n      requests.forEach((request) => request.resolve(undefined));\n    }\n  }\n\n  private *_drainBatch(): Generator<T> {\n    while (true) {\n      const entry = this._entries[0];\n      if (entry === undefined) {\n        return;\n      }\n      if (entry.kind === 'stop') {\n        this._isStopped = true;\n        this._entries.shift();\n        return;\n      }\n      const settled = entry.settled;\n      if (settled === undefined || settled.status === 'rejected') {\n        return;\n      }\n      this._entries.shift();\n      this._release();\n      yield settled.value;\n    }\n  }\n}\n"]}