File size: 1,561 Bytes
d491dc1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | import json
import logging
from typing import Optional, Dict, Any
logger = logging.getLogger("hyperflow.store_context")
class StoreContextCache:
"""
Cross-request store context memory layer.
Persists dark store context (profitability score, PSI status, last forecast) in Redis.
"""
KEY_PREFIX = "hf:store:"
TTL_SECONDS = 300
def __init__(self, redis_client=None):
self.redis = redis_client
self._local_cache: Dict[str, dict] = {}
async def set_context(self, store_id: str, context: dict) -> None:
key = f"{self.KEY_PREFIX}{store_id}"
self._local_cache[key] = context
if self.redis:
try:
if hasattr(self.redis, "setex") and callable(self.redis.setex):
await self.redis.setex(key, self.TTL_SECONDS, json.dumps(context))
elif hasattr(self.redis, "set"):
self.redis.set(key, json.dumps(context), ex=self.TTL_SECONDS)
except Exception as e:
logger.warning(f"[StoreContextCache] Redis write error: {e}")
async def get_context(self, store_id: str) -> Optional[dict]:
key = f"{self.KEY_PREFIX}{store_id}"
if self.redis:
try:
if hasattr(self.redis, "get"):
raw = self.redis.get(key)
if raw:
return json.loads(raw)
except Exception as e:
logger.warning(f"[StoreContextCache] Redis read error: {e}")
return self._local_cache.get(key)
|