add retry for db

1 files changed, 34 insertions, 2 deletions

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}")