流程圖

代碼解釋

#!/bin/bash# 如果任何命令返回非零狀態（失敗），立即退出腳本
set -e# 函數：從.env文件加載環境變量
load_env_file() {# 獲取當前腳本所在的目錄路徑local script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"local env_file="$script_dir/.env"# 檢查.env文件是否存在if [ -f "$env_file" ]; thenecho "從以下位置加載環境變量: $env_file"# 設置所有變量為環境變量set -asource "$env_file" set +aelseecho "警告: 未找到.env文件: $env_file"fi
}# 加載環境變量
load_env_file# 取消可能由Docker守護進程設置的HTTP代理
export http_proxy=""; export https_proxy=""; export no_proxy=""; export HTTP_PROXY=""; export HTTPS_PROXY=""; export NO_PROXY=""
# 設置Python模塊搜索路徑
export PYTHONPATH=$(pwd)# 設置動態鏈接庫路徑
export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu/
# 獲取jemalloc庫的路徑
JEMALLOC_PATH=$(pkg-config --variable=libdir jemalloc)/libjemalloc.so# 設置Python解釋器
PY=python3# 如果未設置WS變量或WS小于1，則設置默認工作進程數為1
if [[ -z "$WS" || $WS -lt 1 ]]; thenWS=1
fi# 每個任務執行器和服務器的最大重試次數
MAX_RETRIES=5# 控制終止的標志
STOP=false# 存儲子進程PID的數組
PIDS=()# 設置NLTK數據目錄路徑
export NLTK_DATA="./nltk_data"# 清理函數：處理終止信號
cleanup() {echo "接收到終止信號。正在關閉..."STOP=true# 終止所有子進程for pid in "${PIDS[@]}"; doif kill -0 "$pid" 2>/dev/null; thenecho "正在終止進程 $pid"kill "$pid"fidoneexit 0
}# 捕獲SIGINT和SIGTERM信號，調用cleanup函數
trap cleanup SIGINT SIGTERM# 函數：使用重試邏輯執行task_executor.py
task_exe(){local task_id=$1local retry_count=0while ! $STOP && [ $retry_count -lt $MAX_RETRIES ]; doecho "為任務 $task_id 啟動task_executor.py (嘗試 $((retry_count+1)))"# 使用jemalloc預加載執行Python腳本LD_PRELOAD=$JEMALLOC_PATH $PY rag/svr/task_executor.py "$task_id"EXIT_CODE=$?if [ $EXIT_CODE -eq 0 ]; thenecho "任務 $task_id 的task_executor.py成功退出。"breakelseecho "任務 $task_id 的task_executor.py失敗，退出代碼 $EXIT_CODE。正在重試..." >&2retry_count=$((retry_count + 1))sleep 2fidoneif [ $retry_count -ge $MAX_RETRIES ]; thenecho "任務 $task_id 的task_executor.py在 $MAX_RETRIES 次嘗試后失敗。正在退出..." >&2cleanupfi
}# 函數：使用重試邏輯執行ragflow_server.py
run_server(){local retry_count=0while ! $STOP && [ $retry_count -lt $MAX_RETRIES ]; doecho "啟動ragflow_server.py (嘗試 $((retry_count+1)))"$PY api/ragflow_server.pyEXIT_CODE=$?if [ $EXIT_CODE -eq 0 ]; thenecho "ragflow_server.py成功退出。"breakelseecho "ragflow_server.py失敗，退出代碼 $EXIT_CODE。正在重試..." >&2retry_count=$((retry_count + 1))sleep 2fidoneif [ $retry_count -ge $MAX_RETRIES ]; thenecho "ragflow_server.py在 $MAX_RETRIES 次嘗試后失敗。正在退出..." >&2cleanupfi
}# 啟動任務執行器進程
for ((i=0;i<WS;i++))
dotask_exe "$i" &PIDS+=($!)
done# 啟動主服務器進程
run_server &
PIDS+=($!)# 等待所有后臺進程完成
wait