feat(celery): 保证同时只有一个beat在运行 (#4723)

* feat(celery): 保证同时只有一个beat在运行

* fix: 修复代码拼写错误

* fix: 修复拼写

* fix: remove import

Co-authored-by: ibuler <ibuler@qq.com>

utils/start_celery_beat.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+#
+import os
+import sys
+import subprocess
+
+import redis_lock
+from redis import Redis
+
+BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+APPS_DIR = os.path.join(BASE_DIR, 'apps')
+
+sys.path.insert(0, BASE_DIR)
+from apps.jumpserver.const import CONFIG
+
+os.environ.setdefault('PYTHONOPTIMIZE', '1')
+if os.getuid() == 0:
+    os.environ.setdefault('C_FORCE_ROOT', '1')
+
+redis = Redis(host=CONFIG.REDIS_HOST, port=CONFIG.REDIS_PORT, password=CONFIG.REDIS_PASSWORD)
+scheduler = "django_celery_beat.schedulers:DatabaseScheduler"
+
+cmd = [
+    'celery', 'beat',
+    '-A', 'ops',
+    '-l', 'INFO',
+    '--scheduler', scheduler,
+    '--max-interval', '60'
+]
+
+with redis_lock.Lock(redis, name="beat-distribute-start-lock", expire=60, auto_renewal=True):
+    print("Get beat lock start to run it")
+    code = subprocess.call(cmd, cwd=APPS_DIR)
+    sys.exit(code)