t3-code-android-nightly/.repos/alchemy-effect/benchmark/container/wrangler/worker.ts

/// <reference types="@cloudflare/workers-types" />

import { Container, getContainer } from "@cloudflare/containers";

interface Env {
  BENCH_CONTAINER: DurableObjectNamespace<BenchContainer>;
  CRASH_CONTAINER: DurableObjectNamespace<CrashContainer>;
}

/**
 * Plain `@cloudflare/containers` container — no Alchemy involved. The DO proxies
 * `fetch` to the Bun HTTP server on port 8080; the first `fetch` blocks through
 * the container's cold start (the helper calls `startAndWaitForPorts`
 * internally), so timing a single `fetch` measures start → reachable.
 */
export class BenchContainer extends Container<Env> {
  defaultPort = 8080;
  sleepAfter = "2m";
}

/**
 * Crash-loop control: its image exits immediately, so `fetch` can never reach
 * it. Native surfaces the fatal crash from its monitor and returns a 5xx; we
 * time how fast that happens to baseline Alchemy's fail-fast behaviour.
 */
export class CrashContainer extends Container<Env> {
  defaultPort = 8080;
  sleepAfter = "2m";
}

export default {
  async fetch(request, env): Promise<Response> {
    const url = new URL(request.url);

    // `/start?name=X` boots the container instance named `X` and reports the
    // cold-start-to-reachable latency measured inside the Worker.
    if (url.pathname === "/start") {
      const name = url.searchParams.get("name") ?? "default";
      const container = getContainer(env.BENCH_CONTAINER, name);
      const start = Date.now();
      try {
        const res = await container.fetch(new Request("http://container/"));
        await res.text();
        return Response.json({ ms: Date.now() - start, status: res.status });
      } catch (err) {
        return Response.json(
          { error: String(err), ms: Date.now() - start },
          { status: 500 },
        );
      }
    }

    // `/crashloop?name=X` attempts to reach a container that crash-loops and
    // reports how long until the failure surfaces (native returns a 5xx). `ok`
    // is true only on a 2xx — always false here.
    if (url.pathname === "/crashloop") {
      const name = url.searchParams.get("name") ?? "default";
      const container = getContainer(env.CRASH_CONTAINER, name);
      const start = Date.now();
      try {
        const res = await container.fetch(new Request("http://container/"));
        await res.text();
        return Response.json({ ms: Date.now() - start, ok: res.ok });
      } catch (err) {
        return Response.json({
          ms: Date.now() - start,
          ok: false,
          error: String(err),
        });
      }
    }

    return new Response("ok");
  },
} satisfies ExportedHandler<Env>;