Function createAsyncMessageQueue

src/common/utils/asyncMessageQueue.ts:24–72

()

 * ```
 */
export function createAsyncMessageQueue<T>(): {
  push: (msg: T) => void;
  iterate: () => AsyncGenerator<T>;
  end: () => void;
} {
  const queue: T[] = [];
  let resolveNext: (() => void) | null = null;
  let ended = false;

  const push = (msg: T) => {
    if (ended) return;
    queue.push(msg);
    // Signal that new messages are available
    if (resolveNext) {
      const resolve = resolveNext;
      resolveNext = null;
      resolve();
    }
  };

  async function* iterate(): AsyncGenerator<T> {
    while (!ended) {
      // Yield all queued messages synchronously (no async boundaries)
      // This ensures all messages from a batch are processed in the same
      // event loop tick, preventing premature renders
      while (queue.length > 0) {
        yield queue.shift()!;
      }
      // Wait for more messages
      await new Promise<void>((resolve) => {
        resolveNext = resolve;
      });
    }
    // Yield any remaining messages after end() is called
    while (queue.length > 0) {
      yield queue.shift()!;
    }
  }

  const end = () => {
    ended = true;
    // Wake up the iterator so it can exit
    if (resolveNext) {
      resolveNext();
    }
  };

  return { push, iterate, end };
}