Class TimeoutMiddleware

ai/ai.ts:2388–2471

class TimeoutMiddleware implements Middleware {
  private configManagerPromise: Promise<ConfigManager>;

  constructor(configManagerPromise: Promise<ConfigManager>) {
    this.configManagerPromise = configManagerPromise;
  }

  async process<T>(
    input: T,
    next: (input: T, token?: AbortToken) => Promise<any>,
    token?: AbortToken,
  ): Promise<any> {
    const config = (await this.configManagerPromise).getConfig();
    const timeoutMs = config.timeout * 1000;

    const timeoutController = new AbortController();
    const timeoutId = setTimeout(
      () => timeoutController.abort(`请求超时: ${timeoutMs}ms`),
      timeoutMs,
    );

    try {
      const combined = this.combine(timeoutController, token);
      combined.signal.addEventListener("abort", () => clearTimeout(timeoutId), {
        once: true,
      });
      return await next(input, combined);
    } finally {
      clearTimeout(timeoutId);
    }
  }

  private combine(
    timeoutController: AbortController,
    externalToken?: AbortToken,
  ): AbortToken {
    const controller = new AbortController();

    if (timeoutController.signal.aborted)
      controller.abort(timeoutController.signal.reason);
    else
      timeoutController.signal.addEventListener(
        "abort",
        () => controller.abort(timeoutController.signal.reason),
        {
          once: true,
        },
      );

    if (externalToken) {
      if (externalToken.aborted) controller.abort(externalToken.reason);
      else
        externalToken.signal.addEventListener(
          "abort",
          () => controller.abort(externalToken.reason),
          {
            once: true,
          },
        );