celery worker log format following LOG_FORMAT env#9404 (#10016)

Signed-off-by: root <root@localhost.localdomain> Co-authored-by: root <root@localhost.localdomain>
langgenius · Nov 8, 2024 · 7c2a9b0 · 7c2a9b0
1 parent 888d7e6
commit 7c2a9b0
Show file tree

Hide file tree

Showing 3 changed files with 290 additions and 274 deletions.
diff --git a/api/extensions/ext_celery.py b/api/extensions/ext_celery.py
@@ -1,5 +1,6 @@
 from datetime import timedelta
 
+import pytz
 from celery import Celery, Task
 from celery.schedules import crontab
 from flask import Flask
@@ -43,6 +44,11 @@ def __call__(self, *args: object, **kwargs: object) -> object:
         result_backend=dify_config.CELERY_RESULT_BACKEND,
         broker_transport_options=broker_transport_options,
         broker_connection_retry_on_startup=True,
+        worker_log_format=dify_config.LOG_FORMAT,
+        worker_task_log_format=dify_config.LOG_FORMAT,
+        worker_logfile=dify_config.LOG_FILE,
+        worker_hijack_root_logger=False,
+        timezone=pytz.timezone(dify_config.LOG_TZ),
     )
 
     if dify_config.BROKER_USE_SSL: