@kubb/core 5.0.0-beta.69 → 5.0.0-beta.70

package/dist/index.js

@@ -95,47 +95,6 @@ function randomCliColor(text) {
 }
 //#endregion
 //#region ../../internals/utils/src/promise.ts
-function* chunks(arr, size) {
-	for (let i = 0; i < arr.length; i += size) yield arr.slice(i, i + size);
-}
-/**
-* Slices `source` into batches of `concurrency` items and awaits `process` for each batch.
-* Accepts both plain arrays (sync) and `AsyncIterable` (streaming).
-*
-* `process` controls whether items inside a batch run in parallel; this helper only
-* controls batch size and per-batch flushing.
-*
-* @example
-* ```ts
-* // parallel dispatch inside each batch
-* await forBatches(schemas, (batch) => Promise.all(batch.map(process)), { concurrency: 8 })
-*
-* // async iterable with a flush after every batch
-* await forBatches(stream.schemas, (batch) => dispatch(batch), { concurrency: 8, flush })
-* ```
-*/
-async function forBatches(source, process, options) {
-	const { concurrency, flush } = options;
-	if (Array.isArray(source)) {
-		for (const batch of chunks(source, concurrency)) {
-			await process(batch);
-			if (flush) await flush();
-		}
-		return;
-	}
-	const batch = [];
-	for await (const item of source) {
-		batch.push(item);
-		if (batch.length >= concurrency) {
-			await process(batch.splice(0));
-			if (flush) await flush();
-		}
-	}
-	if (batch.length > 0) {
-		await process(batch.splice(0));
-		if (flush) await flush();
-	}
-}
 /** Returns `true` when `result` is a thenable `Promise`.
 *
 * @example
@@ -445,7 +404,7 @@ function createAdapter(build) {
 /**
 * Docs major version, derived from the package version so the link tracks the published major.
 */
-const docsMajor = "5.0.0-beta.69".split(".")[0] ?? "5";
+const docsMajor = "5.0.0-beta.70".split(".")[0] ?? "5";
 /**
 * Narrows a {@link Diagnostic} to the variant for `code`, or `null` when it does not match.
 *
@@ -1910,27 +1869,28 @@ var KubbDriver = class {
 			checkAllowedNames: false,
 			emit: emitsOperationHook ? (node, ctx) => this.hooks.emit("kubb:generate:operation", node, ctx) : null
 		};
+		const dispatchPass = async (state, nodes, dispatch) => {
+			let sinceFlush = 0;
+			for (const node of nodes) {
+				await dispatchNode(state, node, dispatch);
+				if (++sinceFlush >= 8) {
+					sinceFlush = 0;
+					await flushPending();
+				}
+			}
+			if (sinceFlush > 0) await flushPending();
+		};
 		for (const state of states) {
-			const needsCollectedOperations = emitsOperationsHook || state.generators.some((gen) => !!gen.operations);
-			const collectedOperations = needsCollectedOperations ? [] : void 0;
-			await forBatches(schemasBuffer, (nodes) => Promise.all(nodes.map((node) => dispatchNode(state, node, schemaDispatch))), {
-				concurrency: 8,
-				flush: flushPending
-			});
-			await forBatches(operationsBuffer, (nodes) => {
-				if (needsCollectedOperations) collectedOperations?.push(...nodes);
-				return Promise.all(nodes.map((node) => dispatchNode(state, node, operationDispatch)));
-			}, {
-				concurrency: 8,
-				flush: flushPending
-			});
-			if (!state.failed && needsCollectedOperations) try {
+			const needsOperationsAggregate = emitsOperationsHook || state.generators.some((gen) => !!gen.operations);
+			await dispatchPass(state, schemasBuffer, schemaDispatch);
+			await dispatchPass(state, operationsBuffer, operationDispatch);
+			if (!state.failed && needsOperationsAggregate) try {
 				const { plugin, generatorContext, generators } = state;
 				const ctx = {
 					...generatorContext,
 					options: plugin.options
 				};
-				const pluginOperations = (collectedOperations ?? []).reduce((acc, node) => {
+				const pluginOperations = operationsBuffer.reduce((acc, node) => {
 					const resolved = resolveForPlugin(state, node);
 					if (resolved) acc.push(resolved.transformedNode);
 					return acc;