@trpc/server 11.0.0-rc.589 → 11.0.0-rc.590

package/src/unstable-core-do-not-import/stream/utils/asyncIterable.ts

@@ -27,6 +27,7 @@ export async function* withCancel<T>(
 interface TakeWithGraceOptions {
   count: number;
   gracePeriodMs: number;
+  onCancel?: () => void;
 }
 
 /**
@@ -36,7 +37,7 @@ interface TakeWithGraceOptions {
  */
 export async function* takeWithGrace<T>(
   iterable: AsyncIterable<T>,
-  { count, gracePeriodMs }: TakeWithGraceOptions,
+  { count, gracePeriodMs, onCancel }: TakeWithGraceOptions,
 ): AsyncGenerator<T> {
   const iterator = iterable[Symbol.asyncIterator]();
   const timer = createPromiseTimer(gracePeriodMs);
@@ -53,7 +54,7 @@ export async function* takeWithGrace<T>(
       }
       yield result.value;
       if (--count === 0) {
-        timer.start();
+        timer.start().promise.then(onCancel, noop);
       }
     }
   } finally {

package/src/unstable-core-do-not-import/stream/utils/promiseTimer.ts

@@ -17,7 +17,7 @@ export function createPromiseTimer(ms: number) {
 
   function start(): PromiseTimer {
     if (timeout != null) {
-      throw new Error("PromiseTimer already started.");
+      throw new Error('PromiseTimer already started.');
     }
     timeout = setTimeout(deferred.resolve, ms);
     return timer;

package/src/unstable-core-do-not-import/utils.ts

@@ -68,3 +68,51 @@ export function noop(): void {}
 export function identity<T>(it: T): T {
   return it;
 }
+
+/**
+ * Generic runtime assertion function. Throws, if the condition is not `true`.
+ *
+ * Can be used as a slightly less dangerous variant of type assertions. Code
+ * mistakes would be revealed at runtime then (hopefully during testing).
+ */
+export function assert(
+  condition: boolean,
+  msg = 'no additional info',
+): asserts condition {
+  if (!condition) {
+    throw new Error(`AssertionError: ${msg}`);
+  }
+}
+
+export function sleep(ms = 0): Promise<void> {
+  return new Promise<void>((res) => setTimeout(res, ms));
+}
+
+/**
+ * Ponyfill for
+ * [`AbortSignal.any`](https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal/any_static).
+ */
+export function abortSignalsAnyPonyfill(signals: AbortSignal[]): AbortSignal {
+  if (typeof AbortSignal.any === 'function') {
+    return AbortSignal.any(signals);
+  }
+
+  const ac = new AbortController();
+
+  for (const signal of signals) {
+    if (signal.aborted) {
+      trigger();
+    } else if (!ac.signal.aborted) {
+      signal.addEventListener('abort', trigger, { once: true });
+    }
+  }
+
+  return ac.signal;
+
+  function trigger() {
+    ac.abort();
+    for (const signal of signals) {
+      signal.removeEventListener('abort', trigger);
+    }
+  }
+}