9568f4dbd6
* cleanup pylint comments Signed-off-by: Jens Langhammer <jens@goauthentik.io> * remove more Signed-off-by: Jens Langhammer <jens@goauthentik.io> * fix url name Signed-off-by: Jens Langhammer <jens@goauthentik.io> * *: use ExtractHour instead of ExtractDay Signed-off-by: Jens Langhammer <jens@goauthentik.io> Signed-off-by: Jens Langhammer <jens@goauthentik.io>
22 lines
605 B
Python
22 lines
605 B
Python
"""admin signals"""
|
|
from django.dispatch import receiver
|
|
|
|
from authentik.admin.api.tasks import TaskInfo
|
|
from authentik.admin.apps import GAUGE_WORKERS
|
|
from authentik.root.celery import CELERY_APP
|
|
from authentik.root.monitoring import monitoring_set
|
|
|
|
|
|
@receiver(monitoring_set)
|
|
def monitoring_set_workers(sender, **kwargs):
|
|
"""Set worker gauge"""
|
|
count = len(CELERY_APP.control.ping(timeout=0.5))
|
|
GAUGE_WORKERS.set(count)
|
|
|
|
|
|
@receiver(monitoring_set)
|
|
def monitoring_set_tasks(sender, **kwargs):
|
|
"""Set task gauges"""
|
|
for task in TaskInfo.all().values():
|
|
task.set_prom_metrics()
|