feat(celery): 添加celery的health check接口

This commit is contained in:
ibuler
2020-12-09 16:27:04 +08:00
committed by Jiangjie.Bai
parent 32dbab2e34
commit 80b03e73f6
2 changed files with 46 additions and 0 deletions

View File

@@ -0,0 +1,20 @@
from django.core.management.base import BaseCommand, CommandError
class Command(BaseCommand):
help = 'Ops manage commands'
def add_arguments(self, parser):
parser.add_argument('check_celery', nargs='?', help='Check celery health')
def handle(self, *args, **options):
from ops.celery.utils import get_celery_status, get_beat_status
ok = get_celery_status()
if not ok:
raise CommandError('Celery worker unhealthy')
ok = get_beat_status()
if not ok:
raise CommandError('Beat unhealthy')