Task & Run Hooks

[ericallam]

We should give users the ability to hook into the task and run lifecycle

import { prisma } from "./db";import { prisma } from "./db";
import { TriggerClient } from "@trigger.dev/sdk";

const client = new TriggerClient({
  id: "my-app",
  hooks: {
    onTaskCompleted: (task, job, payload, ctx) =>
      prisma.tasks.upsert({
        where: { id: task.idempotencyKey },
        create: { id: task.idempotencyKey, data: task.output },
        update: { data: task.output },
      }), // keep a local record of all tasks if you want to
    beforeTaskRun: async (task, job, payload, ctx) => {
      const existingTask = await prisma.tasks.findUnique({
        id: task.idempotencyKey,
      });

      return existingTask?.data; // Can keep a local cache of tasks and provide output
    },
    onTaskFailed: async (task, error, job, payload, ctx) => {
      // Do something when a task fails
    },
    onRunFinished: async (job, run, payload, ctx) => {
      // Do something when a run is completed/cancelled/failed
    },
    onRunCompleted: async (job, run, payload, ctx) => {
      // Do something when a run is completed
    },
    onRunCancelled: async (job, run, payload, ctx) => {
      // Do something when a run is cancelled
    },
    onRunFailed: async (job, run, payload, ctx) => {
      // Do something when a run is cancelled
    },
  },
});

// They can also be defined on at the job level
client.defineJob({
  id: "my-job",
  onTaskCompleted: (task, payload, ctx) => {},
  beforeTaskRun: async (task, payload, ctx) => {
  },
  onTaskFailed: async (task, error, payload, ctx) => {
    // Do something when a task fails
  },
  onRunFinished: async (run, payload, ctx) => {
    // Do something when a run is completed/cancelled/failed
  },
  onRunCompleted: async (run, payload, ctx) => {
    // Do something when a run is completed
  },
  onRunCancelled: async (run, payload, ctx) => {
    // Do something when a run is cancelled
  },
  onRunFailed: async (run, payload, ctx) => {
    // Do something when a run is cancelled
  },
});