fix: external reject size scheduler

nmathew98 · Mar 29, 2024 · 2751085 · 2751085
1 parent 56e90aa
commit 2751085
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/shared/src/batcher/index.ts b/packages/shared/src/batcher/index.ts
@@ -60,12 +60,14 @@ export const makeIntervalScheduler = (ms?: number) => {
 
 export const makeSizeScheduler = (minimumSize: number) => {
 	let externalResolve: null | ((value: any) => void) = null;
+	let externalReject: null | ((reason: unknown) => void) = null;
 	let pendingExecution = Promise.resolve();
 
 	return (executor: any, pending: any[]) => {
 		if (pending.length < minimumSize) {
-			const next: Promise<void> = new Promise(resolve => {
+			const next: Promise<void> = new Promise((resolve, reject) => {
 				externalResolve = resolve;
+				externalReject = reject;
 			});
 
 			pendingExecution = pendingExecution.then(() => next);
@@ -74,7 +76,11 @@ export const makeSizeScheduler = (minimumSize: number) => {
 		}
 
 		const consumed = pending.splice(0, Infinity);
-		externalResolve?.(executor(consumed));
+		try {
+			externalResolve?.(executor(consumed));
+		} catch (error: unknown) {
+			externalReject?.(error);
+		}
 
 		return pendingExecution;
 	};