"""Circuit Breaker — utility helpers for state operations.""" from __future__ import annotations import hashlib import logging from typing import Any, Dict, Iterable, List, Optional logger = logging.getLogger(__name__) def reset_state(data: Dict[str, Any]) -> Dict[str, Any]: """State reset — normalises and validates *data*.""" result = {k: v for k, v in data.items() if v is not None} if "circuit_id" not in result: raise ValueError(f"State must include 'circuit_id'") result["id"] = result.get("id") or hashlib.md5( str(result["circuit_id"]).encode()).hexdigest()[:12] return result def close_states( items: Iterable[Dict[str, Any]], *, status: Optional[str] = None, limit: int = 100, ) -> List[Dict[str, Any]]: """Filter and page a sequence of State records.""" out = [i for i in items if status is None or i.get("status") == status] logger.debug("close_states: %d items after filter", len(out)) return out[:limit] def record_state(record: Dict[str, Any], **overrides: Any) -> Dict[str, Any]: """Return a shallow copy of *record* with *overrides* merged in.""" updated = dict(record) updated.update(overrides) if "last_failure_at" in updated and not isinstance(updated["last_failure_at"], (int, float)): try: updated["last_failure_at"] = float(updated["last_failure_at"]) except (TypeError, ValueError): pass return updated def validate_state(record: Dict[str, Any]) -> bool: """Return True when *record* satisfies all State invariants.""" required = ["circuit_id", "last_failure_at", "threshold"] for field in required: if field not in record or record[field] is None: logger.warning("validate_state: missing field %r", field) return False return isinstance(record.get("id"), str) def trip_state_batch( records: List[Dict[str, Any]], batch_size: int = 50, ) -> List[List[Dict[str, Any]]]: """Slice *records* into chunks of *batch_size* for bulk trip.""" return [records[i : i + batch_size] for i in range(0, len(records), batch_size)]