Add retries to updateStats

polymerdao · Apr 25, 2024 · eed50bf · eed50bf
1 parent 7f31fb9
commit eed50bf
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/src/stats.ts b/src/stats.ts
@@ -1,4 +1,6 @@
 import { Context } from "@/generated";
+import retry from "async-retry";
+import { defaultRetryOpts } from "./retry";
 
 export enum StatName {
   SendPackets = 'SendPackets',
@@ -15,15 +17,18 @@ export enum StatName {
 }
 
 export async function updateStats(context: Context, id: StatName) {
-  await context.db.Stat.upsert({
-    id: id,
-    create: {
-      val: 1,
-    },
-    update: ({current}) => {
-      return {
-        val: current.val + 1
-      }
-    }
-  });
+  await retry(async bail => {
+      await context.db.Stat.upsert({
+        id: id,
+        create: {
+          val: 1,
+        },
+        update: ({current}) => {
+          return {
+            val: current.val + 1
+          }
+        }
+      });
+    }, defaultRetryOpts
+  )
 }