trigger.dev 3.0.0-beta.0 → 3.0.0-beta.2

package/dist/workers/prod/worker-facade.js

@@ -0,0 +1,183 @@
+// src/workers/prod/worker-facade.ts
+import {
+  ProdChildToWorkerMessages,
+  ProdWorkerToChildMessages,
+  TaskExecutor,
+  ZodIpcConnection,
+  DurableClock,
+  clock
+} from "@trigger.dev/core/v3";
+import "source-map-support/register.js";
+import {
+  ConsoleInterceptor,
+  OtelTaskLogger,
+  ProdRuntimeManager,
+  TaskRunErrorCodes,
+  TriggerTracer,
+  logger,
+  runtime
+} from "@trigger.dev/core/v3";
+
+// package.json
+var version = "3.0.0-beta.2";
+
+// src/workers/prod/worker-facade.ts
+__WORKER_SETUP__;
+__IMPORTED_PROJECT_CONFIG__;
+var otelTracer = tracingSDK.getTracer("trigger-prod-worker", version);
+var otelLogger = tracingSDK.getLogger("trigger-prod-worker", version);
+var durableClock = new DurableClock();
+clock.setGlobalClock(durableClock);
+var tracer = new TriggerTracer({ tracer: otelTracer, logger: otelLogger });
+var consoleInterceptor = new ConsoleInterceptor(otelLogger);
+var otelTaskLogger = new OtelTaskLogger({
+  logger: otelLogger,
+  tracer,
+  level: "info"
+});
+logger.setGlobalTaskLogger(otelTaskLogger);
+var TaskFileImports = {};
+var TaskFiles = {};
+__TASKS__;
+function getTasks() {
+  const result = [];
+  for (const [importName, taskFile] of Object.entries(TaskFiles)) {
+    const fileImports = TaskFileImports[importName];
+    for (const [exportName, task] of Object.entries(fileImports ?? {})) {
+      if (task.__trigger) {
+        result.push({
+          id: task.__trigger.id,
+          exportName,
+          packageVersion: task.__trigger.packageVersion,
+          filePath: taskFile.filePath,
+          queue: task.__trigger.queue,
+          retry: task.__trigger.retry,
+          machine: task.__trigger.machine,
+          fns: task.__trigger.fns
+        });
+      }
+    }
+  }
+  return result;
+}
+function getTaskMetadata() {
+  const result = getTasks();
+  return result.map((task) => {
+    const { fns, ...metadata } = task;
+    return metadata;
+  });
+}
+var tasks = getTasks();
+runtime.registerTasks(tasks);
+var taskExecutors = /* @__PURE__ */ new Map();
+for (const task of tasks) {
+  taskExecutors.set(
+    task.id,
+    new TaskExecutor(task, {
+      tracer,
+      tracingSDK,
+      consoleInterceptor,
+      projectConfig: __PROJECT_CONFIG__,
+      importedConfig,
+      handleErrorFn: handleError
+    })
+  );
+}
+var _execution;
+var _isRunning = false;
+var zodIpc = new ZodIpcConnection({
+  listenSchema: ProdWorkerToChildMessages,
+  emitSchema: ProdChildToWorkerMessages,
+  process,
+  handlers: {
+    EXECUTE_TASK_RUN: async ({ execution, traceContext, metadata }, sender) => {
+      if (_isRunning) {
+        console.error("Worker is already running a task");
+        await sender.send("TASK_RUN_COMPLETED", {
+          execution,
+          result: {
+            ok: false,
+            id: execution.attempt.id,
+            error: {
+              type: "INTERNAL_ERROR",
+              code: TaskRunErrorCodes.TASK_ALREADY_RUNNING
+            }
+          }
+        });
+        return;
+      }
+      process.title = `trigger-prod-worker: ${execution.task.id} ${execution.run.id}`;
+      const executor = taskExecutors.get(execution.task.id);
+      if (!executor) {
+        console.error(`Could not find executor for task ${execution.task.id}`);
+        await sender.send("TASK_RUN_COMPLETED", {
+          execution,
+          result: {
+            ok: false,
+            id: execution.attempt.id,
+            error: {
+              type: "INTERNAL_ERROR",
+              code: TaskRunErrorCodes.COULD_NOT_FIND_EXECUTOR
+            }
+          }
+        });
+        return;
+      }
+      try {
+        _execution = execution;
+        _isRunning = true;
+        const result = await executor.execute(execution, metadata, traceContext);
+        return sender.send("TASK_RUN_COMPLETED", {
+          execution,
+          result
+        });
+      } finally {
+        _execution = void 0;
+        _isRunning = false;
+      }
+    },
+    TASK_RUN_COMPLETED_NOTIFICATION: async ({ completion, execution }) => {
+      prodRuntimeManager.resumeTask(completion, execution);
+    },
+    WAIT_COMPLETED_NOTIFICATION: async () => {
+      prodRuntimeManager.resumeAfterRestore();
+    },
+    CLEANUP: async ({ flush, kill }, sender) => {
+      if (kill) {
+        await tracingSDK.flush();
+        await sender.send("READY_TO_DISPOSE", void 0);
+      } else {
+        if (flush) {
+          await tracingSDK.flush();
+        }
+      }
+    }
+  }
+});
+var prodRuntimeManager = new ProdRuntimeManager(zodIpc, {
+  waitThresholdInMs: parseInt(process.env.TRIGGER_RUNTIME_WAIT_THRESHOLD_IN_MS ?? "30000", 10)
+});
+runtime.setGlobalRuntimeManager(prodRuntimeManager);
+zodIpc.send("TASKS_READY", { tasks: getTaskMetadata() }).catch((err) => {
+  console.error("Failed to send TASKS_READY message", err);
+});
+process.title = "trigger-prod-worker";
+async function asyncHeartbeat(initialDelayInSeconds = 30, intervalInSeconds = 5) {
+  async function _doHeartbeat() {
+    while (true) {
+      if (_isRunning && _execution) {
+        try {
+          await zodIpc.send("TASK_HEARTBEAT", { id: _execution.attempt.id });
+        } catch (err) {
+          console.error("Failed to send HEARTBEAT message", err);
+        }
+      }
+      await new Promise((resolve) => setTimeout(resolve, 1e3 * intervalInSeconds));
+    }
+  }
+  await new Promise((resolve) => setTimeout(resolve, 1e3 * initialDelayInSeconds));
+  return _doHeartbeat();
+}
+asyncHeartbeat(5).catch((err) => {
+  console.error("Failed to start asyncHeartbeat", err);
+});

package/dist/workers/prod/worker-setup.js

@@ -0,0 +1,35 @@
+// src/workers/prod/worker-setup.ts
+import { Resource } from "@opentelemetry/resources";
+import {
+  SemanticInternalAttributes,
+  TracingSDK
+} from "@trigger.dev/core/v3";
+__SETUP_IMPORTED_PROJECT_CONFIG__;
+var tracingSDK = new TracingSDK({
+  url: process.env.OTEL_EXPORTER_OTLP_ENDPOINT ?? "http://0.0.0.0:4318",
+  resource: new Resource({
+    [SemanticInternalAttributes.CLI_VERSION]: "3.0.0"
+  }),
+  instrumentations: setupImportedConfig?.instrumentations ?? [],
+  diagLogLevel: process.env.OTEL_LOG_LEVEL ?? "none"
+});
+process.on("uncaughtException", (error, origin) => {
+  process.send?.({
+    type: "EVENT",
+    message: {
+      type: "UNCAUGHT_EXCEPTION",
+      payload: {
+        error: {
+          name: error.name,
+          message: error.message,
+          stack: error.stack
+        },
+        origin
+      },
+      version: "v1"
+    }
+  });
+});
+export {
+  tracingSDK
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "trigger.dev",
-  "version": "3.0.0-beta.0",
+  "version": "3.0.0-beta.2",
   "description": "A Command-Line Interface for Trigger.dev (v3) projects",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -54,7 +54,7 @@
     "typescript": "^5.3.3",
     "vitest": "^0.34.4",
     "xdg-app-paths": "^8.3.0",
-    "@trigger.dev/core-apps": "3.0.0-beta.0",
+    "@trigger.dev/core-apps": "3.0.0-beta.1",
     "@trigger.dev/tsconfig": "0.0.0"
   },
   "watch": {
@@ -75,7 +75,7 @@
     "@opentelemetry/sdk-trace-base": "^1.22.0",
     "@opentelemetry/sdk-trace-node": "^1.22.0",
     "@opentelemetry/semantic-conventions": "^1.22.0",
-    "@trigger.dev/core": "^3.0.0-beta.0",
+    "@trigger.dev/core": "^3.0.0-beta.1",
     "@types/degit": "^2.8.3",
     "chalk": "^5.2.0",
     "chokidar": "^3.5.3",