huanghaosheng/VLoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加了worker的启动

Browse Source
This commit is contained in:
Leon
2026-01-05 15:13:24 +08:00
parent 39645868a2
commit bf61401688
1 changed files with 26 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
start.sh
View File

@@ -11,6 +11,7 @@ RUN_DIR="${RUN_DIR:-$BACKEND_DIR/.run}"
START_REDIS="${START_REDIS:-1}" START_REDIS="${START_REDIS:-1}"
START_RABBITMQ="${START_RABBITMQ:-1}" START_RABBITMQ="${START_RABBITMQ:-1}"
START_BACKEND="${START_BACKEND:-1}" START_BACKEND="${START_BACKEND:-1}"
START_WORKER="${START_WORKER:-1}"
START_FRONTEND="${START_FRONTEND:-1}" START_FRONTEND="${START_FRONTEND:-1}"
# docker compose (for dependencies like RabbitMQ): # docker compose (for dependencies like RabbitMQ):
@@ -47,6 +48,7 @@ require_cmd() {
} }
BACKEND_PID="" BACKEND_PID=""
WORKER_PID=""
FRONTEND_PID="" FRONTEND_PID=""
cleanup() { cleanup() {
set +e set +e
@@ -54,6 +56,10 @@ cleanup() {
echo "[start.sh] Stopping frontend (pid=$FRONTEND_PID)" echo "[start.sh] Stopping frontend (pid=$FRONTEND_PID)"
kill "$FRONTEND_PID" >/dev/null 2>&1 || true kill "$FRONTEND_PID" >/dev/null 2>&1 || true
fi fi
if [ -n "${WORKER_PID:-}" ]; then
echo "[start.sh] Stopping worker (pid=$WORKER_PID)"
kill "$WORKER_PID" >/dev/null 2>&1 || true
fi
if [ -n "${BACKEND_PID:-}" ]; then if [ -n "${BACKEND_PID:-}" ]; then
echo "[start.sh] Stopping backend (pid=$BACKEND_PID)" echo "[start.sh] Stopping backend (pid=$BACKEND_PID)"
kill "$BACKEND_PID" >/dev/null 2>&1 || true kill "$BACKEND_PID" >/dev/null 2>&1 || true
@@ -64,6 +70,9 @@ cleanup() {
if [ -n "${FRONTEND_PID:-}" ]; then if [ -n "${FRONTEND_PID:-}" ]; then
taskkill //PID "$FRONTEND_PID" //T //F >/dev/null 2>&1 || true taskkill //PID "$FRONTEND_PID" //T //F >/dev/null 2>&1 || true
fi fi
if [ -n "${WORKER_PID:-}" ]; then
taskkill //PID "$WORKER_PID" //T //F >/dev/null 2>&1 || true
fi
if [ -n "${BACKEND_PID:-}" ]; then if [ -n "${BACKEND_PID:-}" ]; then
taskkill //PID "$BACKEND_PID" //T //F >/dev/null 2>&1 || true taskkill //PID "$BACKEND_PID" //T //F >/dev/null 2>&1 || true
fi fi
@@ -76,7 +85,7 @@ cleanup() {
} }
trap cleanup INT TERM EXIT trap cleanup INT TERM EXIT
if [ "$START_BACKEND" = "1" ]; then if [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ]; then
require_dir backend "$BACKEND_DIR" require_dir backend "$BACKEND_DIR"
mkdir -p "$RUN_DIR" mkdir -p "$RUN_DIR"
require_cmd go require_cmd go
@@ -166,6 +175,14 @@ start_backend_bg() {
echo "[start.sh] Backend PID: $BACKEND_PID" echo "[start.sh] Backend PID: $BACKEND_PID"
} }
start_worker_bg() {
echo "[start.sh] Starting worker (background)"
(cd "$BACKEND_DIR" && go run ./cmd/worker) &
WORKER_PID=$!
echo "$WORKER_PID" >"$RUN_DIR/worker.pid"
echo "[start.sh] Worker PID: $WORKER_PID"
}
start_frontend_bg() { start_frontend_bg() {
if [ "$FRONTEND_INSTALL" = "1" ] || { [ "$FRONTEND_INSTALL" = "auto" ] && [ ! -d "$FRONTEND_DIR/node_modules" ]; }; then if [ "$FRONTEND_INSTALL" = "1" ] || { [ "$FRONTEND_INSTALL" = "auto" ] && [ ! -d "$FRONTEND_DIR/node_modules" ]; }; then
echo "[start.sh] Installing frontend deps" echo "[start.sh] Installing frontend deps"
@@ -179,11 +196,11 @@ start_frontend_bg() {
echo "[start.sh] Frontend PID: $FRONTEND_PID" echo "[start.sh] Frontend PID: $FRONTEND_PID"
} }
if [ "$START_RABBITMQ" = "1" ] && [ "$START_BACKEND" = "1" ]; then if [ "$START_RABBITMQ" = "1" ] && { [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ]; }; then
start_rabbitmq_compose start_rabbitmq_compose
fi fi
if [ "$START_REDIS" = "1" ] && [ "$START_BACKEND" = "1" ]; then if [ "$START_REDIS" = "1" ] && { [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ]; }; then
start_redis start_redis
fi fi
@@ -191,13 +208,17 @@ if [ "$START_BACKEND" = "1" ]; then
start_backend_bg start_backend_bg
fi fi
if [ "$START_WORKER" = "1" ]; then
start_worker_bg
fi
if [ "$START_FRONTEND" = "1" ]; then if [ "$START_FRONTEND" = "1" ]; then
start_frontend_bg start_frontend_bg
fi fi
if [ "$START_BACKEND" = "1" ] || [ "$START_FRONTEND" = "1" ]; then if [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ] || [ "$START_FRONTEND" = "1" ]; then
echo "[start.sh] Press Ctrl+C to stop." echo "[start.sh] Press Ctrl+C to stop."
wait wait
else else
echo "[start.sh] Nothing to start. Set START_BACKEND=1 and/or START_FRONTEND=1." echo "[start.sh] Nothing to start. Set START_BACKEND=1 and/or START_WORKER=1 and/or START_FRONTEND=1."
fi fi