From 1f4420a3d8063f57a7491963c777645dadcd5482 Mon Sep 17 00:00:00 2001
From: RhodeCode Admin <admin@rhodecode.com>
Date: Thu, 21 Dec 2023 10:15:16 +0100
Subject: [PATCH] fix(celery): setup statsd on celery

---
 rhodecode/lib/celerylib/loader.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/rhodecode/lib/celerylib/loader.py b/rhodecode/lib/celerylib/loader.py
index a00772a1..e8435593 100644
--- a/rhodecode/lib/celerylib/loader.py
+++ b/rhodecode/lib/celerylib/loader.py
@@ -162,7 +162,11 @@ def _init_celery(app_type=''):
 
     log.debug('Got Pyramid ENV: %s', env)
 
-    celery_settings = get_celery_config(env['registry'].settings)
+    settings = env['registry'].settings
+    celery_settings = get_celery_config(settings)
+
+    # init and bootstrap StatsdClient
+    StatsdClient.setup(settings)
 
     setup_celery_app(
         app=env['app'], root=env['root'], request=env['request'],
@@ -188,10 +192,11 @@ def on_beat_init(sender=None, conf=None, **kwargs):
 def task_prerun_signal(task_id, task, args, **kwargs):
     ping_db()
     statsd = StatsdClient.statsd
+
     if statsd:
         task_repr = getattr(task, 'name', task)
         statsd.incr('rhodecode_celery_task_total', tags=[
-            'task:{}'.format(task_repr),
+            f'task:{task_repr}',
             'mode:async'
         ])