wehub-resource-sync
68 lines
1.9 KiB
TypeScript
68 lines
1.9 KiB
TypeScript
import type { TaskRunError } from "@trigger.dev/core/v3/schemas";
|
|
import { TaskRunErrorCodes } from "@trigger.dev/core/v3/schemas";
|
|
import type { TaskRun } from "@trigger.dev/database";
|
|
import type { AuthenticatedEnvironment } from "~/services/apiAuth.server";
|
|
import { marqs } from "../marqs/index.server";
|
|
|
|
export type EnqueueRunOptions = {
|
|
env: AuthenticatedEnvironment;
|
|
run: TaskRun;
|
|
dependentRun?: { queue: string; id: string };
|
|
};
|
|
|
|
export type EnqueueRunResult =
|
|
| {
|
|
ok: true;
|
|
}
|
|
| {
|
|
ok: false;
|
|
error: TaskRunError;
|
|
};
|
|
|
|
export async function enqueueRun({
|
|
env,
|
|
run,
|
|
dependentRun,
|
|
}: EnqueueRunOptions): Promise<EnqueueRunResult> {
|
|
// If this is a triggerAndWait or batchTriggerAndWait,
|
|
// we need to add the parent run to the reserve concurrency set
|
|
// to free up concurrency for the children to run
|
|
// In the case of a recursive queue, reserving concurrency can fail, which means there is a deadlock and we need to fail the run
|
|
|
|
// TODO: reserveConcurrency can fail because of a deadlock, we need to handle that case
|
|
const wasEnqueued = await marqs.enqueueMessage(
|
|
env,
|
|
run.queue,
|
|
run.id,
|
|
{
|
|
type: "EXECUTE",
|
|
taskIdentifier: run.taskIdentifier,
|
|
projectId: env.projectId,
|
|
environmentId: env.id,
|
|
environmentType: env.type,
|
|
},
|
|
run.concurrencyKey ?? undefined,
|
|
run.queueTimestamp ?? undefined,
|
|
dependentRun
|
|
? { messageId: dependentRun.id, recursiveQueue: dependentRun.queue === run.queue }
|
|
: undefined
|
|
);
|
|
|
|
if (!wasEnqueued) {
|
|
const error = {
|
|
type: "INTERNAL_ERROR",
|
|
code: TaskRunErrorCodes.RECURSIVE_WAIT_DEADLOCK,
|
|
message: `This run will never execute because it was triggered recursively and the task has no remaining concurrency available`,
|
|
} satisfies TaskRunError;
|
|
|
|
return {
|
|
ok: false,
|
|
error,
|
|
};
|
|
}
|
|
|
|
return {
|
|
ok: true,
|
|
};
|
|
}
|