entrypoint.sh

#!/bin/bash
set -e

ENVIRONMENT=${ENVIRONMENT:-production}

echo "ENVIRONMENT=${ENVIRONMENT}"

help() {
  echo "Docker entrypoint script."
  echo ""
  echo "Usage:"
  echo ""
  echo "manage -- run CLI to management command"
  echo "shell -- run management shell"
  echo ""
  echo "run_periodic -- start Huey periodic jobs runner with optional code reloading"
  echo "run_worker -- start a Huey workers with optional code reloading"
  echo "run_server -- start Flask / gunicorn server"
  echo ""
  echo "For live code reloading set ENVIRONMENT=development"
  echo ""
  echo "Example:"
  echo ""
  echo "docker <CONTAINER> exec manage database create-tables"
  echo "docker -it <CONTAINER> exec shell"
  echo ""
}

run_worker() {
  echo "Starting Huey worker (${ENVIRONMENT}) ..."
  if [ ${ENVIRONMENT} = "development" ]; then
    exec watchmedo auto-restart -d=./redash/ -d=./dingolytics/ -p=*.py -R -- \
      python -m dingolytics.worker default
  else
    exec python -m dingolytics.worker default
  fi
}

run_periodic() {
  echo "Starting Huey periodic worker (${ENVIRONMENT}) ..."
  if [ ${ENVIRONMENT} = "development" ]; then
    exec watchmedo auto-restart -d=./redash/ -d=./dingolytics/ -p=*.py -R -- \
      python -m dingolytics.worker periodic
  else
    exec python -m dingolytics.worker periodic
  fi
}

run_server() {
  echo "Starting application server (${ENVIRONMENT}) ..."
  if [ ${ENVIRONMENT} = "development" ]; then
    export FLASK_DEBUG=1
    exec ./manage.py runserver -h 0.0.0.0
  else
    # Recycle gunicorn workers every n-th request.
    # See http://docs.gunicorn.org/en/stable/settings.html#max-requests
    MAX_REQUESTS=${MAX_REQUESTS:-1000}
    MAX_REQUESTS_JITTER=${MAX_REQUESTS_JITTER:-100}
    TIMEOUT=${GUNICORN_TIMEOUT:-60}
    exec gunicorn -b 0.0.0.0:5000 --name redash \
      -w${GUNICORN_WEB_WORKERS:-4} redash.main:app \
      --max-requests $MAX_REQUESTS --max-requests-jitter \
      $MAX_REQUESTS_JITTER --timeout $TIMEOUT
  fi
}

case "$1" in
  help)
    shift
    help
    ;;
  manage)
    shift
    exec ./manage.py $*
    ;;
  shell)
    exec ./manage.py shell
    ;;
  run_server)
    shift
    run_server
    ;;
  run_worker)
    shift
    run_worker
    ;;
  run_periodic)
    shift
    run_periodic
    ;;
  *)
    exec "$@"
    ;;
esac