mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-13 01:30:31 +08:00
Add minimal prometheus exporter support
Prometheus https://prometheus.io/ is one of the popular monitoring systems. It pulls service's endpoint to get metrics in a simple text format https://prometheus.io/docs/instrumenting/exposition_formats/
This commit is contained in:
parent
7ce7a6d035
commit
8f56e1ab8b
@ -22,6 +22,8 @@
|
|||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
|
from typing import Any
|
||||||
|
from typing import Dict
|
||||||
from typing import List
|
from typing import List
|
||||||
|
|
||||||
from aiohttp.web import Request
|
from aiohttp.web import Request
|
||||||
@ -34,6 +36,22 @@ from ..info import InfoManager
|
|||||||
|
|
||||||
from ..http import exposed_http
|
from ..http import exposed_http
|
||||||
from ..http import make_json_response
|
from ..http import make_json_response
|
||||||
|
from ..http import make_text_response
|
||||||
|
|
||||||
|
|
||||||
|
# ====
|
||||||
|
def _build_metrics(metrics: List[str], name: str, value: Any) -> None:
|
||||||
|
if isinstance(value, bool):
|
||||||
|
value = 1 if value else 0
|
||||||
|
if isinstance(value, (int, float)):
|
||||||
|
metrics.append(f"# TYPE {name} gauge")
|
||||||
|
metrics.append(f"{name} {value}")
|
||||||
|
elif isinstance(value, dict):
|
||||||
|
for key, val in value.items():
|
||||||
|
if key == "parsed_flags":
|
||||||
|
_build_metrics(metrics, name, val)
|
||||||
|
else:
|
||||||
|
_build_metrics(metrics, f"{name}_{key}", val)
|
||||||
|
|
||||||
|
|
||||||
# =====
|
# =====
|
||||||
@ -59,3 +77,17 @@ class InfoApi:
|
|||||||
subval=(lambda field: check_string_in_list(field, "info field", subs)),
|
subval=(lambda field: check_string_in_list(field, "info field", subs)),
|
||||||
name="info fields list",
|
name="info fields list",
|
||||||
))) or subs)
|
))) or subs)
|
||||||
|
|
||||||
|
@exposed_http("GET", "/export/prometheus/metrics", False)
|
||||||
|
async def __metrics_handler(self, _: Request) -> Response:
|
||||||
|
data = await asyncio.gather(self.__info_manager.get_submanager("hw").get_state())
|
||||||
|
if data is None:
|
||||||
|
return make_text_response("error", 500)
|
||||||
|
else:
|
||||||
|
data_exists: Dict[Any, Any] = data
|
||||||
|
health = data_exists[0]["health"]
|
||||||
|
|
||||||
|
metrics: List[str] = []
|
||||||
|
_build_metrics(metrics, "pikvm", health)
|
||||||
|
|
||||||
|
return make_text_response("\n".join(metrics))
|
||||||
|
|||||||
@ -147,6 +147,23 @@ def make_json_exception(err: Exception, status: Optional[int]=None) -> aiohttp.w
|
|||||||
}, status=status)
|
}, status=status)
|
||||||
|
|
||||||
|
|
||||||
|
def make_text_response(
|
||||||
|
result: str,
|
||||||
|
status: int=200,
|
||||||
|
set_cookies: Optional[Dict[str, str]]=None,
|
||||||
|
) -> aiohttp.web.Response:
|
||||||
|
|
||||||
|
response = aiohttp.web.Response(
|
||||||
|
text=result,
|
||||||
|
status=status,
|
||||||
|
content_type="text/plain",
|
||||||
|
)
|
||||||
|
if set_cookies:
|
||||||
|
for (key, value) in set_cookies.items():
|
||||||
|
response.set_cookie(key, value)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
# =====
|
# =====
|
||||||
async def get_multipart_field(reader: aiohttp.MultipartReader, name: str) -> aiohttp.BodyPartReader:
|
async def get_multipart_field(reader: aiohttp.MultipartReader, name: str) -> aiohttp.BodyPartReader:
|
||||||
field = await reader.next()
|
field = await reader.next()
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user