@supergrowthai/tq 1.0.7 → 1.0.9

package/dist/index.mjs

@@ -43,13 +43,14 @@ var __callDispose = (stack, error, hasError) => {
   };
   return next();
 };
-import { I, M, P } from "./PrismaAdapter-Z2vLslDJ.js";
+import { I, M, P } from "./PrismaAdapter-BD8f3tk9.js";
 import { getEnvironmentQueueName } from "@supergrowthai/mq";
 import { L as Logger, a as LogLevel } from "./client-BAiCkZv7.js";
 import { tId } from "./utils/task-id-gen.mjs";
 import { Actions } from "./core/Actions.mjs";
 import { AsyncActions } from "./core/async/AsyncActions.mjs";
 import moment from "moment";
+import * as os from "os";
 import { AsyncTaskManager } from "./core/async/AsyncTaskManager.mjs";
 class LockManager {
   constructor(cacheProvider, options) {
@@ -167,10 +168,19 @@ class TaskStore {
     await this.databaseAdapter.markTasksAsExecuted(tasks);
   }
   /**
-   * Marks tasks as failed and increments retry count
+   * Marks tasks as failed — uses upsert to handle store_on_failure tasks
+   * that were never persisted to DB
    */
   async markTasksAsFailed(tasks) {
-    await this.databaseAdapter.markTasksAsFailed(tasks);
+    const tasksWithStatus = tasks.map((task) => ({
+      ...task,
+      status: "failed",
+      execution_stats: {
+        ...task.execution_stats || {},
+        failed_at: /* @__PURE__ */ new Date()
+      }
+    }));
+    await this.databaseAdapter.upsertTasks(tasksWithStatus);
   }
   /**
    * Marks tasks as successful/completed
@@ -182,8 +192,9 @@ class TaskStore {
    * Marks tasks as ignored with proper task context
    */
   async markTasksAsIgnored(tasks) {
-    const updatedTasks = tasks.map((task) => ({
+    const tasksWithStatus = tasks.map((task) => ({
       ...task,
+      status: "ignored",
       execution_stats: {
         ...task.execution_stats || {},
         error: "No executor found for task type",
@@ -191,7 +202,7 @@ class TaskStore {
         ignored_at: /* @__PURE__ */ new Date()
       }
     }));
-    await this.databaseAdapter.markTasksAsIgnored(updatedTasks);
+    await this.databaseAdapter.upsertTasks(tasksWithStatus);
   }
   /**
    * Updates multiple tasks with specific updates
@@ -220,19 +231,11 @@ class TaskStore {
     await this.databaseAdapter.cleanupTasks(twoDaysAgo, now);
   }
   /**
-   * Updates tasks for retry with new execution time and retry count
+   * Updates tasks for retry — uses upsert to handle store_on_failure tasks
+   * that were never persisted to DB
    */
   async updateTasksForRetry(tasks) {
-    const updates = tasks.map((task) => ({
-      id: task.id,
-      updates: {
-        execute_at: task.execute_at,
-        status: task.status,
-        execution_stats: task.execution_stats,
-        updated_at: /* @__PURE__ */ new Date()
-      }
-    }));
-    await this.databaseAdapter.updateTasks(updates);
+    await this.databaseAdapter.upsertTasks(tasks);
   }
 }
 var commonjsGlobal = typeof globalThis !== "undefined" ? globalThis : typeof window !== "undefined" ? window : typeof global !== "undefined" ? global : typeof self !== "undefined" ? self : {};
@@ -953,7 +956,7 @@ class TaskHandler {
     this.queueStats = /* @__PURE__ */ new Map();
     this.logger = new Logger("TaskHandler", LogLevel.INFO);
     this.config = config || {};
-    this.workerId = `${(void 0)()}-${process.pid}-${Date.now()}`;
+    this.workerId = `${os.hostname()}-${process.pid}-${Date.now()}`;
     this.workerStartedAt = /* @__PURE__ */ new Date();
     this.taskStore = new TaskStore(databaseAdapter);
     this.taskRunner = new TaskRunner(
@@ -1176,6 +1179,7 @@ class TaskHandler {
       }
       await this.postProcessTasks({ failedTasks, newTasks, successTasks }).catch((err) => {
         this.logger.error("Failed to postProcessTasks", err);
+        throw err;
       });
       if (!this.queueStats.has(streamName)) {
         this.queueStats.set(streamName, {
@@ -1231,7 +1235,7 @@ class TaskHandler {
   buildWorkerInfo() {
     return {
       worker_id: this.workerId,
-      hostname: (void 0)(),
+      hostname: os.hostname(),
       pid: process.pid,
       started_at: this.workerStartedAt,
       enabled_queues: this.enabledQueues
@@ -1328,6 +1332,28 @@ class TaskHandler {
         }
         const matureTasks = await this.taskStore.getMatureTasks(Date.now());
         this.logger.debug(`Found ${matureTasks.length} mature tasks to process`);
+        if (matureTasks.length > 0) {
+          try {
+            const tasksWithIds = matureTasks.filter((t) => t.id);
+            const dedupKeys = tasksWithIds.map((t) => `mature_dedup:${t.id}`);
+            if (dedupKeys.length > 0) {
+              const previousPickers = await this.cacheAdapter.mget(...dedupKeys);
+              for (let i = 0; i < dedupKeys.length; i++) {
+                if (previousPickers[i]) {
+                  const task = tasksWithIds[i];
+                  this.logger.warn(`DUPLICATE_MATURE_PICK: task ${task.id} (${task.type}) already picked by ${previousPickers[i]}`);
+                }
+              }
+              const pipeline = this.cacheAdapter.pipeline();
+              for (const key of dedupKeys) {
+                pipeline.set(key, INSTANCE_ID, 120);
+              }
+              await pipeline.exec();
+            }
+          } catch (err) {
+            this.logger.warn(`Duplicate pick detection failed (best-effort): ${err}`);
+          }
+        }
         await this.addTasks(matureTasks);
       } catch (error) {
         this.logger.error(`Error processing tasks: ${error}`);