Skip to content

Commit 45bdaf5

Browse files
committed
refactor: remove scheduler service and related code
1 parent 9b04eee commit 45bdaf5

File tree

4 files changed

+12
-16
lines changed

4 files changed

+12
-16
lines changed

apps/common/management/commands/services/command.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ class Services(TextChoices):
1111
gunicorn = 'gunicorn', 'gunicorn'
1212
celery_default = 'celery_default', 'celery_default'
1313
local_model = 'local_model', 'local_model'
14-
scheduler = 'scheduler', 'scheduler'
1514
web = 'web', 'web'
1615
celery = 'celery', 'celery'
1716
celery_model = 'celery_model', 'celery_model'
@@ -25,7 +24,6 @@ def get_service_object_class(cls, name):
2524
cls.gunicorn.value: services.GunicornService,
2625
cls.celery_default: services.CeleryDefaultService,
2726
cls.local_model: services.GunicornLocalModelService,
28-
cls.scheduler: services.SchedulerService,
2927
}
3028
return services_map.get(name)
3129

@@ -41,13 +39,10 @@ def celery_services(cls):
4139
def task_services(cls):
4240
return cls.celery_services()
4341

44-
@classmethod
45-
def scheduler_services(cls):
46-
return [cls.scheduler]
4742

4843
@classmethod
4944
def all_services(cls):
50-
return cls.web_services() + cls.task_services() + cls.scheduler_services()
45+
return cls.web_services() + cls.task_services()
5146

5247
@classmethod
5348
def export_services_values(cls):

apps/maxkb/wsgi/web.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -44,15 +44,5 @@ def post_handler():
4444
DatabaseModelManage.init()
4545

4646

47-
def post_scheduler_handler():
48-
from common import job
49-
50-
job.run()
51-
52-
5347
# 启动后处理函数
5448
post_handler()
55-
56-
# 仅在scheduler中启动定时任务,dev local_model celery 不需要
57-
if os.environ.get('ENABLE_SCHEDULER') == '1':
58-
post_scheduler_handler()

apps/ops/celery/signal_handler.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,20 @@
1717
logger = logging.getLogger(__file__)
1818
safe_str = lambda x: x
1919

20+
def init_scheduler():
21+
from common import job
22+
23+
job.run()
24+
2025

2126
@worker_ready.connect
2227
def on_app_ready(sender=None, headers=None, **kwargs):
2328
if cache.get("CELERY_APP_READY", 0) == 1:
2429
return
2530
cache.set("CELERY_APP_READY", 1, 10)
31+
# 初始化定时任务
32+
init_scheduler()
33+
2634
tasks = get_after_app_ready_tasks()
2735
logger.debug("Work ready signal recv")
2836
logger.debug("Start need start task: [{}]".format(", ".join(tasks)))

apps/users/apps.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
class UsersConfig(AppConfig):
55
default_auto_field = 'django.db.models.BigAutoField'
66
name = 'users'
7+
8+
def ready(self):
9+
from ops.celery import signal_handler

0 commit comments

Comments
 (0)