checking before decrementing pending count

tensorlakeai · Aug 3, 2024 · 15ed42e · 15ed42e
1 parent c40ca1e
commit 15ed42e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/crates/indexify_internal_api/src/lib.rs b/crates/indexify_internal_api/src/lib.rs
@@ -107,12 +107,17 @@ impl TaskAnalytics {
 
     pub fn success(&mut self) {
         self.successful_tasks += 1;
-        self.pending_tasks -= 1;
+        // This is for upgrade path from older versions
+        if self.pending_tasks > 0 {
+            self.pending_tasks -= 1;
+        }
     }
 
     pub fn fail(&mut self) {
         self.failed_tasks += 1;
-        self.pending_tasks -= 1;
+        if self.pending_tasks > 0 {
+            self.pending_tasks -= 1;
+        }
     }
 }