feat: add class Monitoring

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(): ...