From ad152ff4143fcc804cb5e587c80065a4c48101f6 Mon Sep 17 00:00:00 2001
From: Rohit Rajan <rohit.rajan031101@gmail.com>
Date: Mon, 8 Dec 2025 18:49:46 +0530
Subject: [PATCH] chore: register schedule worker

---
 server/src/schedule-worker.ts  | 5 +++--
 server/src/storage/schedule.ts | 3 +++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/server/src/schedule-worker.ts b/server/src/schedule-worker.ts
index 63c7bad4..75c578c2 100644
--- a/server/src/schedule-worker.ts
+++ b/server/src/schedule-worker.ts
@@ -13,7 +13,7 @@ if (!process.env.DB_USER || !process.env.DB_PASSWORD || !process.env.DB_HOST ||
 
 const pgBossConnectionString = `postgresql://${process.env.DB_USER}:${encodeURIComponent(process.env.DB_PASSWORD)}@${process.env.DB_HOST}:${process.env.DB_PORT}/${process.env.DB_NAME}`;
 
-const pgBoss = new PgBoss({
+export const pgBoss = new PgBoss({
   connectionString: pgBossConnectionString,
   max: 3,
   expireInHours: 23,
@@ -87,8 +87,9 @@ async function registerScheduledWorkflowWorker() {
 
 /**
  * Register a worker for a specific queue
+ * Exported to allow dynamic registration when new schedules are created
  */
-async function registerWorkerForQueue(queueName: string) {
+export async function registerWorkerForQueue(queueName: string) {
   try {
     if (registeredQueues.has(queueName)) {
       return;
diff --git a/server/src/storage/schedule.ts b/server/src/storage/schedule.ts
index 919f64b2..ed524d9b 100644
--- a/server/src/storage/schedule.ts
+++ b/server/src/storage/schedule.ts
@@ -5,6 +5,7 @@
 import { v4 as uuid } from 'uuid';
 import logger from '../logger';
 import { pgBossClient } from './pgboss';
+import { registerWorkerForQueue } from '../schedule-worker';
 
 /**
  * Utility function to schedule a cron job using PgBoss
@@ -28,6 +29,8 @@ export async function scheduleWorkflow(id: string, userId: string,  cronExpressi
       { tz: timezone }
     );
 
+    await registerWorkerForQueue(queueName);
+
     logger.log('info', `Scheduled workflow job for robot ${id}`);
   } catch (error: unknown) {
     const errorMessage = error instanceof Error ? error.message : String(error);