SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-25 21:41:27 +00:00
Code Issues Projects Releases Wiki Activity

fix: upd disks query

Browse source
This commit is contained in:
dettlaff 2024-07-16 06:41:06 +04:00
parent 092b8deb9c
commit f357607f68
1 changed files with 24 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
selfprivacy_api/utils/prometheus.py
View file

@ -22,51 +22,48 @@ class PrometheusQueryResult:
class PrometheusQueries: class PrometheusQueries:
@staticmethod @staticmethod
def _send_query(query: str, start: datetime, end: datetime, step: int): def _send_query(query: str, start: int, end: int, step: int):
try: try:
response = requests.get( response = requests.get(
f"{PROMETHEUS_URL}/api/v1/query_range", f"{PROMETHEUS_URL}/api/v1/query",
params={ params={
"query": query, "query": query,
"start": int(start.timestamp()), "start": start,
"end": int(start.timestamp()), "end": end,
"step": step, "step": step,
}, },
) )
if response.status_code != 200: if response.status_code != 200:
raise Exception("Prometheus returned unexpected HTTP status code") raise Exception("Prometheus returned unexpected HTTP status code")
json = response.json() json = response.json()
if json["status"] != "success":
raise Exception("Prometheus returned unexpected status")
result = json["data"]
return PrometheusQueryResult( return PrometheusQueryResult(
result_type=result["resultType"], result=result["result"] result_type=json["result_type"], result=json["result"]
) )
except Exception as error: except Exception as error:
raise Exception("Prometheus request failed! " + str(error)) raise Exception("Prometheus request failed! " + str(error))
@staticmethod @staticmethod
def cpu_usage( def cpu_usage(
start: Optional[datetime] = None, start: Optional[int] = None,
end: Optional[datetime] = None, end: Optional[int] = None,
step: int = 60, # seconds step: int = 60, # seconds
) -> PrometheusQueryResult: ) -> PrometheusQueryResult:
""" """
Get CPU information. Get CPU information.
Args: Args:
start (datetime, optional): timestamp indicating the start time of metrics to fetch start (int, optional): Unix timestamp indicating the start time.
Defaults to 20 minutes ago if not provided. Defaults to 20 minutes ago if not provided.
end (datetime, optional): timestamp indicating the end time of metrics to fetch end (int, optional): Unix timestamp indicating the end time.
Defaults to current time if not provided. Defaults to current time if not provided.
step (int): Interval in seconds for querying disk usage data. step (int): Interval in seconds for querying disk usage data.
""" """
if not start: if not start:
start = datetime.now() - timedelta(minutes=20) start = int((datetime.now() - timedelta(minutes=20)).timestamp())
if not end: if not end:
end = datetime.now() end = int(datetime.now().timestamp())
query = '100 - (avg by (instance) (rate(node_cpu_seconds_total{mode="idle"}[5m])) * 100)' query = '100 - (avg by (instance) (rate(node_cpu_seconds_total{mode="idle"}[5m])) * 100)'
@ -74,26 +71,26 @@ class PrometheusQueries:
@staticmethod @staticmethod
def memory_usage( def memory_usage(
start: Optional[datetime] = None, start: Optional[int] = None,
end: Optional[datetime] = None, end: Optional[int] = None,
step: int = 60, # seconds step: int = 60, # seconds
) -> PrometheusQueryResult: ) -> PrometheusQueryResult:
""" """
Get memory usage. Get memory usage.
Args: Args:
start (datetime, optional): timestamp indicating the start time of metrics to fetch start (int, optional): Unix timestamp indicating the start time.
Defaults to 20 minutes ago if not provided. Defaults to 20 minutes ago if not provided.
end (datetime, optional): timestamp indicating the end time of metrics to fetch end (int, optional): Unix timestamp indicating the end time.
Defaults to current time if not provided. Defaults to current time if not provided.
step (int): Interval in seconds for querying memory usage data. step (int): Interval in seconds for querying memory usage data.
""" """
if not start: if not start:
start = datetime.now() - timedelta(minutes=20) start = int((datetime.now() - timedelta(minutes=20)).timestamp())
if not end: if not end:
end = datetime.now() end = int(datetime.now().timestamp())
query = "100 - (100 * (node_memory_MemAvailable_bytes / node_memory_MemTotal_bytes))" query = "100 - (100 * (node_memory_MemAvailable_bytes / node_memory_MemTotal_bytes))"
@ -101,27 +98,27 @@ class PrometheusQueries:
@staticmethod @staticmethod
def disk_usage( def disk_usage(
start: Optional[datetime] = None, start: Optional[int] = None,
end: Optional[datetime] = None, end: Optional[int] = None,
step: int = 60, # seconds step: int = 60, # seconds
) -> PrometheusQueryResult: ) -> PrometheusQueryResult:
""" """
Get disk usage information. Get disk usage information.
Args: Args:
start (datetime, optional): timestamp indicating the start time of metrics to fetch start (int, optional): Unix timestamp indicating the start time.
Defaults to 20 minutes ago if not provided. Defaults to 20 minutes ago if not provided.
end (datetime, optional): timestamp indicating the end time of metrics to fetch end (int, optional): Unix timestamp indicating the end time.
Defaults to current time if not provided. Defaults to current time if not provided.
step (int): Interval in seconds for querying disk usage data. step (int): Interval in seconds for querying disk usage data.
""" """
if not start: if not start:
start = datetime.now() - timedelta(minutes=20) start = int((datetime.now() - timedelta(minutes=20)).timestamp())
if not end: if not end:
end = datetime.now() end = int(datetime.now().timestamp())
query = '100 - (100 * ((node_filesystem_avail_bytes{mountpoint="/",fstype!="rootfs"} ) / (node_filesystem_size_bytes{mountpoint="/",fstype!="rootfs"}) ))' 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, end, step) return PrometheusQueries._send_query(query, start, end, step)