SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-22 12:11:26 +00:00
Code Issues Projects Releases Wiki Activity

feat: add _send_request

Browse source
This commit is contained in:
def 2024-06-10 05:27:37 +03:00
parent cbdcb4a63f
commit 2e92557e84
1 changed files with 18 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
selfprivacy_api/graphql/queries/monitoring.py
View file

@ -25,13 +25,15 @@ class MonitoringInfo:
class Monitoring:
"""GraphQL queries to get prometheus monitoring information."""
@strawberry.field
def status(self) -> MonitoringInfo:
"""Get status"""
endpoint = "/api/v1/status/runtimeinfo"
def _send_request(self, endpoint="/api/v1/query", query=None) -> MonitoringInfo:
try:
if query:
response = requests.get(
f"{PROMETHEUS_URL}{endpoint}", params={"query": query}
)
else:
response = requests.get(f"{PROMETHEUS_URL}{endpoint}")
return MonitoringInfo(
http_response=response.status_code,
output=(
@ -44,47 +46,25 @@ class Monitoring:
error=str(error),
)
@strawberry.field
def status(self) -> MonitoringInfo:
"""Get status"""
endpoint = "/api/v1/status/runtimeinfo"
return _send_request(endpoint=endpoint)
@strawberry.field
def build(self) -> MonitoringInfo:
"""Get build information"""
endpoint = "/api/v1/status/buildinfo"
return _send_request(endpoint=endpoint)
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):
@strawberry.field
def cpu_usage(self) -> MonitoringInfo:
"""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),
)
return _send_request(query=query)
def disks_usage(): ...