zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace logging calls with maxkb_logger for consistency

Browse Source
This commit is contained in:
CaptainB 2025-06-25 17:36:07 +08:00
parent 3a2294e53b
commit 40fcc5bcb9
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/ops/celery/signal_handler.py
View File

@ -10,6 +10,7 @@ from celery.signals import (
from django.core.cache import cache from django.core.cache import cache
from django_celery_beat.models import PeriodicTask from django_celery_beat.models import PeriodicTask
from common.utils.logger import maxkb_logger
from .decorator import get_after_app_ready_tasks, get_after_app_shutdown_clean_tasks from .decorator import get_after_app_ready_tasks, get_after_app_shutdown_clean_tasks
from .logger import CeleryThreadTaskFileHandler from .logger import CeleryThreadTaskFileHandler
@ -65,7 +66,7 @@ def add_celery_logger_handler(sender=None, logger=None, loglevel=None, format=No
@task_revoked.connect @task_revoked.connect
def on_task_revoked(request, terminated, signum, expired, **kwargs): def on_task_revoked(request, terminated, signum, expired, **kwargs):
logging.getLogger('max_kb').info('task_revoked', terminated) maxkb_logger.info('task_revoked', terminated)
@task_prerun.connect @task_prerun.connect

3
apps/ops/celery/utils.py
View File

@ -9,6 +9,7 @@ from django_celery_beat.models import (
PeriodicTasks PeriodicTasks
) )
from common.utils.logger import maxkb_logger
from maxkb.const import PROJECT_DIR from maxkb.const import PROJECT_DIR
logger = logging.getLogger(__file__) logger = logging.getLogger(__file__)
@ -62,7 +63,7 @@ def get_celery_status():
active_queue_worker = set([n.split('@')[0] for n in active_nodes if n]) active_queue_worker = set([n.split('@')[0] for n in active_nodes if n])
# Celery Worker 数量: 2 # Celery Worker 数量: 2
if len(active_queue_worker) < 2: if len(active_queue_worker) < 2:
logging.getLogger('max_kb').info("Not all celery worker worked") maxkb_logger.info("Not all celery worker worked")
return False return False
else: else:
return True return True