Cache TB data: src/data/data_providers.py

src/data/data_providers.py

@@ -872,6 +872,8 @@ class EnergyService(BaseService):
     def __init__(self, tb_client: Any = None):
         self._tb = tb_client
         self._breaker = CircuitBreaker(threshold=3, cooldown_sec=300)
+        self._current_cache = TTLCache(ttl_seconds=300, redis_prefix="energy:")      # 5 min
+        self._daily_cache = TTLCache(ttl_seconds=900, redis_prefix="energy_daily:")   # 15 min
 
     def _client(self):
         if self._tb is None:
@@ -884,28 +886,36 @@ class EnergyService(BaseService):
     # ------------------------------------------------------------------
 
     def get_current(self) -> Dict[str, Any]:
-        """Latest power reading from the Plant asset."""
+        """Latest power reading from the Plant asset (5-min TTL cache)."""
+        cached = self._current_cache.get("current")
+        if cached is not None:
+            return cached
         if self._breaker.is_open:
             return {"error": "ThingsBoard circuit breaker open — retrying in 5 min"}
         try:
             vals = self._client().get_asset_latest("Plant", ["power", "production"])
             power_w = vals.get("power")
             self._breaker.record_success()
-            return {
+            result = {
                 "power_kw": round(power_w / 1000, 1) if power_w else None,
                 "source": "ThingsBoard Plant asset",
             }
+            self._current_cache.set("current", result)
+            return result
         except Exception as exc:
             self._breaker.record_failure()
             return {"error": f"Energy current failed: {exc}"}
 
     def get_daily_production(self, target_date: Optional[str] = None) -> Dict[str, Any]:
-        """Accumulated energy production for a single day (real TB data).
+        """Accumulated energy production for a single day (real TB data, 15-min TTL cache).
 
         Returns dict with daily_kwh, peak_hour, hourly_profile.
         """
         try:
             target = target_date or str(date.today())
+            cached = self._daily_cache.get(f"daily:{target}")
+            if cached is not None:
+                return cached
             day_start = pd.Timestamp(target, tz="UTC")
             day_end = day_start + pd.Timedelta(days=1)
 
@@ -943,7 +953,7 @@ class EnergyService(BaseService):
                     peak_kwh = kwh
                     peak_hour = h
 
-            return {
+            result = {
                 "date": target,
                 "daily_kwh": round(total_kwh, 1),
                 "peak_hour": peak_hour,
@@ -951,6 +961,8 @@ class EnergyService(BaseService):
                 "hourly_profile": hourly_profile,
                 "source": "ThingsBoard Plant asset",
             }
+            self._daily_cache.set(f"daily:{target}", result)
+            return result
         except Exception as exc:
             return {"date": target_date, "daily_kwh": None, "error": f"Energy fetch failed: {exc}"}