Function waitForPromiseWithTimeout

src/node/runtime/sshConnectionPool.ts:122–167

(
  promise: Promise<T>,
  timeoutMs: number,
  abortSignal?: AbortSignal,
  timeoutError?: Error
)

}

async function waitForPromiseWithTimeout<T>(
  promise: Promise<T>,
  timeoutMs: number,
  abortSignal?: AbortSignal,
  timeoutError?: Error
): Promise<T> {
  if (abortSignal?.aborted) {
    throw new Error("Operation aborted");
  }

  return await new Promise<T>((resolve, reject) => {
    let settled = false;

    const cleanup = () => {
      clearTimeout(timer);
      abortSignal?.removeEventListener("abort", onAbort);
    };

    const finish = (handler: () => void) => {
      if (settled) {
        return;
      }
      settled = true;
      cleanup();
      handler();
    };

    const onAbort = () => {
      finish(() => reject(new Error("Operation aborted")));
    };

    const timer = setTimeout(() => {
      finish(() => reject(timeoutError ?? new Error("Operation timed out")));
    }, timeoutMs);

    abortSignal?.addEventListener("abort", onAbort);
    promise.then(
      (value) => {
        finish(() => resolve(value));
      },
      (error) => {
        finish(() => reject(error instanceof Error ? error : new Error(String(error))));
      }
    );
  });
}

/**
 * SSH Connection Pool