#!/bin/bash

######################
## 因为内容相似 会影响在线服务（打满ES）所以暂时关闭
######################

# 设置 Python 路径，让脚本能找到 db_service, config, deepwalk 等模块
export PYTHONPATH=/home/tw/recommendation/offline_tasks:$PYTHONPATH

cd /home/tw/recommendation/offline_tasks

# mkdir bak___before_rm_run_all_py
# mv output logs nohup.out bak___before_rm_run_all_py/
# mkdir output
# mkdir logs

# 检测并激活conda环境
if [ "$CONDA_DEFAULT_ENV" != "tw" ]; then
    echo "当前环境不是 tw，正在激活 tw 环境..."
    source /home/tw/miniconda3/etc/profile.d/conda.sh
    conda activate tw
    echo "已激活 tw 环境"
else
    echo "当前已经在 tw 环境中，无需重复激活"
fi

# ============================================================================
# 配置区域
# ============================================================================

# 算法参数
LOOKBACK_DAYS=768
TOP_N=50
DEBUG_MODE="--debug"  # 留空则不开启debug

# Redis配置

# 内存监控阈值
MEM_WARN_THRESHOLD=25  # GB
MEM_KILL_THRESHOLD=35  # GB

# ============================================================================
# 工具函数
# ============================================================================

# 内存监控函数
check_memory() {
    local pid=$1
    local task_name=$2
    
    while kill -0 $pid 2>/dev/null; do
        local mem_mb=$(ps -p $pid -o rss= 2>/dev/null | awk '{print int($1/1024)}')
        
        if [ -n "$mem_mb" ]; then
            local mem_gb=$(echo "scale=2; $mem_mb/1024" | bc)
            local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
            
            if [ $(echo "$mem_gb >= $MEM_KILL_THRESHOLD" | bc) -eq 1 ]; then
                echo "[$timestamp] ❌ [$task_name] 内存超限！${mem_gb}GB, 强制终止" | tee -a logs/memory_monitor.log
                kill -9 $pid
                break
            elif [ $(echo "$mem_gb >= $MEM_WARN_THRESHOLD" | bc) -eq 1 ]; then
                echo "[$timestamp] ⚠️  [$task_name] 内存警告: ${mem_gb}GB" | tee -a logs/memory_monitor.log
            fi
        fi
        
        sleep 60
    done
}

# 运行任务函数
run_task() {
    local task_name=$1
    local task_cmd=$2
    
    echo ""
    echo "======================================================================"
    echo "[$task_name] 开始 - $(date '+%Y-%m-%d %H:%M:%S')"
    echo "======================================================================"
    
    eval $task_cmd &
    local pid=$!
    
    # 启动内存监控
    check_memory $pid "$task_name" &
    local monitor_pid=$!
    
    # 等待任务完成
    wait $pid
    local exit_code=$?
    
    # 停止内存监控
    kill $monitor_pid 2>/dev/null
    
    if [ $exit_code -eq 0 ]; then
        echo "✓ [$task_name] 完成"
        return 0
    else
        echo "✗ [$task_name] 失败，退出码: $exit_code"
        return $exit_code
    fi
}

# ============================================================================
# 环境准备
# ============================================================================

# 清理旧进程
ps -ef | grep "python3.*scripts" | grep -v grep | awk '{print $2}' | xargs kill -9 2>/dev/null

# 创建必要目录
mkdir -p logs output

echo "======================================================================"
echo "开始运行离线推荐任务 - $(date '+%Y-%m-%d %H:%M:%S')"
echo "配置: lookback_days=$LOOKBACK_DAYS, top_n=$TOP_N"
echo "内存监控: 警告=${MEM_WARN_THRESHOLD}GB, 终止=${MEM_KILL_THRESHOLD}GB"
echo "======================================================================"

# ============================================================================
# 前置任务
# ============================================================================

# 前置任务1: 获取商品属性
run_task "前置任务1: 获取商品属性" \
    "python3 scripts/fetch_item_attributes.py $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  商品属性获取失败，但继续执行"
fi

# 前置任务2: 生成Session文件
run_task "前置任务2: 生成Session文件" \
    "python3 scripts/generate_session.py --lookback_days $LOOKBACK_DAYS --format both $DEBUG_MODE"
if [ $? -ne 0 ]; then  
    echo "❌ Session文件生成失败，退出"
    exit 1
fi

# 前置任务3: C++ Swing算法
echo ""
echo "======================================================================"
echo "[前置任务3: C++ Swing算法] 开始 - $(date '+%Y-%m-%d %H:%M:%S')"
echo "======================================================================"
cd collaboration
bash run.sh
SWING_EXIT=$?
cd ..

if [ $SWING_EXIT -eq 0 ]; then
    echo "✓ [前置任务3: C++ Swing算法] 完成"
else
    echo "⚠️  [前置任务3: C++ Swing算法] 失败，但继续执行"
fi

# ============================================================================
# i2i相似度任务
# ============================================================================

# Task 1: Python Swing算法
run_task "Task 1: Python Swing算法" \
    "python3 scripts/i2i_swing.py --lookback_days $LOOKBACK_DAYS --top_n $TOP_N --use_daily_session $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  Python Swing失败，但继续执行"
fi

# Task 2: Session W2V
run_task "Task 2: Session W2V" \
    "python3 scripts/i2i_session_w2v.py --lookback_days $LOOKBACK_DAYS --top_n $TOP_N --save_model $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  Session W2V失败，但继续执行"
fi

# Task 3: DeepWalk
run_task "Task 3: DeepWalk" \
    "python3 scripts/i2i_deepwalk.py --lookback_days $LOOKBACK_DAYS --top_n $TOP_N --save_model --save_graph $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  DeepWalk失败，但继续执行"
fi
# 
# # Task 4: 内容相似度
# run_task "Task 4: 内容相似度" \
#     "python3 scripts/i2i_content_similar.py --top_n $TOP_N $DEBUG_MODE"
# if [ $? -ne 0 ]; then
#     echo "⚠️  内容相似度失败，但继续执行"
# fi
# 
# Task 5: Item行为相似度
run_task "Task 5: Item行为相似度" \
    "python3 scripts/i2i_item_behavior.py --lookback_days $LOOKBACK_DAYS --top_n $TOP_N $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  Item行为相似度失败，但继续执行"
fi

# Task 6: Tag分类相似度
run_task "Task 6: Tag分类相似度" \
    "python3 scripts/tag_category_similar.py --lookback_days $LOOKBACK_DAYS --top_n $TOP_N $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  Tag分类相似度失败，但继续执行"
fi

# ============================================================================
# 兴趣聚合任务
# ============================================================================

# Task 7: 兴趣聚合
run_task "Task 7: 兴趣聚合" \
    "python3 scripts/interest_aggregation.py --lookback_days $LOOKBACK_DAYS --top_n 1000 $DEBUG_MODE"
if [ $? -ne 0 ]; then
    echo "⚠️  兴趣聚合失败，但继续执行"
fi

# ============================================================================
# 加载到Redis
# ============================================================================

echo ""
echo "======================================================================"
echo "[加载到Redis] 开始 - $(date '+%Y-%m-%d %H:%M:%S')"
echo "======================================================================"

python3 scripts/load_index_to_redis.py
LOAD_EXIT=$?

if [ $LOAD_EXIT -eq 0 ]; then
    echo "✓ [加载到Redis] 完成"
else
    echo "❌ [加载到Redis] 失败，退出码: $LOAD_EXIT"
    exit 1
fi

# ============================================================================
# 完成
# ============================================================================

echo ""
echo "======================================================================"
echo "所有任务完成 - $(date '+%Y-%m-%d %H:%M:%S')"
echo "======================================================================"
echo ""
echo "输出文件位置:"
echo "  - 商品属性: output/item_attributes_mappings.json"
echo "  - Session文件: output/session.txt.*"
echo "  - C++ Swing: collaboration/output/swing_similar.txt"
echo "  - Python算法: output/i2i_*.txt"
echo "  - Item行为相似度: output/i2i_item_behavior_*.txt"
echo "  - Tag分类相似度: output/tag_category_similar_*.txt"
echo "  - 兴趣聚合: output/interest_aggregation_*.txt"
echo "  - 日志: logs/"
echo ""