#!/bin/bash

echo "🔍 查找所有 multiprocessing.spawn 相关进程..."

# 查找所有 multiprocessing.spawn 进程
SPAWN_PROCESSES=$(ps aux | grep "multiprocessing.spawn" | grep -v grep | awk '{print $2}')

if [ -z "$SPAWN_PROCESSES" ]; then
    echo "✅ 没有找到 multiprocessing.spawn 进程"
else
    echo "📋 找到以下 multiprocessing.spawn 进程:"
    ps aux | grep "multiprocessing.spawn" | grep -v grep
    
    echo ""
    echo "🛑 强制终止所有 multiprocessing.spawn 进程..."
    for pid in $SPAWN_PROCESSES; do
        echo "强制终止进程 $pid..."
        kill -9 $pid 2>/dev/null
    done
    
    echo "✅ multiprocessing.spawn 进程已清理"
fi

# 查找所有 multiprocessing.resource_tracker 进程
echo ""
echo "🔍 查找所有 multiprocessing.resource_tracker 相关进程..."

TRACKER_PROCESSES=$(ps aux | grep "multiprocessing.resource_tracker" | grep -v grep | awk '{print $2}')

if [ -z "$TRACKER_PROCESSES" ]; then
    echo "✅ 没有找到 multiprocessing.resource_tracker 进程"
else
    echo "📋 找到以下 multiprocessing.resource_tracker 进程:"
    ps aux | grep "multiprocessing.resource_tracker" | grep -v grep
    
    echo ""
    echo "🛑 强制终止所有 multiprocessing.resource_tracker 进程..."
    for pid in $TRACKER_PROCESSES; do
        echo "强制终止进程 $pid..."
        kill -9 $pid 2>/dev/null
    done
    
    echo "✅ multiprocessing.resource_tracker 进程已清理"
fi

# 查找所有 python agent 相关进程
echo ""
echo "🔍 查找所有 python agent 相关进程..."

AGENT_PROCESSES=$(ps aux | grep "python.*agent" | grep -v grep | awk '{print $2}')

if [ -z "$AGENT_PROCESSES" ]; then
    echo "✅ 没有找到 python agent 进程"
else
    echo "📋 找到以下 python agent 进程:"
    ps aux | grep "python.*agent" | grep -v grep
    
    echo ""
    echo "🛑 强制终止所有 python agent 进程..."
    for pid in $AGENT_PROCESSES; do
        echo "强制终止进程 $pid..."
        kill -9 $pid 2>/dev/null
    done
    
    echo "✅ python agent 进程已清理"
fi

echo ""
echo "🏁 清理完成"