From 84a098ce64380ea884101702c966d1bf71d9ef40 Mon Sep 17 00:00:00 2001
From: Julian Bez <julian@posthog.com>
Date: Mon, 8 Apr 2024 13:22:06 +0100
Subject: [PATCH] feat(celery): Add prefetch multiplier to flags (#21399)

Since the env var doesn't make it to the subprocess
---
 bin/docker-worker-celery | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bin/docker-worker-celery b/bin/docker-worker-celery
index bbd9949d883..494a9276beb 100755
--- a/bin/docker-worker-celery
+++ b/bin/docker-worker-celery
@@ -76,6 +76,8 @@ FLAGS+=("-n node@%h")
 # Restart worker process after it exceeds this much memory usage (to mitigate memory leaks)
 [[ -n "${CELERY_MAX_MEMORY_PER_CHILD}" ]]    && FLAGS+=" --max-memory-per-child $CELERY_MAX_MEMORY_PER_CHILD"
 
+[[ -n "${CELERY_WORKER_PREFETCH_MULTIPLIER}" ]] && FLAGS+=" --prefetch-multiplier $CELERY_WORKER_PREFETCH_MULTIPLIER"
+
 if [[ -z "${CELERY_WORKER_QUEUES}" ]]; then
   source ./bin/celery-queues.env
 fi