wehub-resource-sync
82 lines
2.3 KiB
TypeScript
82 lines
2.3 KiB
TypeScript
import { RunEngineVersion, type RuntimeEnvironmentType } from "@trigger.dev/database";
|
|
import { $replica } from "~/db.server";
|
|
import {
|
|
findCurrentWorkerFromEnvironment,
|
|
getCurrentWorkerDeploymentEngineVersion,
|
|
} from "./models/workerDeployment.server";
|
|
|
|
// Co-locate the per-env run-ops residency/mint decision next to the
|
|
// engine-version decision. determineEngineVersion is intentionally left untouched so its
|
|
// read-only callers (presenters, admin routes, pauseQueue) never pay the mint flag read.
|
|
export { resolveRunIdMintKind, type RunIdMintKind } from "./runOpsMigration/runOpsMintKind.server";
|
|
|
|
type Environment = {
|
|
id: string;
|
|
type: RuntimeEnvironmentType;
|
|
project: {
|
|
id: string;
|
|
engine: RunEngineVersion;
|
|
};
|
|
};
|
|
|
|
export async function determineEngineVersion({
|
|
environment,
|
|
workerVersion,
|
|
engineVersion: version,
|
|
}: {
|
|
environment: Environment;
|
|
workerVersion?: string;
|
|
engineVersion?: RunEngineVersion;
|
|
}): Promise<RunEngineVersion> {
|
|
if (version) {
|
|
return version;
|
|
}
|
|
|
|
// If the project is V1, then none of the background workers are running V2
|
|
if (environment.project.engine === RunEngineVersion.V1) {
|
|
return "V1";
|
|
}
|
|
|
|
/**
|
|
* The project has V2 enabled so it *could* be V2.
|
|
*/
|
|
|
|
// A specific worker version is requested
|
|
if (workerVersion) {
|
|
const worker = await $replica.backgroundWorker.findUnique({
|
|
select: {
|
|
engine: true,
|
|
},
|
|
where: {
|
|
projectId_runtimeEnvironmentId_version: {
|
|
projectId: environment.project.id,
|
|
runtimeEnvironmentId: environment.id,
|
|
version: workerVersion,
|
|
},
|
|
},
|
|
});
|
|
|
|
if (!worker) {
|
|
throw new Error(`Worker not found: environment: ${environment.id} version: ${workerVersion}`);
|
|
}
|
|
|
|
return worker.engine;
|
|
}
|
|
|
|
// Dev: use the latest BackgroundWorker
|
|
if (environment.type === "DEVELOPMENT") {
|
|
const backgroundWorker = await findCurrentWorkerFromEnvironment(environment);
|
|
return backgroundWorker?.engine ?? "V1";
|
|
}
|
|
|
|
// Deployed: use the latest deployed BackgroundWorker
|
|
const currentDeploymentEngineVersion = await getCurrentWorkerDeploymentEngineVersion(
|
|
environment.id
|
|
);
|
|
if (currentDeploymentEngineVersion) {
|
|
return currentDeploymentEngineVersion;
|
|
}
|
|
|
|
return environment.project.engine;
|
|
}
|