超过时间/内存限制时Docker linux kill进程脚本

#!/bin/bash

# 设置超时时间（单位：秒）
timeout=60

# 设置内存限制（单位：字节）
memory_limit=1073741824

# 获取容器中所有进程的ID
pids=$(docker exec <container_id> ps -eo pid --no-headers)

for pid in $pids; do
    # 获取进程的运行时间和内存使用量
    runtime=$(docker exec <container_id> ps -p $pid -o etime --no-headers)
    memory_usage=$(docker exec <container_id> cat /proc/$pid/status | grep VmRSS | awk '{print $2}')

    # 判断是否超过时间限制
    if [[ $runtime =~ ":" ]]; then
        hours=$(echo $runtime | awk -F: '{print $1}')
        minutes=$(echo $runtime | awk -F: '{print $2}')
        seconds=$(echo $runtime | awk -F: '{print $3}')
        total_seconds=$((hours*3600 + minutes*60 + seconds))
    else
        total_seconds=$runtime
    fi

    if [[ $total_seconds -gt $timeout ]]; then
        echo "Process $pid exceeded the time limit. Killing..."
        docker exec <container_id> kill $pid
    fi

    # 判断是否超过内存限制
    if [[ $memory_usage -gt $memory_limit ]]; then
        echo "Process $pid exceeded the memory limit. Killing..."
        docker exec <container_id> kill $pid
    fi
done