feat: initial commit - Phase 1 & 2 core features

This commit is contained in:
hiderfong
2026-04-22 17:07:33 +08:00
commit 1773bda06b
25005 changed files with 6252106 additions and 0 deletions
@@ -0,0 +1,5 @@
from .exposition import make_aiohttp_handler
__all__ = [
"make_aiohttp_handler",
]
@@ -0,0 +1,39 @@
from __future__ import annotations
from aiohttp import hdrs, web
from aiohttp.typedefs import Handler
from ..exposition import _bake_output
from ..registry import Collector, REGISTRY
def make_aiohttp_handler(
registry: Collector = REGISTRY,
disable_compression: bool = False,
) -> Handler:
"""Create a aiohttp handler which serves the metrics from a registry."""
async def prometheus_handler(request: web.Request) -> web.Response:
# Prepare parameters
params = {key: request.query.getall(key) for key in request.query.keys()}
accept_header = ",".join(request.headers.getall(hdrs.ACCEPT, []))
accept_encoding_header = ""
# Bake output
status, headers, output = _bake_output(
registry,
accept_header,
accept_encoding_header,
params,
# use AIOHTTP's compression
disable_compression=True,
)
response = web.Response(
status=int(status.split(" ")[0]),
headers=headers,
body=output,
)
if not disable_compression:
response.enable_compression()
return response
return prometheus_handler