chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,79 @@
|
||||
import type { TypedResponse } from "@remix-run/server-runtime";
|
||||
import { json } from "@remix-run/server-runtime";
|
||||
import type { WaitForDurationResponseBody } from "@trigger.dev/core/v3";
|
||||
import { WaitForDurationRequestBody } from "@trigger.dev/core/v3";
|
||||
import { RunId } from "@trigger.dev/core/v3/isomorphic";
|
||||
|
||||
import { z } from "zod";
|
||||
import { prisma } from "~/db.server";
|
||||
import { logger } from "~/services/logger.server";
|
||||
import { createActionApiRoute } from "~/services/routeBuilders/apiBuilder.server";
|
||||
import { resolveIdempotencyKeyTTL } from "~/utils/idempotencyKeys.server";
|
||||
import { engine } from "~/v3/runEngine.server";
|
||||
import { runStore } from "~/v3/runStore.server";
|
||||
|
||||
const { action } = createActionApiRoute(
|
||||
{
|
||||
body: WaitForDurationRequestBody,
|
||||
params: z.object({
|
||||
runFriendlyId: z.string(),
|
||||
}),
|
||||
method: "POST",
|
||||
},
|
||||
async ({ authentication, body, params }): Promise<TypedResponse<WaitForDurationResponseBody>> => {
|
||||
const { runFriendlyId } = params;
|
||||
const runId = RunId.toId(runFriendlyId);
|
||||
|
||||
try {
|
||||
const run = await runStore.findRun(
|
||||
{
|
||||
id: runId,
|
||||
runtimeEnvironmentId: authentication.environment.id,
|
||||
},
|
||||
prisma
|
||||
);
|
||||
|
||||
if (!run) {
|
||||
throw new Response("You don't have permissions for this run", { status: 401 });
|
||||
}
|
||||
|
||||
const idempotencyKeyExpiresAt = body.idempotencyKeyTTL
|
||||
? resolveIdempotencyKeyTTL(body.idempotencyKeyTTL)
|
||||
: undefined;
|
||||
|
||||
const { waitpoint } = await engine.createDateTimeWaitpoint({
|
||||
// Co-locate the waitpoint with the run that blocks on it (run-ops split): a run-ops run lives
|
||||
// on the dedicated DB, but the minted waitpoint id is always a cuid, so without the run id
|
||||
// the waitpoint would route to the control-plane DB and the block edge would never resolve.
|
||||
runId: run.id,
|
||||
projectId: authentication.environment.project.id,
|
||||
environmentId: authentication.environment.id,
|
||||
completedAfter: body.date,
|
||||
idempotencyKey: body.idempotencyKey,
|
||||
idempotencyKeyExpiresAt: idempotencyKeyExpiresAt,
|
||||
});
|
||||
|
||||
const _waitResult = await engine.blockRunWithWaitpoint({
|
||||
runId: run.id,
|
||||
waitpoints: waitpoint.id,
|
||||
projectId: authentication.environment.project.id,
|
||||
organizationId: authentication.environment.organization.id,
|
||||
});
|
||||
|
||||
return json({
|
||||
waitUntil: body.date,
|
||||
waitpoint: {
|
||||
id: waitpoint.friendlyId,
|
||||
},
|
||||
});
|
||||
} catch (error) {
|
||||
logger.error("Failed to wait for duration dev", {
|
||||
environmentId: authentication.environment.id,
|
||||
error,
|
||||
});
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
export { action };
|
||||
Reference in New Issue
Block a user