bulltrackers-module 1.0.745 → 1.0.746

package/functions/computation-system-v2/framework/storage/StorageManager.js

@@ -347,15 +347,40 @@ class StorageManager {
         const table = 'computation_checkpoints';
         const fullTable = `\`${this.config.bigquery.projectId}.${this.config.bigquery.dataset}.${table}\``;
         try {
+            // FIX: Use subquery with ROW_NUMBER to find the TRUE latest state per computation.
+            // We only count it as a zombie if the LATEST row is 'running'.
+            // This ignores 'running' rows that have a newer (or same-time) 'completed' sibling.
             const query = `
                 SELECT computation_name, date, checkpoint_id, last_updated
-                FROM ${fullTable}
-                WHERE status = 'running' AND last_updated < TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL @minutes MINUTE)
+                FROM (
+                    SELECT 
+                        computation_name, 
+                        date, 
+                        checkpoint_id, 
+                        last_updated, 
+                        status,
+                        ROW_NUMBER() OVER (
+                            PARTITION BY computation_name, date 
+                            ORDER BY last_updated DESC,
+                            CASE status 
+                                WHEN 'completed' THEN 1 
+                                WHEN 'failed' THEN 2 
+                                ELSE 3 
+                            END ASC
+                        ) as rn
+                    FROM ${fullTable}
+                )
+                WHERE rn = 1 
+                  AND status = 'running' 
+                  AND last_updated < TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL @minutes MINUTE)
                 LIMIT 50
             `;
             const [rows] = await this.bigquery.query({ query, params: { minutes: minutesThreshold }, location: this.config.bigquery.location });
             return rows.map(r => ({ name: r.computation_name, date: r.date.value || r.date, checkpointId: r.checkpoint_id }));
-        } catch (e) { return []; }
+        } catch (e) { 
+            console.error(`[Storage] findZombies failed: ${e.message}`);
+            return []; 
+        }
     }
 
     async completeCheckpoint(dateStr, computationName, checkpointId) {
@@ -373,11 +398,21 @@ class StorageManager {
         const table = 'computation_checkpoints';
         const fullTable = `\`${this.config.bigquery.projectId}.${this.config.bigquery.dataset}.${table}\``;
         try {
+            // FIX: Added Tie-Breaker logic to ORDER BY
+            // If timestamps are identical, 'completed' (1) comes before 'failed' (2) before 'running' (3).
+            // This ensures we never accidentally pick a "running" row when a "completed" one exists at the exact same ms.
             const query = `
                 SELECT checkpoint_id, status, processed_count, last_entity_id, completed_batches, worker_instance_id, last_updated, attempts, code_hash, started_at
                 FROM ${fullTable}
                 WHERE date = @date AND computation_name = @computationName
-                ORDER BY last_updated DESC LIMIT 1
+                ORDER BY 
+                    last_updated DESC,
+                    CASE status 
+                        WHEN 'completed' THEN 1 
+                        WHEN 'failed' THEN 2 
+                        ELSE 3 
+                    END ASC
+                LIMIT 1
             `;
             const [rows] = await this.bigquery.query({ query, params: { date: dateStr, computationName }, location: this.config.bigquery.location });
             if (rows.length === 0) return null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bulltrackers-module",
-  "version": "1.0.745",
+  "version": "1.0.746",
   "description": "Helper Functions for Bulltrackers.",
   "main": "index.js",
   "files": [