Function runStep

src/server/mcp-shutdown.ts:56–105

(
  name: string,
  timeoutMs: number,
  operation: () => Promise<T>,
)

}

async function runStep<T>(
  name: string,
  timeoutMs: number,
  operation: () => Promise<T>,
): Promise<ShutdownStepOutcome<T>> {
  const startedAt = Date.now();
  let timeoutHandle: NodeJS.Timeout | null = null;

  try {
    const timeoutPromise = new Promise<RunStepRaceOutcome<T>>((resolve) => {
      timeoutHandle = setTimeout(() => resolve({ kind: 'timed_out' }), timeoutMs);
      timeoutHandle.unref?.();
    });

    const operationOutcome = operation()
      .then((value): RunStepRaceOutcome<T> => ({ kind: 'value', value }))
      .catch(
        (error): RunStepRaceOutcome<T> => ({
          kind: 'error',
          error: toErrorMessage(error),
        }),
      );
    const outcome = await Promise.race([operationOutcome, timeoutPromise]);

    if (outcome.kind === 'timed_out') {
      return {
        status: 'timed_out',
        durationMs: Date.now() - startedAt,
      };
    }

    if (outcome.kind === 'error') {
      return {
        status: 'failed',
        durationMs: Date.now() - startedAt,
        error: outcome.error,
      };
    }

    return {
      status: 'completed',
      durationMs: Date.now() - startedAt,
      value: outcome.value,
    };
  } finally {
    if (timeoutHandle) {
      clearTimeout(timeoutHandle);
    }
  }
}

const FAILURE_REASONS: ReadonlySet<McpShutdownReason> = new Set([
  'startup-failure',