start_celery.sh

#!/bin/bash

# Celery Worker 启动脚本
# 用于启动AI-Dev-Insight项目的Celery worker

echo "🚀 启动 AI-Dev-Insight Celery Worker..."

# 检查是否已有Celery worker在运行
EXISTING_WORKER=$(ps aux | grep "celery.*worker" | grep -v grep | wc -l)
if [ $EXISTING_WORKER -gt 0 ]; then
    echo "⚠️  检测到已有Celery worker在运行"
    echo "📋 当前运行的worker进程："
    ps aux | grep "celery.*worker" | grep -v grep
    read -p "是否要停止现有worker并重新启动？(y/N): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        echo "🛑 停止现有worker..."
        pkill -f "celery.*worker"
        sleep 2
    else
        echo "❌ 取消启动，保持现有worker运行"
        exit 0
    fi
fi

# 检查Redis连接
echo "📡 检查Redis连接..."
python -c "
import redis
from config.config import get_redis_url
try:
    client = redis.from_url(get_redis_url())
    result = client.ping()
    print(f'✅ Redis连接成功: {result}')
except Exception as e:
    print(f'❌ Redis连接失败: {e}')
    exit(1)
"

if [ $? -ne 0 ]; then
    echo "❌ Redis连接失败，请检查Redis服务是否运行"
    exit 1
fi

# 启动Celery worker
echo "🔧 启动Celery worker..."
echo "💡 提示：按 Ctrl+C 可以停止worker"
celery -A tasks.celery_app worker \
    --loglevel=info \
    --queues=crawl,default,supplement,schedule,maintenance,stats \
    --concurrency=4 \
    --max-tasks-per-child=1000 \
    --time-limit=300 \
    --soft-time-limit=240

echo "🛑 Celery worker 已停止"