mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
77 lines
2.5 KiB
Bash
77 lines
2.5 KiB
Bash
|
#!/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
|
||
|
|
start uvicorn uvicorn api.app:app --host 0.0.0.0 --port "$UVICORN_BASE_PORT" --workers "$FASTAPI_WORKERS"
|
||
|
|
|
||
|
|
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
|