diff --git a/selfprivacy_api/graphql/queries/monitoring.py b/selfprivacy_api/graphql/queries/monitoring.py new file mode 100644 index 0000000..f116e02 --- /dev/null +++ b/selfprivacy_api/graphql/queries/monitoring.py @@ -0,0 +1,90 @@ +"""Prometheus monitoring queries.""" + +# pylint: disable=too-few-public-methods +import typing +from typing import Optional + +import strawberry +import requests + +# TODO: добавить файл в schema.py + +PROMETHEUS_URL = "http://localhost:9090" + + +@strawberry.type +class MonitoringInfo: + """ """ + + http_response: int + output: Optional[str] = None + error: Optional[str] = None + + +@strawberry.type +class Monitoring: + """GraphQL queries to get prometheus monitoring information.""" + + @strawberry.field + def status(self) -> MonitoringInfo: + """Get status""" + endpoint = "/api/v1/status/runtimeinfo" + + try: + response = requests.get(f"{PROMETHEUS_URL}{endpoint}") + return MonitoringInfo( + http_response=response.status_code, + output=( + json.dumps(response.json()) if response.status_code == 200 else None + ), + ) + except requests.RequestException as error: + return MonitoringInfo( + http_response=500, # копилот предложил использовать тут 500 + error=str(error), + ) + + @strawberry.field + def build(self) -> MonitoringInfo: + """Get build information""" + endpoint = "/api/v1/status/buildinfo" + + try: + response = requests.get(f"{PROMETHEUS_URL}{endpoint}") + return MonitoringInfo( + http_response=response.status_code, + output=( + json.dumps(response.json()) if response.status_code == 200 else None + ), + ) + except requests.RequestException as error: + return MonitoringInfo( + http_response=500, + error=str(error), + ) + + def cpu_usage(self): + """Get CPU information""" + endpoint = "/api/v1/query" + query = """100 - (avg by (instance) (irate(node_cpu_seconds_total{job="node",mode="idle"}[5m])) * 100)""" + # TODO: не представляю рабочий ли этот запрос вообще + # https://overflow.hostux.net/questions/34923788/prometheus-convert-cpu-user-seconds-to-cpu-usage + # ему бы еще сделать кастомные минуты, чтоб разное время указывать, но так чтоб нельзя было инжектить + + try: + response = requests.get( + f"{PROMETHEUS_URL}{endpoint}", params={"query": query} + ) + return MonitoringInfo( + http_response=response.status_code, + output=( + json.dumps(response.json()) if response.status_code == 200 else None + ), + ) + except requests.RequestException as error: + return MonitoringInfo( + http_response=500, + error=str(error), + ) + + def disks_usage(): ...