fastapi-alertengine 1.0.0__tar.gz → 1.0.2__tar.gz

{fastapi_alertengine-1.0.0 → fastapi_alertengine-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-alertengine
-Version: 1.0.0
+Version: 1.0.2
 Summary: FastAPI AlertEngine: fintech-grade request monitoring (p95 latency + error rate) without Prometheus.
 Author: Tandem Media
 Project-URL: Homepage, https://github.com/Tandem-Media/fastapi-alertengine

fastapi_alertengine-1.0.2/fastapi_alertengine/__init__.py

@@ -0,0 +1,7 @@
+# fastapi_alertengine/__init__.py
+
+from .engine import AlertEngine           # or from .alert_engine import AlertEngine
+from .middleware import RequestMetricsMiddleware
+from .client import get_alert_engine
+
+__all__ = ["AlertEngine", "RequestMetricsMiddleware", "get_alert_engine"]

fastapi_alertengine-1.0.2/fastapi_alertengine/app_main.py

@@ -0,0 +1,150 @@
+import time
+from typing import Callable
+
+import redis
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+from starlette.middleware.base import BaseHTTPMiddleware
+
+app = FastAPI()
+
+
+class RequestMetricsMiddleware(BaseHTTPMiddleware):
+    def __init__(self, app, redis_url: str = "redis://localhost:6379/0"):
+        super().__init__(app)
+        self.redis = redis.Redis.from_url(redis_url, decode_responses=True)
+
+    async def dispatch(self, request: Request, call_next: Callable):
+        start = time.time()
+        try:
+            response = await call_next(request)
+            status_code = response.status_code
+        except Exception:
+            status_code = 500
+            response = JSONResponse({"detail": "Internal server error"}, status_code=500)
+
+        duration_ms = (time.time() - start) * 1000
+        # Write a minimal metric to Redis Stream
+        self.redis.xadd(
+            "request_metrics",
+            {
+                "path": request.url.path,
+                "status": status_code,
+                "duration_ms": f"{duration_ms:.2f}",
+            },
+            maxlen=1000,
+        )
+        return response
+
+
+class AlertEngine:
+    """
+    AlertEngine Lite:
+    - Reads from Redis Stream `request_metrics`
+    - Tracks rolling error rate and P95 latency
+    - For now: prints alerts to console (no email/slack yet)
+    """
+
+    def __init__(
+        self,
+        redis_url: str = "redis://localhost:6379/0",
+        stream_key: str = "request_metrics",
+        group: str = "alert_engine",
+        consumer: str = "alert_engine_1",
+        error_rate_threshold: float = 0.05,
+        p95_latency_threshold_ms: float = 500.0,
+        window_size: int = 100,
+    ):
+        self.redis = redis.Redis.from_url(redis_url, decode_responses=True)
+        self.stream_key = stream_key
+        self.group = group
+        self.consumer = consumer
+        self.error_rate_threshold = error_rate_threshold
+        self.p95_latency_threshold_ms = p95_latency_threshold_ms
+        self.window_size = window_size
+
+        # Create consumer group if it doesn't exist
+        try:
+            self.redis.xgroup_create(self.stream_key, self.group, id="$", mkstream=True)
+        except redis.ResponseError as e:
+            # Group already exists
+            if "BUSYGROUP" not in str(e):
+                raise
+
+    def _calculate_alerts(self, events: list[dict]) -> None:
+        if not events:
+            return
+
+        durations = []
+        errors = 0
+
+        for e in events:
+            fields = e["fields"]
+            status = int(fields.get("status", 500))
+            duration_ms = float(fields.get("duration_ms", 0.0))
+            durations.append(duration_ms)
+            if status >= 500:
+                errors += 1
+
+        error_rate = errors / len(events)
+        p95_latency = self._p95(durations)
+
+        if error_rate >= self.error_rate_threshold:
+            print(
+                f"[ALERT] High error rate: {error_rate:.2%} "
+                f"(threshold {self.error_rate_threshold:.2%})"
+            )
+
+        if p95_latency >= self.p95_latency_threshold_ms:
+            print(
+                f"[ALERT] High P95 latency: {p95_latency:.1f} ms "
+                f"(threshold {self.p95_latency_threshold_ms:.1f} ms)"
+            )
+
+    @staticmethod
+    def _p95(values: list[float]) -> float:
+        if not values:
+            return 0.0
+        values_sorted = sorted(values)
+        idx = int(0.95 * (len(values_sorted) - 1))
+        return values_sorted[idx]
+
+    def poll_once(self, count: int = 200, block_ms: int = 1000) -> None:
+        """
+        Polls Redis Streams once and prints alerts if thresholds are crossed.
+        Can be called in a loop or a background task.
+        """
+        resp = self.redis.xreadgroup(
+            groupname=self.group,
+            consumername=self.consumer,
+            streams={self.stream_key: ">"},
+            count=count,
+            block=block_ms,
+        )
+
+        if not resp:
+            return
+
+        # resp is a list of (stream, [ (id, fields), ... ])
+        events = []
+        ids_to_ack = []
+        for stream_name, entries in resp:
+            for entry_id, fields in entries:
+                events.append({"id": entry_id, "fields": fields})
+                ids_to_ack.append(entry_id)
+
+        # Use only the latest window_size events for alert calculations
+        events = events[-self.window_size :]
+        self._calculate_alerts(events)
+
+        if ids_to_ack:
+            self.redis.xack(self.stream_key, self.group, *ids_to_ack)
+
+
+# Wire up middleware
+app.add_middleware(RequestMetricsMiddleware)
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}

fastapi_alertengine-1.0.2/fastapi_alertengine/client.py

@@ -0,0 +1,17 @@
+# fastapi_alertengine/client.py
+
+from functools import lru_cache
+
+from .engine import AlertEngine  # or from .alert_engine import AlertEngine if you didn't rename
+from .config import AlertConfig
+
+
+@lru_cache(maxsize=1)
+def get_alert_engine(config: AlertConfig | None = None) -> AlertEngine:
+    """
+    Return a singleton AlertEngine instance.
+
+    If a config is provided on first call, it is used to initialize the engine.
+    Subsequent calls ignore the config argument and return the same instance.
+    """
+    return AlertEngine(config=config)

{fastapi_alertengine-1.0.0 → fastapi_alertengine-1.0.2}/fastapi_alertengine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-alertengine
-Version: 1.0.0
+Version: 1.0.2
 Summary: FastAPI AlertEngine: fintech-grade request monitoring (p95 latency + error rate) without Prometheus.
 Author: Tandem Media
 Project-URL: Homepage, https://github.com/Tandem-Media/fastapi-alertengine

fastapi_alertengine-1.0.2/fastapi_alertengine.egg-info/SOURCES.txt

@@ -0,0 +1,22 @@
+README.md
+pyproject.toml
+setup.cfg
+./fastapi_alertengine/__init__.py
+./fastapi_alertengine/app_main.py
+./fastapi_alertengine/client.py
+./fastapi_alertengine/config.py
+./fastapi_alertengine/engine.py.py
+./fastapi_alertengine/middleware.py
+./fastapi_alertengine/storage.py.py
+fastapi_alertengine/__init__.py
+fastapi_alertengine/app_main.py
+fastapi_alertengine/client.py
+fastapi_alertengine/config.py
+fastapi_alertengine/engine.py.py
+fastapi_alertengine/middleware.py
+fastapi_alertengine/storage.py.py
+fastapi_alertengine.egg-info/PKG-INFO
+fastapi_alertengine.egg-info/SOURCES.txt
+fastapi_alertengine.egg-info/dependency_links.txt
+fastapi_alertengine.egg-info/requires.txt
+fastapi_alertengine.egg-info/top_level.txt

fastapi_alertengine-1.0.2/fastapi_alertengine.egg-info/top_level.txt

@@ -0,0 +1 @@
+fastapi_alertengine

{fastapi_alertengine-1.0.0 → fastapi_alertengine-1.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fastapi-alertengine"
-version = "1.0.0"
+version = "1.0.2"
 description = "FastAPI AlertEngine: fintech-grade request monitoring (p95 latency + error rate) without Prometheus."
 readme = "README.md"
 requires-python = ">=3.10"

fastapi_alertengine-1.0.2/setup.cfg

@@ -0,0 +1,16 @@
+[metadata]
+name = fastapi-alertengine
+version = "1.0.2"
+
+[options]
+packages = find:
+package_dir = 
+	= .
+
+[options.packages.find]
+where = .
+
+[egg_info]
+tag_build = 
+tag_date = 0
+

fastapi_alertengine-1.0.0/fastapi_alertengine.egg-info/SOURCES.txt

@@ -1,7 +0,0 @@
-README.md
-pyproject.toml
-fastapi_alertengine.egg-info/PKG-INFO
-fastapi_alertengine.egg-info/SOURCES.txt
-fastapi_alertengine.egg-info/dependency_links.txt
-fastapi_alertengine.egg-info/requires.txt
-fastapi_alertengine.egg-info/top_level.txt

fastapi_alertengine-1.0.0/fastapi_alertengine.egg-info/top_level.txt

@@ -1 +0,0 @@
-

fastapi_alertengine-1.0.0/setup.cfg

@@ -1,4 +0,0 @@
-[egg_info]
-tag_build = 
-tag_date = 0
-