wehub-resource-sync
97 lines
3.2 KiB
TypeScript
97 lines
3.2 KiB
TypeScript
import { packetRequiresOffloading, stringifyIO } from "@trigger.dev/core/v3";
|
|
import { type AuthenticatedEnvironment } from "~/services/apiAuth.server";
|
|
import { BaseService } from "./baseService.server";
|
|
import { env } from "~/env.server";
|
|
import { handleMetadataPacket } from "~/utils/packets";
|
|
import { type RunTemplateData } from "../taskRunTemplate";
|
|
|
|
export class TaskRunTemplateService extends BaseService {
|
|
public async call(environment: AuthenticatedEnvironment, data: RunTemplateData) {
|
|
const { triggerSource } = data;
|
|
|
|
switch (triggerSource) {
|
|
case "STANDARD": {
|
|
const packet = { data: JSON.stringify(data.payload), dataType: "application/json" };
|
|
|
|
const { needsOffloading } = packetRequiresOffloading(
|
|
packet,
|
|
env.TASK_PAYLOAD_OFFLOAD_THRESHOLD
|
|
);
|
|
|
|
if (needsOffloading) {
|
|
// we currently disallow large payloads in task run templates
|
|
throw new Error("Payload too large");
|
|
}
|
|
|
|
const metadataPacket = data.metadata
|
|
? handleMetadataPacket(
|
|
data.metadata,
|
|
"application/json",
|
|
env.TASK_RUN_METADATA_MAXIMUM_SIZE
|
|
)
|
|
: undefined;
|
|
|
|
const taskRunTemplate = await this._prisma.taskRunTemplate.create({
|
|
data: {
|
|
taskSlug: data.taskIdentifier,
|
|
triggerSource: "STANDARD",
|
|
label: data.label,
|
|
payload: packet.data,
|
|
payloadType: packet.dataType,
|
|
metadata: metadataPacket?.data,
|
|
metadataType: metadataPacket?.dataType,
|
|
queue: data.queue,
|
|
ttlSeconds: data.ttlSeconds,
|
|
concurrencyKey: data.concurrencyKey,
|
|
maxAttempts: data.maxAttempts,
|
|
maxDurationSeconds: data.maxDurationSeconds,
|
|
tags: data.tags ?? [],
|
|
machinePreset: data.machine,
|
|
projectId: environment.projectId,
|
|
organizationId: environment.organizationId,
|
|
},
|
|
});
|
|
|
|
return taskRunTemplate;
|
|
}
|
|
case "SCHEDULED": {
|
|
const payload = {
|
|
scheduleId: "sched_1234",
|
|
type: "IMPERATIVE",
|
|
timestamp: data.timestamp,
|
|
lastTimestamp: data.lastTimestamp,
|
|
timezone: data.timezone,
|
|
externalId: data.externalId,
|
|
upcoming: [],
|
|
};
|
|
const payloadPacket = await stringifyIO(payload);
|
|
|
|
const taskRunTemplate = await this._prisma.taskRunTemplate.create({
|
|
data: {
|
|
taskSlug: data.taskIdentifier,
|
|
triggerSource: "SCHEDULED",
|
|
label: data.label,
|
|
payload: payloadPacket.data,
|
|
payloadType: payloadPacket.dataType,
|
|
queue: data.queue,
|
|
ttlSeconds: data.ttlSeconds,
|
|
concurrencyKey: data.concurrencyKey,
|
|
maxAttempts: data.maxAttempts,
|
|
maxDurationSeconds: data.maxDurationSeconds,
|
|
tags: data.tags ?? [],
|
|
machinePreset: data.machine,
|
|
projectId: environment.projectId,
|
|
organizationId: environment.organizationId,
|
|
},
|
|
});
|
|
|
|
return taskRunTemplate;
|
|
}
|
|
default: {
|
|
triggerSource satisfies never;
|
|
throw new Error("Invalid trigger source");
|
|
}
|
|
}
|
|
}
|
|
}
|