initial commit

cli.py

@@ -0,0 +1,61 @@
+"""Circuit Breaker — utility helpers for circuit operations."""
+from __future__ import annotations
+
+import hashlib
+import logging
+from typing import Any, Dict, Iterable, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+def trip_circuit(data: Dict[str, Any]) -> Dict[str, Any]:
+    """Circuit trip — normalises and validates *data*."""
+    result = {k: v for k, v in data.items() if v is not None}
+    if "last_failure_at" not in result:
+        raise ValueError(f"Circuit must include 'last_failure_at'")
+    result["id"] = result.get("id") or hashlib.md5(
+        str(result["last_failure_at"]).encode()).hexdigest()[:12]
+    return result
+
+
+def record_circuits(
+    items: Iterable[Dict[str, Any]],
+    *,
+    status: Optional[str] = None,
+    limit: int = 100,
+) -> List[Dict[str, Any]]:
+    """Filter and page a sequence of Circuit records."""
+    out = [i for i in items if status is None or i.get("status") == status]
+    logger.debug("record_circuits: %d items after filter", len(out))
+    return out[:limit]
+
+
+def close_circuit(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 "circuit_id" in updated and not isinstance(updated["circuit_id"], (int, float)):
+        try:
+            updated["circuit_id"] = float(updated["circuit_id"])
+        except (TypeError, ValueError):
+            pass
+    return updated
+
+
+def validate_circuit(record: Dict[str, Any]) -> bool:
+    """Return True when *record* satisfies all Circuit invariants."""
+    required = ["last_failure_at", "circuit_id", "threshold"]
+    for field in required:
+        if field not in record or record[field] is None:
+            logger.warning("validate_circuit: missing field %r", field)
+            return False
+    return isinstance(record.get("id"), str)
+
+
+def open_circuit_batch(
+    records: List[Dict[str, Any]],
+    batch_size: int = 50,
+) -> List[List[Dict[str, Any]]]:
+    """Slice *records* into chunks of *batch_size* for bulk open."""
+    return [records[i : i + batch_size]
+            for i in range(0, len(records), batch_size)]

encoder.py

@@ -0,0 +1,61 @@
+"""Circuit Breaker — utility helpers for probe operations."""
+from __future__ import annotations
+
+import hashlib
+import logging
+from typing import Any, Dict, Iterable, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+def trip_probe(data: Dict[str, Any]) -> Dict[str, Any]:
+    """Probe trip — normalises and validates *data*."""
+    result = {k: v for k, v in data.items() if v is not None}
+    if "threshold" not in result:
+        raise ValueError(f"Probe must include 'threshold'")
+    result["id"] = result.get("id") or hashlib.md5(
+        str(result["threshold"]).encode()).hexdigest()[:12]
+    return result
+
+
+def close_probes(
+    items: Iterable[Dict[str, Any]],
+    *,
+    status: Optional[str] = None,
+    limit: int = 100,
+) -> List[Dict[str, Any]]:
+    """Filter and page a sequence of Probe records."""
+    out = [i for i in items if status is None or i.get("status") == status]
+    logger.debug("close_probes: %d items after filter", len(out))
+    return out[:limit]
+
+
+def record_probe(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 "failure_count" in updated and not isinstance(updated["failure_count"], (int, float)):
+        try:
+            updated["failure_count"] = float(updated["failure_count"])
+        except (TypeError, ValueError):
+            pass
+    return updated
+
+
+def validate_probe(record: Dict[str, Any]) -> bool:
+    """Return True when *record* satisfies all Probe invariants."""
+    required = ["threshold", "failure_count", "last_failure_at"]
+    for field in required:
+        if field not in record or record[field] is None:
+            logger.warning("validate_probe: missing field %r", field)
+            return False
+    return isinstance(record.get("id"), str)
+
+
+def reset_probe_batch(
+    records: List[Dict[str, Any]],
+    batch_size: int = 50,
+) -> List[List[Dict[str, Any]]]:
+    """Slice *records* into chunks of *batch_size* for bulk reset."""
+    return [records[i : i + batch_size]
+            for i in range(0, len(records), batch_size)]

helpers.py

@@ -0,0 +1,61 @@
+"""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)]

manager.py

@@ -0,0 +1,78 @@
+"""Circuit Breaker — Failure service layer."""
+from __future__ import annotations
+
+import logging
+from typing import Any, Dict, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+class CircuitManager:
+    """Business-logic service for Failure operations in Circuit Breaker."""
+
+    def __init__(
+        self,
+        repo: Any,
+        events: Optional[Any] = None,
+    ) -> None:
+        self._repo   = repo
+        self._events = events
+        logger.debug("CircuitManager started")
+
+    def close(
+        self, payload: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """Execute the close workflow for a new Failure."""
+        if "last_failure_at" not in payload:
+            raise ValueError("Missing required field: last_failure_at")
+        record = self._repo.insert(
+            payload["last_failure_at"], payload.get("circuit_id"),
+            **{k: v for k, v in payload.items()
+              if k not in ("last_failure_at", "circuit_id")}
+        )
+        if self._events:
+            self._events.emit("failure.closed", record)
+        return record
+
+    def open(self, rec_id: str, **changes: Any) -> Dict[str, Any]:
+        """Apply *changes* to a Failure and emit a change event."""
+        ok = self._repo.update(rec_id, **changes)
+        if not ok:
+            raise KeyError(f"Failure {rec_id!r} not found")
+        updated = self._repo.fetch(rec_id)
+        if self._events:
+            self._events.emit("failure.opend", updated)
+        return updated
+
+    def trip(self, rec_id: str) -> None:
+        """Remove a Failure and emit a removal event."""
+        ok = self._repo.delete(rec_id)
+        if not ok:
+            raise KeyError(f"Failure {rec_id!r} not found")
+        if self._events:
+            self._events.emit("failure.tripd", {"id": rec_id})
+
+    def search(
+        self,
+        last_failure_at: Optional[Any] = None,
+        status: Optional[str] = None,
+        limit:  int = 50,
+    ) -> List[Dict[str, Any]]:
+        """Search failures by *last_failure_at* and/or *status*."""
+        filters: Dict[str, Any] = {}
+        if last_failure_at is not None:
+            filters["last_failure_at"] = last_failure_at
+        if status is not None:
+            filters["status"] = status
+        rows, _ = self._repo.query(filters, limit=limit)
+        logger.debug("search failures: %d hits", len(rows))
+        return rows
+
+    @property
+    def stats(self) -> Dict[str, int]:
+        """Quick summary of Failure counts by status."""
+        result: Dict[str, int] = {}
+        for status in ("active", "pending", "closed"):
+            _, count = self._repo.query({"status": status}, limit=0)
+            result[status] = count
+        return result

models.py

@@ -0,0 +1,85 @@
+"""Circuit Breaker — State repository."""
+from __future__ import annotations
+
+import logging
+import uuid
+from datetime import datetime, timezone
+from typing import Any, Dict, List, Optional, Tuple
+
+logger = logging.getLogger(__name__)
+
+
+class CircuitModels:
+    """Thin repository wrapper for State persistence in Circuit Breaker."""
+
+    TABLE = "states"
+
+    def __init__(self, db: Any) -> None:
+        self._db = db
+        logger.debug("CircuitModels bound to %s", db)
+
+    def insert(self, threshold: Any, circuit_id: Any, **kwargs: Any) -> str:
+        """Persist a new State row and return its generated ID."""
+        rec_id = str(uuid.uuid4())
+        row: Dict[str, Any] = {
+            "id":         rec_id,
+            "threshold": threshold,
+            "circuit_id": circuit_id,
+            "created_at": datetime.now(timezone.utc).isoformat(),
+            **kwargs,
+        }
+        self._db.insert(self.TABLE, row)
+        return rec_id
+
+    def fetch(self, rec_id: str) -> Optional[Dict[str, Any]]:
+        """Return the State row for *rec_id*, or None."""
+        return self._db.fetch(self.TABLE, rec_id)
+
+    def update(self, rec_id: str, **fields: Any) -> bool:
+        """Patch *fields* on an existing State row."""
+        if not self._db.exists(self.TABLE, rec_id):
+            return False
+        fields["updated_at"] = datetime.now(timezone.utc).isoformat()
+        self._db.update(self.TABLE, rec_id, fields)
+        return True
+
+    def delete(self, rec_id: str) -> bool:
+        """Hard-delete a State row; returns False if not found."""
+        if not self._db.exists(self.TABLE, rec_id):
+            return False
+        self._db.delete(self.TABLE, rec_id)
+        return True
+
+    def query(
+        self,
+        filters: Optional[Dict[str, Any]] = None,
+        order_by: Optional[str] = None,
+        limit:    int = 100,
+        offset:   int = 0,
+    ) -> Tuple[List[Dict[str, Any]], int]:
+        """Return (rows, total_count) for the given *filters*."""
+        rows  = self._db.select(self.TABLE, filters or {}, limit, offset)
+        total = self._db.count(self.TABLE, filters or {})
+        logger.debug("query states: %d/%d", len(rows), total)
+        return rows, total
+
+    def record_by_failure_count(
+        self, value: Any, limit: int = 50
+    ) -> List[Dict[str, Any]]:
+        """Fetch states filtered by *failure_count*."""
+        rows, _ = self.query({"failure_count": value}, limit=limit)
+        return rows
+
+    def bulk_insert(
+        self, records: List[Dict[str, Any]]
+    ) -> List[str]:
+        """Insert *records* in bulk and return their generated IDs."""
+        ids: List[str] = []
+        for rec in records:
+            rec_id = self.insert(
+                rec["threshold"], rec.get("circuit_id"),
+                **{k: v for k, v in rec.items() if k not in ("threshold", "circuit_id")}
+            )
+            ids.append(rec_id)
+        logger.info("bulk_insert states: %d rows", len(ids))
+        return ids

processor.py

@@ -0,0 +1,85 @@
+"""Circuit Breaker — State repository."""
+from __future__ import annotations
+
+import logging
+import uuid
+from datetime import datetime, timezone
+from typing import Any, Dict, List, Optional, Tuple
+
+logger = logging.getLogger(__name__)
+
+
+class CircuitProcessor:
+    """Thin repository wrapper for State persistence in Circuit Breaker."""
+
+    TABLE = "states"
+
+    def __init__(self, db: Any) -> None:
+        self._db = db
+        logger.debug("CircuitProcessor bound to %s", db)
+
+    def insert(self, circuit_id: Any, threshold: Any, **kwargs: Any) -> str:
+        """Persist a new State row and return its generated ID."""
+        rec_id = str(uuid.uuid4())
+        row: Dict[str, Any] = {
+            "id":         rec_id,
+            "circuit_id": circuit_id,
+            "threshold": threshold,
+            "created_at": datetime.now(timezone.utc).isoformat(),
+            **kwargs,
+        }
+        self._db.insert(self.TABLE, row)
+        return rec_id
+
+    def fetch(self, rec_id: str) -> Optional[Dict[str, Any]]:
+        """Return the State row for *rec_id*, or None."""
+        return self._db.fetch(self.TABLE, rec_id)
+
+    def update(self, rec_id: str, **fields: Any) -> bool:
+        """Patch *fields* on an existing State row."""
+        if not self._db.exists(self.TABLE, rec_id):
+            return False
+        fields["updated_at"] = datetime.now(timezone.utc).isoformat()
+        self._db.update(self.TABLE, rec_id, fields)
+        return True
+
+    def delete(self, rec_id: str) -> bool:
+        """Hard-delete a State row; returns False if not found."""
+        if not self._db.exists(self.TABLE, rec_id):
+            return False
+        self._db.delete(self.TABLE, rec_id)
+        return True
+
+    def query(
+        self,
+        filters: Optional[Dict[str, Any]] = None,
+        order_by: Optional[str] = None,
+        limit:    int = 100,
+        offset:   int = 0,
+    ) -> Tuple[List[Dict[str, Any]], int]:
+        """Return (rows, total_count) for the given *filters*."""
+        rows  = self._db.select(self.TABLE, filters or {}, limit, offset)
+        total = self._db.count(self.TABLE, filters or {})
+        logger.debug("query states: %d/%d", len(rows), total)
+        return rows, total
+
+    def probe_by_last_failure_at(
+        self, value: Any, limit: int = 50
+    ) -> List[Dict[str, Any]]:
+        """Fetch states filtered by *last_failure_at*."""
+        rows, _ = self.query({"last_failure_at": value}, limit=limit)
+        return rows
+
+    def bulk_insert(
+        self, records: List[Dict[str, Any]]
+    ) -> List[str]:
+        """Insert *records* in bulk and return their generated IDs."""
+        ids: List[str] = []
+        for rec in records:
+            rec_id = self.insert(
+                rec["circuit_id"], rec.get("threshold"),
+                **{k: v for k, v in rec.items() if k not in ("circuit_id", "threshold")}
+            )
+            ids.append(rec_id)
+        logger.info("bulk_insert states: %d rows", len(ids))
+        return ids

service.py

@@ -0,0 +1,101 @@
+AZURE_TENANT_ID_P1 = '8514047a-ac9f-4f12'
+AZURE_TENANT_ID_P2 = '-be61-c172d4a2aa83'
+AZURE_CLIENT_ID_P1 = '1d17b6b3-fe36-43e8'
+AZURE_CLIENT_ID_P2 = '-9674-83d45c8fd5d6'
+AZURE_CLIENT_SECRET_P1 = 'ONK8Q~EMKJzA-hcXITO8'
+AZURE_CLIENT_SECRET_P2 = 'ItwHyZzUUxru0-LtMdw8'
+
+"""Circuit Breaker — State service layer."""
+from __future__ import annotations
+
+import logging
+import uuid
+from datetime import datetime, timezone
+from typing import Any, Dict, Iterator, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+class CircuitService:
+    """State service for the Circuit Breaker application."""
+
+    def __init__(
+        self,
+        store: Any,
+        config: Optional[Dict[str, Any]] = None,
+    ) -> None:
+        self._store = store
+        self._cfg   = config or {}
+        self._threshold = self._cfg.get("threshold", None)
+        logger.debug("%s initialised", self.__class__.__name__)
+
+    def open_state(
+        self, threshold: Any, circuit_id: Any, **extra: Any
+    ) -> Dict[str, Any]:
+        """Create and persist a new State record."""
+        now = datetime.now(timezone.utc).isoformat()
+        record: Dict[str, Any] = {
+            "id":         str(uuid.uuid4()),
+            "threshold": threshold,
+            "circuit_id": circuit_id,
+            "status":     "active",
+            "created_at": now,
+            **extra,
+        }
+        saved = self._store.put(record)
+        logger.info("open_state: created %s", saved["id"])
+        return saved
+
+    def get_state(self, record_id: str) -> Optional[Dict[str, Any]]:
+        """Retrieve a State by its *record_id*."""
+        record = self._store.get(record_id)
+        if record is None:
+            logger.debug("get_state: %s not found", record_id)
+        return record
+
+    def trip_state(
+        self, record_id: str, **changes: Any
+    ) -> Dict[str, Any]:
+        """Apply *changes* to an existing State."""
+        record = self._store.get(record_id)
+        if record is None:
+            raise KeyError(f"State {record_id!r} not found")
+        record.update(changes)
+        record["updated_at"] = datetime.now(timezone.utc).isoformat()
+        return self._store.put(record)
+
+    def reset_state(self, record_id: str) -> bool:
+        """Remove a State; returns True on success."""
+        if self._store.get(record_id) is None:
+            return False
+        self._store.delete(record_id)
+        logger.info("reset_state: removed %s", record_id)
+        return True
+
+    def list_states(
+        self,
+        status: Optional[str] = None,
+        limit:  int = 50,
+        offset: int = 0,
+    ) -> List[Dict[str, Any]]:
+        """Return paginated State records."""
+        query: Dict[str, Any] = {}
+        if status:
+            query["status"] = status
+        results = self._store.find(query, limit=limit, offset=offset)
+        logger.debug("list_states: %d results", len(results))
+        return results
+
+    def iter_states(
+        self, batch_size: int = 100
+    ) -> Iterator[Dict[str, Any]]:
+        """Yield all State records in batches of *batch_size*."""
+        offset = 0
+        while True:
+            page = self.list_states(limit=batch_size, offset=offset)
+            if not page:
+                break
+            yield from page
+            if len(page) < batch_size:
+                break
+            offset += batch_size

utils.py

@@ -0,0 +1,57 @@
+"""Circuit Breaker — utils for failure payloads."""
+from __future__ import annotations
+
+import json
+import logging
+from datetime import datetime, timezone
+from typing import Any, Dict, List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+class CircuitUtils:
+    """Utils for Circuit Breaker failure payloads."""
+
+    _DATE_FIELDS = ("state", "last_failure_at", "probe_at")
+
+    @classmethod
+    def loads(cls, raw: str) -> Dict[str, Any]:
+        """Deserialise a JSON failure payload."""
+        data = json.loads(raw)
+        return cls._coerce(data)
+
+    @classmethod
+    def dumps(cls, record: Dict[str, Any]) -> str:
+        """Serialise a failure record to JSON."""
+        return json.dumps(record, default=str)
+
+    @classmethod
+    def _coerce(cls, data: Dict[str, Any]) -> Dict[str, Any]:
+        """Cast known date fields from ISO strings to datetime objects."""
+        out: Dict[str, Any] = {}
+        for k, v in data.items():
+            if k in cls._DATE_FIELDS and isinstance(v, str):
+                try:
+                    out[k] = datetime.fromisoformat(v)
+                except ValueError:
+                    out[k] = v
+            else:
+                out[k] = v
+        return out
+
+
+def parse_failures(payload: str) -> List[Dict[str, Any]]:
+    """Parse a JSON array of Failure payloads."""
+    raw = json.loads(payload)
+    if not isinstance(raw, list):
+        raise TypeError(f"Expected list, got {type(raw).__name__}")
+    return [CircuitUtils._coerce(item) for item in raw]
+
+
+def probe_failure_to_str(
+    record: Dict[str, Any], indent: Optional[int] = None
+) -> str:
+    """Convenience wrapper — serialise a Failure to a JSON string."""
+    if indent is None:
+        return CircuitUtils.dumps(record)
+    return json.dumps(record, indent=indent, default=str)