refactor: replace logging calls with maxkb_logger for consistency

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

View File

@ -10,6 +10,7 @@ from celery.signals import (
from django.core.cache import cache
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 .logger import CeleryThreadTaskFileHandler
@ -65,7 +66,7 @@ def add_celery_logger_handler(sender=None, logger=None, loglevel=None, format=No
@task_revoked.connect
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

View File

@ -9,6 +9,7 @@ from django_celery_beat.models import (
PeriodicTasks
)
from common.utils.logger import maxkb_logger
from maxkb.const import PROJECT_DIR
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])
# Celery 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
else:
return True