如何让主管重新启动挂起的工人？

Question

我有许多由 supervisord 管理的 Python 工作者，如果它们工作正常，它们应该连续打印到标准输出（在每个完成的任务之后）。但是，它们往往会挂起，而且我们很难找到错误。理想情况下，supervisord 会注意到它们在 X 分钟内没有打印并重新启动它们；这些任务是幂等的，所以非优雅重启很好。是否有任何主管功能或插件可以做到这一点？还是其他类似主管的程序开箱即用？

我们已经使用http://superlance.readthedocs.io/en/latest/memmon.html 在内存使用量猛增时终止，这可以缓解一些挂起，但不会导致内存泄漏的挂起仍会导致工作人员陷入停顿。

Answer 1

一种可能的解决方案是将您的 python 脚本包装在一个 bash 脚本中，该脚本将对其进行监视，如果一段时间内没有输出到标准输出，则退出。

例如：

kill-if-hung.sh

#!/usr/bin/env bash
set -e

TIMEOUT=60
LAST_CHANGED="$(date +%s)"

{
    set -e
    while true; do
        sleep 1
        kill -USR1 $$
    done
} &

trap check_output USR1

check_output() {
    CURRENT="$(date +%s)"
    if [[ $((CURRENT - LAST_CHANGED)) -ge $TIMEOUT ]]; then
        echo "Process STDOUT hasn't printed in $TIMEOUT seconds"
        echo "Considering process hung and exiting"
        exit 1
    fi
}

STDOUT_PIPE=$(mktemp -u)
mkfifo $STDOUT_PIPE

trap cleanup EXIT
cleanup() {
    kill -- -$$ # Send TERM to child processes
    [[ -p $STDOUT_PIPE ]] && rm -f $STDOUT_PIPE
}

$@ >$STDOUT_PIPE || exit 2 &

while true; do
    if read tmp; then
        echo "$tmp"
        LAST_CHANGED="$(date +%s)"
    fi
done <$STDOUT_PIPE

然后你将在 supervisord 中运行一个 python 脚本，如：kill-if-hung.sh python -u some-script.py（-u 禁用输出缓冲，或设置PYTHONUNBUFFERED）。

我相信你可以想象一个 python 脚本会做类似的事情。