From 6ddbe77958db1f97ea64ba650a49fa47888d3f40 Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Mon, 28 Apr 2025 00:13:49 -0600
Subject: add retry for db

---
 server/services/process_bids.py | 36 ++++++++++++++++++++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

(limited to 'server/services/process_bids.py')

diff --git a/server/services/process_bids.py b/server/services/process_bids.py
index 40a31cd..edac780 100644
--- a/server/services/process_bids.py
+++ b/server/services/process_bids.py
@@ -5,6 +5,8 @@ from models.bid import Bid
 from models.auth import User
 from gridstatus import ISONE, NYISO, MISO
 from zoneinfo import ZoneInfo
+from sqlalchemy.exc import OperationalError
+import time
 
 MARKET_ISOS = {
     "ISONE": ISONE(),
@@ -87,7 +89,22 @@ def process_bids():
                 bid.pnl = None
                 print(f"Bid {bid.id}: Failed (price too low)")
 
-            db.commit()
+            max_retries = 3
+            for attempt in range(max_retries):
+                try:
+                    db.commit()
+                    break
+                except OperationalError as e:
+                    if "database is locked" in str(e).lower():
+                        if attempt < max_retries - 1:
+                            time.sleep(2)
+                            continue
+                        else:
+                            raise Exception(
+                                "Database is locked, and retries exhausted."
+                            )
+                    else:
+                        raise e
 
         except Exception as e:
             print(f"Error processing submitted bid {bid.id}: {e}")
@@ -122,7 +139,22 @@ def process_bids():
             bid.pnl = pnl
             print(f"Bid {bid.id} Settled | PnL: ${pnl:.2f}")
 
-            db.commit()
+            max_retries = 3
+            for attempt in range(max_retries):
+                try:
+                    db.commit()
+                    break
+                except OperationalError as e:
+                    if "database is locked" in str(e).lower():
+                        if attempt < max_retries - 1:
+                            time.sleep(2)
+                            continue
+                        else:
+                            raise Exception(
+                                "Database is locked, and retries exhausted."
+                            )
+                    else:
+                        raise e
 
         except Exception as e:
             print(f"Error settling executed bid {bid.id}: {e}")
-- 
cgit v1.2.3