41 lines
1.6 KiB
Bash
41 lines
1.6 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
UVICORN_APP="${UVICORN_APP:-videoroll.apps.monolith.main:app}"
|
|
UVICORN_HOST="${UVICORN_HOST:-0.0.0.0}"
|
|
UVICORN_PORT="${UVICORN_PORT:-8000}"
|
|
UVICORN_ROOT_PATH="${UVICORN_ROOT_PATH:-}"
|
|
|
|
CELERY_SUB_APP="${CELERY_SUB_APP:-${CELERY_APP:-videoroll.apps.subtitle_service.worker:celery_app}}"
|
|
CELERY_SUB_QUEUE="${CELERY_SUB_QUEUE:-${CELERY_QUEUE:-subtitle}}"
|
|
CELERY_PUB_APP="${CELERY_PUB_APP:-videoroll.apps.bilibili_publisher.worker:celery_app}"
|
|
CELERY_PUB_QUEUE="${CELERY_PUB_QUEUE:-publish}"
|
|
CELERY_PUB_CONCURRENCY="${CELERY_PUB_CONCURRENCY:-1}"
|
|
|
|
echo "Starting celery worker (subtitle): $CELERY_SUB_APP queue=$CELERY_SUB_QUEUE"
|
|
celery -A "$CELERY_SUB_APP" worker -Q "$CELERY_SUB_QUEUE" -l INFO &
|
|
CELERY_SUB_PID=$!
|
|
|
|
echo "Starting celery worker (publish): $CELERY_PUB_APP queue=$CELERY_PUB_QUEUE concurrency=$CELERY_PUB_CONCURRENCY"
|
|
celery -A "$CELERY_PUB_APP" worker -Q "$CELERY_PUB_QUEUE" -l INFO --concurrency "$CELERY_PUB_CONCURRENCY" &
|
|
CELERY_PUB_PID=$!
|
|
|
|
echo "Starting uvicorn: $UVICORN_APP on $UVICORN_HOST:$UVICORN_PORT (root_path=${UVICORN_ROOT_PATH:-/})"
|
|
uvicorn_cmd=(uvicorn "$UVICORN_APP" --host "$UVICORN_HOST" --port "$UVICORN_PORT")
|
|
if [[ -n "$UVICORN_ROOT_PATH" ]]; then
|
|
uvicorn_cmd+=(--root-path "$UVICORN_ROOT_PATH")
|
|
fi
|
|
"${uvicorn_cmd[@]}" &
|
|
UVICORN_PID=$!
|
|
|
|
term_handler() {
|
|
echo "Stopping..."
|
|
kill -TERM "$UVICORN_PID" "$CELERY_SUB_PID" "$CELERY_PUB_PID" 2>/dev/null || true
|
|
wait "$UVICORN_PID" "$CELERY_SUB_PID" "$CELERY_PUB_PID" 2>/dev/null || true
|
|
}
|
|
|
|
trap term_handler SIGINT SIGTERM
|
|
|
|
wait -n "$UVICORN_PID" "$CELERY_SUB_PID" "$CELERY_PUB_PID"
|
|
term_handler
|