mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
59619e9eaa
* feat: remote setup with docker build option * chore: update documentation * chore: make script run in non tty * chore: add warning about slow build * chore: add more documentation * feat: add FASTAPI_WORKERS parameter * feat: add scaling docs * feat: add update script * fix: fix semver options in update_remote.sh
85 lines
2.9 KiB
Bash
Executable file
85 lines
2.9 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
###############################################################################
|
|
### CONFIGURATION
|
|
###############################################################################
|
|
|
|
BASE_DIR="$(cd "$(dirname "$(dirname "${BASH_SOURCE[0]}")")" && pwd)"
|
|
ENV_FILE="$BASE_DIR/api/.env"
|
|
|
|
ARQ_WORKERS=${ARQ_WORKERS:-1}
|
|
FASTAPI_WORKERS=${FASTAPI_WORKERS:-1}
|
|
UVICORN_BASE_PORT=${UVICORN_BASE_PORT:-8000}
|
|
|
|
cd "$BASE_DIR"
|
|
echo "Starting Dograh Services (DOCKER) at $(date) in BASE_DIR: ${BASE_DIR}"
|
|
|
|
###############################################################################
|
|
### 1) Load env file if mounted (env normally comes from docker-compose)
|
|
###############################################################################
|
|
|
|
if [[ -f "$ENV_FILE" ]]; then
|
|
set -a && . "$ENV_FILE" && set +a
|
|
fi
|
|
|
|
###############################################################################
|
|
### 2) Run migrations
|
|
###############################################################################
|
|
|
|
alembic -c "$BASE_DIR/api/alembic.ini" upgrade head
|
|
|
|
###############################################################################
|
|
### 3) Signal handling — forward TERM/INT to children for clean docker stop
|
|
###############################################################################
|
|
|
|
pids=()
|
|
|
|
shutdown() {
|
|
echo "Received shutdown signal, stopping services..."
|
|
for pid in "${pids[@]}"; do
|
|
kill -TERM "$pid" 2>/dev/null || true
|
|
done
|
|
wait
|
|
exit 0
|
|
}
|
|
|
|
trap shutdown TERM INT
|
|
|
|
start() {
|
|
local name=$1
|
|
shift
|
|
echo "→ Starting $name"
|
|
"$@" &
|
|
pids+=($!)
|
|
echo " $name PID $!"
|
|
}
|
|
|
|
###############################################################################
|
|
### 4) Start services (logs go to stdout for `docker logs`)
|
|
###############################################################################
|
|
|
|
start ari_manager python -m api.services.telephony.ari_manager
|
|
start campaign_orchestrator python -m api.services.campaign.campaign_orchestrator
|
|
|
|
# Spawn FASTAPI_WORKERS independent uvicorn processes on consecutive ports
|
|
# starting at UVICORN_BASE_PORT. nginx upstream (configured in setup_remote.sh)
|
|
# balances across them with least_conn — better than uvicorn --workers for
|
|
# long-lived WebSocket connections, which would otherwise stick to whichever
|
|
# worker accepted them first.
|
|
for ((i=0; i<FASTAPI_WORKERS; i++)); do
|
|
port=$((UVICORN_BASE_PORT + i))
|
|
start "uvicorn$i" uvicorn api.app:app --host 0.0.0.0 --port "$port" --workers 1
|
|
done
|
|
|
|
for ((i=1; i<=ARQ_WORKERS; i++)); do
|
|
start "arq$i" python -m arq api.tasks.arq.WorkerSettings --custom-log-dict api.tasks.arq.LOG_CONFIG
|
|
done
|
|
|
|
###############################################################################
|
|
### 5) Wait — if any service exits, tear the container down so docker restarts
|
|
###############################################################################
|
|
|
|
wait -n
|
|
echo "A service exited; tearing down container."
|
|
shutdown
|