mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-09-20 15:37:50 +00:00
dettlaff
4cd90d0c93
Co-authored-by: nhnn <nhnn@disroot.org> Co-authored-by: Inex Code <inex.code@selfprivacy.org> Reviewed-on: https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api/pulls/120 Co-authored-by: dettlaff <dettlaff@riseup.net> Co-committed-by: dettlaff <dettlaff@riseup.net>
121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
import strawberry
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
from selfprivacy_api.models.services import ServiceStatus
|
|
from selfprivacy_api.services.prometheus import Prometheus
|
|
from selfprivacy_api.utils.monitoring import (
|
|
MonitoringQueries,
|
|
MonitoringQueryError,
|
|
MonitoringValuesResult,
|
|
MonitoringMetricsResult,
|
|
)
|
|
|
|
|
|
@strawberry.type
|
|
class CpuMonitoring:
|
|
start: Optional[datetime]
|
|
end: Optional[datetime]
|
|
step: int
|
|
|
|
@strawberry.field
|
|
def overall_usage(self) -> MonitoringValuesResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.cpu_usage_overall(self.start, self.end, self.step)
|
|
|
|
|
|
@strawberry.type
|
|
class MemoryMonitoring:
|
|
start: Optional[datetime]
|
|
end: Optional[datetime]
|
|
step: int
|
|
|
|
@strawberry.field
|
|
def overall_usage(self) -> MonitoringValuesResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.memory_usage_overall(self.start, self.end, self.step)
|
|
|
|
@strawberry.field
|
|
def average_usage_by_service(self) -> MonitoringMetricsResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.memory_usage_average_by_slice(self.start, self.end)
|
|
|
|
@strawberry.field
|
|
def max_usage_by_service(self) -> MonitoringMetricsResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.memory_usage_max_by_slice(self.start, self.end)
|
|
|
|
|
|
@strawberry.type
|
|
class DiskMonitoring:
|
|
start: Optional[datetime]
|
|
end: Optional[datetime]
|
|
step: int
|
|
|
|
@strawberry.field
|
|
def overall_usage(self) -> MonitoringMetricsResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.disk_usage_overall(self.start, self.end, self.step)
|
|
|
|
|
|
@strawberry.type
|
|
class NetworkMonitoring:
|
|
start: Optional[datetime]
|
|
end: Optional[datetime]
|
|
step: int
|
|
|
|
@strawberry.field
|
|
def overall_usage(self) -> MonitoringMetricsResult:
|
|
if Prometheus().get_status() != ServiceStatus.ACTIVE:
|
|
return MonitoringQueryError(error="Prometheus is not running")
|
|
|
|
return MonitoringQueries.network_usage_overall(self.start, self.end, self.step)
|
|
|
|
|
|
@strawberry.type
|
|
class Monitoring:
|
|
@strawberry.field
|
|
def cpu_usage(
|
|
self,
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60,
|
|
) -> CpuMonitoring:
|
|
return CpuMonitoring(start=start, end=end, step=step)
|
|
|
|
@strawberry.field
|
|
def memory_usage(
|
|
self,
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60,
|
|
) -> MemoryMonitoring:
|
|
return MemoryMonitoring(start=start, end=end, step=step)
|
|
|
|
@strawberry.field
|
|
def disk_usage(
|
|
self,
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60,
|
|
) -> DiskMonitoring:
|
|
return DiskMonitoring(start=start, end=end, step=step)
|
|
|
|
@strawberry.field
|
|
def network_usage(
|
|
self,
|
|
start: Optional[datetime] = None,
|
|
end: Optional[datetime] = None,
|
|
step: int = 60,
|
|
) -> NetworkMonitoring:
|
|
return NetworkMonitoring(start=start, end=end, step=step)
|