87 lines
2.3 KiB
Bash
87 lines
2.3 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# AI-Core gRPC Server Startup Script
|
||
|
|
|
||
|
|
echo "Starting AI-Core Document Parser gRPC Server..."
|
||
|
|
|
||
|
|
# 配置
|
||
|
|
PORT=${1:-50051}
|
||
|
|
|
||
|
|
# 使用虚拟环境
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
|
|
cd "$SCRIPT_DIR"
|
||
|
|
|
||
|
|
# Windows 下使用 PowerShell 的 py 命令或者直接用 venv
|
||
|
|
if [[ "$OSTYPE" == "msys" || "$OSTYPE" == "win32" || -f "venv/Scripts/python.exe" ]]; then
|
||
|
|
if [ -f "venv/Scripts/python.exe" ]; then
|
||
|
|
echo "Using virtual environment Python..."
|
||
|
|
PYTHON_CMD="$SCRIPT_DIR/venv/Scripts/python.exe"
|
||
|
|
elif command -v py &> /dev/null; then
|
||
|
|
echo "Using py launcher..."
|
||
|
|
PYTHON_CMD="py"
|
||
|
|
else
|
||
|
|
echo "Error: Python not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
else
|
||
|
|
# Linux/Mac
|
||
|
|
if [ -d "venv" ]; then
|
||
|
|
echo "Activating virtual environment..."
|
||
|
|
source venv/bin/activate
|
||
|
|
PYTHON_CMD="python"
|
||
|
|
else
|
||
|
|
PYTHON_CMD="python3"
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Using Python: $PYTHON_CMD"
|
||
|
|
$PYTHON_CMD --version
|
||
|
|
|
||
|
|
# Check if requirements are installed
|
||
|
|
$PYTHON_CMD -c "import grpcio" 2>/dev/null
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Installing Python dependencies..."
|
||
|
|
$PYTHON_CMD -m pip install -r requirements.txt
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Error: Failed to install dependencies"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Generate gRPC code if needed
|
||
|
|
if [ ! -f "proto/document_parser_pb2.py" ]; then
|
||
|
|
echo "Generating gRPC code..."
|
||
|
|
$PYTHON_CMD generate_grpc.py
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Error: Failed to generate gRPC code"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
# 检查端口占用并释放
|
||
|
|
echo "Checking port $PORT..."
|
||
|
|
if [[ "$OSTYPE" == "msys" || "$OSTYPE" == "win32" ]]; then
|
||
|
|
# Windows
|
||
|
|
NETSTAT_OUTPUT=$(netstat -ano 2>/dev/null | grep ":$PORT" | grep LISTENING)
|
||
|
|
if [ -n "$NETSTAT_OUTPUT" ]; then
|
||
|
|
echo "Port $PORT is in use, killing process..."
|
||
|
|
PID=$(echo "$NETSTAT_OUTPUT" | awk '{print $NF}' | head -1)
|
||
|
|
if [ -n "$PID" ]; then
|
||
|
|
taskkill //F //PID $PID 2>/dev/null
|
||
|
|
sleep 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
else
|
||
|
|
# Linux/Mac
|
||
|
|
PID=$(lsof -ti:$PORT 2>/dev/null)
|
||
|
|
if [ -n "$PID" ]; then
|
||
|
|
echo "Port $PORT is in use, killing process $PID..."
|
||
|
|
kill $PID 2>/dev/null
|
||
|
|
sleep 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Start the server
|
||
|
|
echo "Starting server on port $PORT..."
|
||
|
|
$PYTHON_CMD main.py --port $PORT --max-workers 10 --log-level INFO
|