chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,75 @@
+import type { PrismaReplicaClient } from "~/db.server";
+import {
+  $replica as defaultLegacyReplica,
+  runOpsNewPrisma as defaultNewPrimary,
+  runOpsNewReplica as defaultNewClient,
+  runOpsSplitReadEnabled as defaultSplitReadEnabled,
+} from "~/db.server";
+import { readThroughRun } from "~/v3/runOpsMigration/readThrough.server";
+
+type ResolveWaitpointDeps = {
+  newClient?: PrismaReplicaClient;
+  legacyReplica?: PrismaReplicaClient;
+  newPrimary?: PrismaReplicaClient;
+  splitEnabled?: boolean;
+  isPastRetention?: (id: string) => boolean;
+};
+
+// Safe defaults matching the deps `complete`/`callback` pass, so a bare caller still fans
+// out to the dedicated run-ops replica (NEW-resident waitpoints) before control-plane.
+export type ResolveWaitpointReadThroughDefaults = {
+  newClient: PrismaReplicaClient;
+  legacyReplica: PrismaReplicaClient;
+  newPrimary: PrismaReplicaClient;
+  splitEnabled: boolean;
+};
+
+const productionDefaults: ResolveWaitpointReadThroughDefaults = {
+  newClient: defaultNewClient,
+  legacyReplica: defaultLegacyReplica,
+  newPrimary: defaultNewPrimary as unknown as PrismaReplicaClient,
+  splitEnabled: defaultSplitReadEnabled,
+};
+
+export async function resolveWaitpointThroughReadThrough<T>(opts: {
+  waitpointId: string;
+  environmentId: string;
+  read: (client: PrismaReplicaClient) => Promise<T | null>;
+  deps?: ResolveWaitpointDeps;
+  defaults?: ResolveWaitpointReadThroughDefaults;
+}): Promise<T | null> {
+  const defaults = opts.defaults ?? productionDefaults;
+
+  const splitEnabled = opts.deps?.splitEnabled ?? defaults.splitEnabled;
+
+  const result = await readThroughRun({
+    runId: opts.waitpointId,
+    environmentId: opts.environmentId,
+    readNew: (client) => opts.read(client),
+    readLegacy: (replica) => opts.read(replica),
+    deps: {
+      splitEnabled,
+      newClient: opts.deps?.newClient ?? defaults.newClient,
+      legacyReplica: opts.deps?.legacyReplica ?? defaults.legacyReplica,
+      isPastRetention: opts.deps?.isPastRetention,
+    },
+  });
+
+  if (result.source === "new" || result.source === "legacy-replica") {
+    return result.value;
+  }
+  // past-retention is an intentional not-found: the token is gone.
+  if (result.source === "past-retention") {
+    return null;
+  }
+
+  // Read-your-writes fallback for a token completed immediately after mint, before it replicated:
+  // re-read from the run-ops PRIMARY only. We deliberately never read the control-plane/legacy
+  // primary here (that is the load the replica-only read-through exists to shed), so a legacy-resident
+  // token that misses its replica stays a miss and the caller retries, rather than adding primary load.
+  const fromNewPrimary = await opts.read(opts.deps?.newPrimary ?? defaults.newPrimary);
+  if (fromNewPrimary != null) {
+    return fromNewPrimary;
+  }
+  return null;
+}