feat: toggle crons to enabled/disabled

hatchet-dev · Apr 24, 2024 · 519167e · 519167e
1 parent d29e90b
commit 519167e
Show file tree

Hide file tree

Showing 7 changed files with 19 additions and 7 deletions.
diff --git a/internal/repository/prisma/dbsqlc/models.go b/internal/repository/prisma/dbsqlc/models.go
diff --git a/internal/repository/prisma/dbsqlc/schema.sql b/internal/repository/prisma/dbsqlc/schema.sql
@@ -597,7 +597,8 @@ CREATE TABLE "WorkflowTriggerCronRef" (
     "parentId" UUID NOT NULL,
     "cron" TEXT NOT NULL,
     "tickerId" UUID,
-    "input" JSONB
+    "input" JSONB,
+    "enabled" BOOLEAN NOT NULL DEFAULT true
 );
 
 -- CreateTable

diff --git a/internal/repository/prisma/dbsqlc/tickers.sql b/internal/repository/prisma/dbsqlc/tickers.sql
@@ -144,11 +144,12 @@ active_cron_schedules AS (
     JOIN 
         latest_workflow_versions l ON versions."workflowId" = l."workflowId" AND versions."order" = l.max_order
     WHERE
-        "tickerId" IS NULL 
+        "enabled" = TRUE AND
+        ("tickerId" IS NULL 
         OR NOT EXISTS (
             SELECT 1 FROM "Ticker" WHERE "id" = cronSchedule."tickerId" AND "isActive" = true AND "lastHeartbeatAt" >= NOW() - INTERVAL '10 seconds'
         )
-        OR "tickerId" = @tickerId::uuid
+        OR "tickerId" = @tickerId::uuid)
     FOR UPDATE SKIP LOCKED
 )
 UPDATE

diff --git a/internal/repository/prisma/dbsqlc/tickers.sql.go b/internal/repository/prisma/dbsqlc/tickers.sql.go
diff --git a/internal/repository/prisma/dbsqlc/workflows.sql.go b/internal/repository/prisma/dbsqlc/workflows.sql.go
diff --git a/prisma/migrations/20240424091046_v0_21_9/migration.sql b/prisma/migrations/20240424091046_v0_21_9/migration.sql
@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "WorkflowTriggerCronRef" ADD COLUMN     "enabled" BOOLEAN NOT NULL DEFAULT true;
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
@@ -415,6 +415,9 @@ model WorkflowTriggerCronRef {
   // the cron expression
   cron String
 
+  // whether this cron is enabled or not
+  enabled Boolean @default(true)
+
   // the assigned ticker
   ticker   Ticker? @relation(fields: [tickerId], references: [id])
   tickerId String? @db.Uuid