mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-23 04:21:29 +00:00
149 lines
4.3 KiB
Python
149 lines
4.3 KiB
Python
"""Prometheus monitoring queries."""
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
import requests
|
|
|
|
import strawberry
|
|
from strawberry.scalars import JSON
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
from datetime import datetime, timedelta
|
|
|
|
PROMETHEUS_URL = "http://localhost:9001"
|
|
|
|
|
|
@strawberry.type
|
|
@dataclass
|
|
class PrometheusQueryResult:
|
|
result_type: str
|
|
result: JSON
|
|
|
|
|
|
class PrometheusQueries:
|
|
@staticmethod
|
|
def _send_query(query: str, start: int, end: int, step: int):
|
|
try:
|
|
response = requests.get(
|
|
f"{PROMETHEUS_URL}/api/v1/query",
|
|
params={
|
|
"query": query,
|
|
"start": start,
|
|
"end": end,
|
|
"step": step,
|
|
},
|
|
)
|
|
if response.status_code != 200:
|
|
raise Exception("Prometheus returned unexpected HTTP status code")
|
|
json = response.json()
|
|
return PrometheusQueryResult(
|
|
result_type=json["data"]["resultType"], result=json["data"]["result"]
|
|
)
|
|
except Exception as error:
|
|
raise Exception("Prometheus request failed! " + str(error))
|
|
|
|
@staticmethod
|
|
def cpu_usage(
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60, # seconds
|
|
) -> PrometheusQueryResult:
|
|
"""
|
|
Get CPU information.
|
|
|
|
Args:
|
|
start (datetime, optional): The start time.
|
|
Defaults to 20 minutes ago if not provided.
|
|
end (datetime, optional): The end time.
|
|
Defaults to current time if not provided.
|
|
step (int): Interval in seconds for querying disk usage data.
|
|
"""
|
|
|
|
if start is None:
|
|
start = datetime.now() - timedelta(minutes=20)
|
|
|
|
if end is None:
|
|
end = datetime.now()
|
|
|
|
start_timestamp = int(start.timestamp())
|
|
end_timestamp = int(end.timestamp())
|
|
|
|
query = '100 - (avg by (instance) (rate(node_cpu_seconds_total{mode="idle"}[5m])) * 100)'
|
|
|
|
return PrometheusQueries._send_query(
|
|
query,
|
|
start_timestamp,
|
|
end_timestamp,
|
|
step,
|
|
)
|
|
|
|
@staticmethod
|
|
def memory_usage(
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60, # seconds
|
|
) -> PrometheusQueryResult:
|
|
"""
|
|
Get memory usage.
|
|
|
|
Args:
|
|
start (datetime, optional): The start time.
|
|
Defaults to 20 minutes ago if not provided.
|
|
end (datetime, optional): The end time.
|
|
Defaults to current time if not provided.
|
|
step (int): Interval in seconds for querying memory usage data.
|
|
"""
|
|
|
|
if start is None:
|
|
start = datetime.now() - timedelta(minutes=20)
|
|
|
|
if end is None:
|
|
end = datetime.now()
|
|
|
|
start_timestamp = int(start.timestamp())
|
|
end_timestamp = int(end.timestamp())
|
|
|
|
query = "100 - (100 * (node_memory_MemAvailable_bytes / node_memory_MemTotal_bytes))"
|
|
|
|
return PrometheusQueries._send_query(
|
|
query,
|
|
start_timestamp,
|
|
end_timestamp,
|
|
step,
|
|
)
|
|
|
|
@staticmethod
|
|
def disk_usage(
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60, # seconds
|
|
) -> PrometheusQueryResult:
|
|
"""
|
|
Get disk usage information.
|
|
|
|
Args:
|
|
start (datetime, optional): The start time.
|
|
Defaults to 20 minutes ago if not provided.
|
|
end (datetime, optional): The end time.
|
|
Defaults to current time if not provided.
|
|
step (int): Interval in seconds for querying disk usage data.
|
|
"""
|
|
|
|
if start is None:
|
|
start = datetime.now() - timedelta(minutes=20)
|
|
|
|
if end is None:
|
|
end = datetime.now()
|
|
|
|
start_timestamp = int(start.timestamp())
|
|
end_timestamp = int(end.timestamp())
|
|
|
|
query = """100 - (100 * sum by (device) (node_filesystem_avail_bytes{fstype!="rootfs"}) / sum by (device) (node_filesystem_size_bytes{fstype!="rootfs"}))"""
|
|
|
|
return PrometheusQueries._send_query(
|
|
query,
|
|
start_timestamp,
|
|
end_timestamp,
|
|
step,
|
|
)
|