From bf6140168834694ed5ffb4099a79eef61ea8eedf Mon Sep 17 00:00:00 2001
From: Leon <147289645+LeoninCS@users.noreply.github.com>
Date: Mon, 5 Jan 2026 15:13:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E4=BA=86worker?=
 =?UTF-8?q?=E7=9A=84=E5=90=AF=E5=8A=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 start.sh | 31 ++++++++++++++++++++++++++-----
 1 file changed, 26 insertions(+), 5 deletions(-)

diff --git a/start.sh b/start.sh
index 578901f..fd9d070 100644
--- a/start.sh
+++ b/start.sh
@@ -11,6 +11,7 @@ RUN_DIR="${RUN_DIR:-$BACKEND_DIR/.run}"
 START_REDIS="${START_REDIS:-1}"
 START_RABBITMQ="${START_RABBITMQ:-1}"
 START_BACKEND="${START_BACKEND:-1}"
+START_WORKER="${START_WORKER:-1}"
 START_FRONTEND="${START_FRONTEND:-1}"
 
 # docker compose (for dependencies like RabbitMQ):
@@ -47,6 +48,7 @@ require_cmd() {
 }
 
 BACKEND_PID=""
+WORKER_PID=""
 FRONTEND_PID=""
 cleanup() {
   set +e
@@ -54,6 +56,10 @@ cleanup() {
     echo "[start.sh] Stopping frontend (pid=$FRONTEND_PID)"
     kill "$FRONTEND_PID" >/dev/null 2>&1 || true
   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
     echo "[start.sh] Stopping backend (pid=$BACKEND_PID)"
     kill "$BACKEND_PID" >/dev/null 2>&1 || true
@@ -64,6 +70,9 @@ cleanup() {
     if [ -n "${FRONTEND_PID:-}" ]; then
       taskkill //PID "$FRONTEND_PID" //T //F >/dev/null 2>&1 || true
     fi
+    if [ -n "${WORKER_PID:-}" ]; then
+      taskkill //PID "$WORKER_PID" //T //F >/dev/null 2>&1 || true
+    fi
     if [ -n "${BACKEND_PID:-}" ]; then
       taskkill //PID "$BACKEND_PID" //T //F >/dev/null 2>&1 || true
     fi
@@ -76,7 +85,7 @@ cleanup() {
 }
 trap cleanup INT TERM EXIT
 
-if [ "$START_BACKEND" = "1" ]; then
+if [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ]; then
   require_dir backend "$BACKEND_DIR"
   mkdir -p "$RUN_DIR"
   require_cmd go
@@ -166,6 +175,14 @@ start_backend_bg() {
   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() {
   if [ "$FRONTEND_INSTALL" = "1" ] || { [ "$FRONTEND_INSTALL" = "auto" ] && [ ! -d "$FRONTEND_DIR/node_modules" ]; }; then
     echo "[start.sh] Installing frontend deps"
@@ -179,11 +196,11 @@ start_frontend_bg() {
   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
 fi
 
-if [ "$START_REDIS" = "1" ] && [ "$START_BACKEND" = "1" ]; then
+if [ "$START_REDIS" = "1" ] && { [ "$START_BACKEND" = "1" ] || [ "$START_WORKER" = "1" ]; }; then
   start_redis
 fi
 
@@ -191,13 +208,17 @@ if [ "$START_BACKEND" = "1" ]; then
   start_backend_bg
 fi
 
+if [ "$START_WORKER" = "1" ]; then
+  start_worker_bg
+fi
+
 if [ "$START_FRONTEND" = "1" ]; then
   start_frontend_bg
 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."
   wait
 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