wehub-resource-sync
76 lines
2.7 KiB
TypeScript
76 lines
2.7 KiB
TypeScript
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;
|
|
}
|