Add healthchecks to docker-compose.yml

Dockerfile

@@ -1,9 +1,9 @@
 FROM docker.io/library/python:3.9-slim
 
-ARG BUILD_DEPS="build-essential curl"
+ARG BUILD_DEPS="build-essential"
 
 RUN set -ex \
-  && apt-get update && apt-get -y --no-install-recommends install $BUILD_DEPS libpq-dev netcat make git \
+  && apt-get update && apt-get -y --no-install-recommends install $BUILD_DEPS libpq-dev make git curl \
   && rm -rf /var/lib/apt/lists/* \
   && curl -sSL "https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py" | python \
   && . $HOME/.poetry/env \

docker-compose.yml

@@ -1,4 +1,4 @@
-version: "3"
+version: "3.8"
 
 x-gunicorn-master: &x-gunicorn-master
   build: .
@@ -22,6 +22,12 @@ x-gunicorn-master: &x-gunicorn-master
     - SQL_USER=postgres
     - SQL_DATABASE=postgres
     - EMAIL_PROVIDER=CONSOLE
+  depends_on:
+    database:
+      condition: service_healthy
+    redis:
+      condition: service_started
+
 
 services:
   redis:
@@ -32,6 +38,11 @@ services:
     environment:
       - POSTGRES_EXTENSIONS=citext
       - POSTGRES_HOST_AUTH_METHOD=trust
+    healthcheck:
+      test: ["CMD", "pg_isready", "-U", "postgres"]
+      interval: 1s
+      timeout: 30s
+      retries: 30
 
   shell:
     image: ghcr.io/ractf/shell:latest
@@ -55,10 +66,23 @@ services:
     command: gunicorn backend.wsgi:application
     depends_on:
       - database
+    healthcheck:
+      test: ["CMD", "curl", "--silent", "--fail", "http://localhost:8000/api/v2/healthcheck/?format=json"]
+      interval: 10s
+      timeout: 50s
+      retries: 5
 
   sockets:
     <<: *x-gunicorn-master
     entrypoint: /app/entrypoints/sockets.sh
     command: gunicorn backend.asgi:application
     depends_on:
-      - backend
+      backend:
+        condition: service_healthy
+
+  watchtower:
+    image: docker.io/containrrr/watchtower
+    volumes:
+      - /var/run/docker.sock:/var/run/docker.sock
+      - /root/.docker/config.json:/config.json
+    command: --interval 5

entrypoints/backend.sh

@@ -8,23 +8,10 @@ else
   rm $prometheus_multiproc_dir/* -rf
 fi
 
-echo "Waiting for postgres... "
-while ! nc -z $SQL_HOST $SQL_PORT
-do
-  sleep 0.69
-done
-echo "Done."
-
 echo "Running migrations..."
 /app/src/manage.py migrate
 echo "Done."
 
-if [ "$LOAD_FIXTURES" ]
-then
-  /app/src/manage.py flush --no-input
-  /app/src/manage.py loaddata test_fixtures
-fi
-
 export GUNICORN_CMD_ARGS="--chdir=/app/src/ --reload --workers=4 --bind=0.0.0.0:8000 --worker-class=gthread"
 if [ -f /etc/newrelic.ini ]
 then

entrypoints/sockets.sh

@@ -1,19 +1,5 @@
 #!/usr/bin/env sh
 
-echo "Waiting for postgres... "
-while ! nc -z $SQL_HOST $SQL_PORT
-do
-  sleep 0.69
-done
-echo "Done."
-
-echo "Waiting for django... "
-while ! nc -z backend 8000
-do
-  sleep 0.69
-done
-echo "Done."
-
 export GUNICORN_CMD_ARGS="--chdir=/app/src/ --reload --workers=4 --bind=0.0.0.0:8000 --worker-class=uvicorn.workers.UvicornWorker"
 if [ -f /etc/newrelic.ini ]
 then