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