fix(budget): potential div by zero

edgeandnode · Sep 6, 2023 · 11cdbaf · 11cdbaf
1 parent fa2b38a
commit 11cdbaf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/graph-gateway/src/budgets.rs b/graph-gateway/src/budgets.rs
@@ -101,9 +101,9 @@ impl Actor {
     }
 
     fn revise_budget(&mut self) {
-        let surplus = self
-            .query_fees_target
-            .saturating_sub(self.recent_fees / USD::try_from(self.recent_query_count).unwrap());
+        let surplus = self.query_fees_target.saturating_sub(
+            self.recent_fees / USD::try_from(self.recent_query_count.max(1)).unwrap(),
+        );
         self.recent_fees = USD::zero();
         self.recent_query_count = 0;