Initial public release

hydra/core/position_tracker.py

@@ -0,0 +1,49 @@
+import json
+from typing import Optional
+
+from hydra.logging.setup import get_logger
+
+logger = get_logger(__name__)
+
+POSITIONS_KEY = "hydra:positions"
+
+
+class PositionTracker:
+    def __init__(self, redis_client):
+        self._redis = redis_client
+
+    def update(self, market: str, symbol: str, qty: float, avg_price: float, side: str, leverage: int = 1, mark_price: float | None = None) -> None:
+        key = f"{POSITIONS_KEY}:{market}:{symbol}"
+        data = {
+            "qty": qty,
+            "avg_price": avg_price,
+            "side": side,
+            "market": market,
+            "symbol": symbol,
+            "leverage": leverage,
+            "mark_price": mark_price or avg_price,
+        }
+        self._redis.set(key, json.dumps(data))
+        logger.info("position_updated", market=market, symbol=symbol, qty=qty, leverage=leverage)
+
+    def get(self, market: str, symbol: str) -> Optional[dict]:
+        key = f"{POSITIONS_KEY}:{market}:{symbol}"
+        raw = self._redis.get(key)
+        return json.loads(raw) if raw else None
+
+    def get_all(self) -> list[dict]:
+        keys = self._redis.keys(f"{POSITIONS_KEY}:*")
+        result = []
+        for k in keys:
+            raw = self._redis.get(k)
+            if raw:
+                result.append(json.loads(raw))
+        return result
+
+    def clear(self, market: str, symbol: str) -> None:
+        key = f"{POSITIONS_KEY}:{market}:{symbol}"
+        self._redis.delete(key)
+        logger.info("position_cleared", market=market, symbol=symbol)
+
+    async def snapshot(self) -> dict:
+        return {"positions": self.get_all()}