@orpc/shared 1.1.1 → 1.2.0

package/dist/index.d.mts

@@ -10,11 +10,17 @@ declare function splitInHalf<T>(arr: readonly T[]): [T[], T[]];
 
 type AnyFunction = (...args: any[]) => any;
 declare function once<T extends () => any>(fn: T): () => ReturnType<T>;
+declare function sequential<A extends any[], R>(fn: (...args: A) => Promise<R>): (...args: A) => Promise<R>;
 
 type OmitChainMethodDeep<T extends object, K extends keyof any> = {
     [P in keyof Omit<T, K>]: T[P] extends AnyFunction ? ((...args: Parameters<T[P]>) => OmitChainMethodDeep<ReturnType<T[P]>, K>) : T[P];
 };
 
+declare class SequentialIdGenerator {
+    private nextId;
+    generate(): number;
+}
+
 type SetOptional<T, K extends keyof T> = Omit<T, K> & Partial<Pick<T, K>>;
 type IntersectPick<T, U> = Pick<T, keyof T & keyof U>;
 type PromiseWithError<T, TError> = Promise<T> & {
@@ -66,6 +72,10 @@ declare function onFinish<T, TOptions extends {
 declare function intercept<TOptions extends InterceptableOptions, TResult, TError>(interceptors: Interceptor<TOptions, TResult, TError>[], options: NoInfer<TOptions>, main: NoInfer<(options: TOptions) => Promisable<TResult>>): Promise<TResult>;
 
 declare function isAsyncIteratorObject(maybe: unknown): maybe is AsyncIteratorObject<any, any, any>;
+interface CreateAsyncIteratorObjectCleanupFn {
+    (reason: 'return' | 'throw' | 'next' | 'dispose'): Promise<void>;
+}
+declare function createAsyncIteratorObject<T, TReturn, TNext>(next: () => Promise<IteratorResult<T, TReturn>>, cleanup: CreateAsyncIteratorObjectCleanupFn): AsyncIteratorObject<T, TReturn, TNext> & AsyncGenerator<T, TReturn, TNext>;
 
 declare function parseEmptyableJSON(text: string | null | undefined): unknown;
 declare function stringifyJSON<T>(value: T): undefined extends T ? undefined | string : string;
@@ -86,8 +96,25 @@ declare function isTypescriptObject(value: unknown): value is object & Record<Pr
 declare function clone<T>(value: T): T;
 declare function get(object: object, path: readonly string[]): unknown;
 
+interface AsyncIdQueueCloseOptions {
+    id?: number;
+    reason?: Error;
+}
+declare class AsyncIdQueue<T> {
+    private readonly openIds;
+    private readonly items;
+    private readonly pendingPulls;
+    get length(): number;
+    open(id: number): void;
+    isOpen(id: number): boolean;
+    push(id: number, item: T): void;
+    pull(id: number): Promise<T>;
+    close({ id, reason }?: AsyncIdQueueCloseOptions): void;
+    assertOpen(id: number): void;
+}
+
 type Value<T, TArgs extends any[] = []> = T | ((...args: TArgs) => Promisable<T>);
 declare function value<T, TArgs extends any[]>(value: Value<T, TArgs>, ...args: NoInfer<TArgs>): Promise<T extends Value<infer U, any> ? U : never>;
 
-export { clone, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, splitInHalf, stringifyJSON, toArray, value };
-export type { AnyFunction, InterceptableOptions, Interceptor, InterceptorOptions, IntersectPick, MaybeOptionalOptions, OmitChainMethodDeep, OnFinishState, PromiseWithError, Registry, Segment, SetOptional, ThrowableError, Value };
+export { AsyncIdQueue, SequentialIdGenerator, clone, createAsyncIteratorObject, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, sequential, splitInHalf, stringifyJSON, toArray, value };
+export type { AnyFunction, AsyncIdQueueCloseOptions, CreateAsyncIteratorObjectCleanupFn, InterceptableOptions, Interceptor, InterceptorOptions, IntersectPick, MaybeOptionalOptions, OmitChainMethodDeep, OnFinishState, PromiseWithError, Registry, Segment, SetOptional, ThrowableError, Value };

package/dist/index.d.ts

@@ -10,11 +10,17 @@ declare function splitInHalf<T>(arr: readonly T[]): [T[], T[]];
 
 type AnyFunction = (...args: any[]) => any;
 declare function once<T extends () => any>(fn: T): () => ReturnType<T>;
+declare function sequential<A extends any[], R>(fn: (...args: A) => Promise<R>): (...args: A) => Promise<R>;
 
 type OmitChainMethodDeep<T extends object, K extends keyof any> = {
     [P in keyof Omit<T, K>]: T[P] extends AnyFunction ? ((...args: Parameters<T[P]>) => OmitChainMethodDeep<ReturnType<T[P]>, K>) : T[P];
 };
 
+declare class SequentialIdGenerator {
+    private nextId;
+    generate(): number;
+}
+
 type SetOptional<T, K extends keyof T> = Omit<T, K> & Partial<Pick<T, K>>;
 type IntersectPick<T, U> = Pick<T, keyof T & keyof U>;
 type PromiseWithError<T, TError> = Promise<T> & {
@@ -66,6 +72,10 @@ declare function onFinish<T, TOptions extends {
 declare function intercept<TOptions extends InterceptableOptions, TResult, TError>(interceptors: Interceptor<TOptions, TResult, TError>[], options: NoInfer<TOptions>, main: NoInfer<(options: TOptions) => Promisable<TResult>>): Promise<TResult>;
 
 declare function isAsyncIteratorObject(maybe: unknown): maybe is AsyncIteratorObject<any, any, any>;
+interface CreateAsyncIteratorObjectCleanupFn {
+    (reason: 'return' | 'throw' | 'next' | 'dispose'): Promise<void>;
+}
+declare function createAsyncIteratorObject<T, TReturn, TNext>(next: () => Promise<IteratorResult<T, TReturn>>, cleanup: CreateAsyncIteratorObjectCleanupFn): AsyncIteratorObject<T, TReturn, TNext> & AsyncGenerator<T, TReturn, TNext>;
 
 declare function parseEmptyableJSON(text: string | null | undefined): unknown;
 declare function stringifyJSON<T>(value: T): undefined extends T ? undefined | string : string;
@@ -86,8 +96,25 @@ declare function isTypescriptObject(value: unknown): value is object & Record<Pr
 declare function clone<T>(value: T): T;
 declare function get(object: object, path: readonly string[]): unknown;
 
+interface AsyncIdQueueCloseOptions {
+    id?: number;
+    reason?: Error;
+}
+declare class AsyncIdQueue<T> {
+    private readonly openIds;
+    private readonly items;
+    private readonly pendingPulls;
+    get length(): number;
+    open(id: number): void;
+    isOpen(id: number): boolean;
+    push(id: number, item: T): void;
+    pull(id: number): Promise<T>;
+    close({ id, reason }?: AsyncIdQueueCloseOptions): void;
+    assertOpen(id: number): void;
+}
+
 type Value<T, TArgs extends any[] = []> = T | ((...args: TArgs) => Promisable<T>);
 declare function value<T, TArgs extends any[]>(value: Value<T, TArgs>, ...args: NoInfer<TArgs>): Promise<T extends Value<infer U, any> ? U : never>;
 
-export { clone, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, splitInHalf, stringifyJSON, toArray, value };
-export type { AnyFunction, InterceptableOptions, Interceptor, InterceptorOptions, IntersectPick, MaybeOptionalOptions, OmitChainMethodDeep, OnFinishState, PromiseWithError, Registry, Segment, SetOptional, ThrowableError, Value };
+export { AsyncIdQueue, SequentialIdGenerator, clone, createAsyncIteratorObject, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, sequential, splitInHalf, stringifyJSON, toArray, value };
+export type { AnyFunction, AsyncIdQueueCloseOptions, CreateAsyncIteratorObjectCleanupFn, InterceptableOptions, Interceptor, InterceptorOptions, IntersectPick, MaybeOptionalOptions, OmitChainMethodDeep, OnFinishState, PromiseWithError, Registry, Segment, SetOptional, ThrowableError, Value };

package/dist/index.mjs

@@ -23,6 +23,26 @@ function once(fn) {
     return result;
   };
 }
+function sequential(fn) {
+  let lastOperationPromise = Promise.resolve();
+  return (...args) => {
+    return lastOperationPromise = lastOperationPromise.catch(() => {
+    }).then(() => {
+      return fn(...args);
+    });
+  };
+}
+
+class SequentialIdGenerator {
+  nextId = 0;
+  generate() {
+    if (this.nextId === Number.MAX_SAFE_INTEGER) {
+      this.nextId = 0;
+      return Number.MAX_SAFE_INTEGER;
+    }
+    return this.nextId++;
+  }
+}
 
 function onStart(callback) {
   return async (options, ...rest) => {
@@ -82,6 +102,62 @@ function isAsyncIteratorObject(maybe) {
   }
   return Symbol.asyncIterator in maybe && typeof maybe[Symbol.asyncIterator] === "function";
 }
+function createAsyncIteratorObject(next, cleanup) {
+  let isExecuteComplete = false;
+  let isDone = false;
+  const iterator = {
+    next: sequential(async () => {
+      if (isDone) {
+        return { done: true, value: void 0 };
+      }
+      try {
+        const result = await next();
+        if (result.done) {
+          isDone = true;
+        }
+        return result;
+      } catch (err) {
+        isDone = true;
+        throw err;
+      } finally {
+        if (isDone && !isExecuteComplete) {
+          isExecuteComplete = true;
+          await cleanup("next");
+        }
+      }
+    }),
+    async return(value) {
+      isDone = true;
+      if (!isExecuteComplete) {
+        isExecuteComplete = true;
+        await cleanup("return");
+      }
+      return { done: true, value };
+    },
+    async throw(err) {
+      isDone = true;
+      if (!isExecuteComplete) {
+        isExecuteComplete = true;
+        await cleanup("throw");
+      }
+      throw err;
+    },
+    /**
+     * asyncDispose symbol only available in esnext, we should fallback to Symbol.for('asyncDispose')
+     */
+    async [Symbol.asyncDispose ?? Symbol.for("asyncDispose")]() {
+      isDone = true;
+      if (!isExecuteComplete) {
+        isExecuteComplete = true;
+        await cleanup("dispose");
+      }
+    },
+    [Symbol.asyncIterator]() {
+      return iterator;
+    }
+  };
+  return iterator;
+}
 
 function parseEmptyableJSON(text) {
   if (!text) {
@@ -142,6 +218,82 @@ function get(object, path) {
   return current;
 }
 
+class AsyncIdQueue {
+  openIds = /* @__PURE__ */ new Set();
+  items = /* @__PURE__ */ new Map();
+  pendingPulls = /* @__PURE__ */ new Map();
+  get length() {
+    return this.openIds.size;
+  }
+  open(id) {
+    this.openIds.add(id);
+  }
+  isOpen(id) {
+    return this.openIds.has(id);
+  }
+  push(id, item) {
+    this.assertOpen(id);
+    const pending = this.pendingPulls.get(id);
+    if (pending?.length) {
+      pending.shift()[0](item);
+      if (pending.length === 0) {
+        this.pendingPulls.delete(id);
+      }
+    } else {
+      const items = this.items.get(id);
+      if (items) {
+        items.push(item);
+      } else {
+        this.items.set(id, [item]);
+      }
+    }
+  }
+  async pull(id) {
+    this.assertOpen(id);
+    const items = this.items.get(id);
+    if (items?.length) {
+      const item = items.shift();
+      if (items.length === 0) {
+        this.items.delete(id);
+      }
+      return item;
+    }
+    return new Promise((resolve, reject) => {
+      const waitingPulls = this.pendingPulls.get(id);
+      const pending = [resolve, reject];
+      if (waitingPulls) {
+        waitingPulls.push(pending);
+      } else {
+        this.pendingPulls.set(id, [pending]);
+      }
+    });
+  }
+  close({ id, reason } = {}) {
+    if (id === void 0) {
+      this.pendingPulls.forEach((pendingPulls, id2) => {
+        pendingPulls.forEach(([, reject]) => {
+          reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id2}] was closed or aborted while waiting for pulling.`));
+        });
+      });
+      this.pendingPulls.clear();
+      this.openIds.clear();
+      this.items.clear();
+      return;
+    }
+    this.pendingPulls.get(id)?.forEach(([, reject]) => {
+      reject(reason ?? new Error(`[AsyncIdQueue] Queue[${id}] was closed or aborted while waiting for pulling.`));
+    });
+    this.pendingPulls.delete(id);
+    this.openIds.delete(id);
+    this.items.delete(id);
+  }
+  assertOpen(id) {
+    if (!this.isOpen(id)) {
+      throw new Error(`[AsyncIdQueue] Cannot access queue[${id}] because it is not open or aborted.`);
+    }
+  }
+}
+
 function value(value2, ...args) {
   if (typeof value2 === "function") {
     return value2(...args);
@@ -149,4 +301,4 @@ function value(value2, ...args) {
   return value2;
 }
 
-export { clone, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, splitInHalf, stringifyJSON, toArray, value };
+export { AsyncIdQueue, SequentialIdGenerator, clone, createAsyncIteratorObject, findDeepMatches, get, intercept, isAsyncIteratorObject, isObject, isTypescriptObject, onError, onFinish, onStart, onSuccess, once, parseEmptyableJSON, resolveMaybeOptionalOptions, sequential, splitInHalf, stringifyJSON, toArray, value };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@orpc/shared",
   "type": "module",
-  "version": "1.1.1",
+  "version": "1.2.0",
   "license": "MIT",
   "homepage": "https://orpc.unnoq.com",
   "repository": {