OCT-1721: Cache often consecutive DB query.

golemfoundation · Jul 4, 2024 · 09a6e86 · 09a6e86
1 parent 61db329
commit 09a6e86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/app/extensions.py b/backend/app/extensions.py
@@ -11,7 +11,7 @@
     generate_cache_key,
 )
 
-from functools import wraps, partial
+from functools import partial
 import lru
 import threading
 import time

diff --git a/backend/app/infrastructure/database/pending_epoch_snapshot.py b/backend/app/infrastructure/database/pending_epoch_snapshot.py
@@ -4,7 +4,7 @@
 from typing_extensions import deprecated
 
 from app.infrastructure.database.models import PendingEpochSnapshot
-from app.extensions import db
+from app.extensions import db, cache
 from app import exceptions
 
 from decimal import Decimal
@@ -20,10 +20,12 @@ def get_by_epoch_num(epoch) -> PendingEpochSnapshot:
     return snapshot
 
 
+@cache.memoize(timeout=15)
 def get_by_epoch(epoch: int) -> Optional[PendingEpochSnapshot]:
     return PendingEpochSnapshot.query.filter_by(epoch=epoch).first()
 
 
+@cache.memoize(timeout=15)
 def get_last_snapshot() -> PendingEpochSnapshot:
     snapshot = (
         db.session.query(PendingEpochSnapshot)