Squash 2041 - Try sending a variable for CELERY_CONCURRENCY for healt…

…h check
department-of-veterans-affairs · Nov 21, 2024 · 52de86e · 52de86e
1 parent 025dcf1
commit 52de86e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/cd/application-deployment/dev/vaec-celery-task-definition.json b/cd/application-deployment/dev/vaec-celery-task-definition.json
@@ -23,6 +23,9 @@
         }
       ],
       "environment": [
+        { "name": "CELERY_CONCURRENCY", 
+          "value": 16
+        },
         {
           "name": "DD_SERVICE",
           "value": "celery"

diff --git a/scripts/check_celery.sh b/scripts/check_celery.sh
@@ -5,8 +5,9 @@ set -e
 grep_string="run_celery.notify_celery\ worker"
 celery_pid_count=$(ps aux | grep -E run_celery.notify_celery\ worker | grep -v grep | wc -l)
 
+TOTAL_CELERY_PIDS=$((CELERY_CONCURRENCY + 1))
 
-if [ $celery_pid_count -ne 17 ]; then
+if [ $celery_pid_count -ne TOTAL_CELERY_PIDS ]; then
   echo -e "There are an incorrect number of Celery PIDs: $celery_pid_count"
   exit 1
 else

diff --git a/scripts/run_celery.sh b/scripts/run_celery.sh
@@ -3,4 +3,4 @@
 set -e
 
 # Necessary to run as exec so the PID is transferred to Celery for the `SIGTERM` sent from ECS
-exec ddtrace-run celery -A run_celery.notify_celery worker --pidfile="/tmp/celery.pid" --loglevel=INFO --concurrency=16
+exec ddtrace-run celery -A run_celery.notify_celery worker --pidfile="/tmp/celery.pid" --loglevel=INFO --concurrency=CELERY_CONCURRENCY