首页 > 代码库 > 防止shell脚本长时间运行导致ssh超时

防止shell脚本长时间运行导致ssh超时

2024-07-03 05:43:56   216人阅读

在一些对安全性要求较高的场景下，ssh的超时时间是管理员预先设置好的，在闲置一段时间后ssh连接会自动断开。这种情况下如果通过ssh执行脚本，而脚本运行时间又比较长的话，会导致ssh客户端和服务器长时间无交互而超时，命令执行失败。
使用bash子进程可以解决这种问题，思路是由子进程执行具体的逻辑代码，而由主进程来监控子进程的执行状态，同时向控制台输出字符来keep alive。
bash创建子进程有多重方式，这里使用“()”，然后用“&”将其放在后台执行。虽然通过“job -p”可以获取到后台进程的进程ID并使用“wait $PID”的方式能够监控到子进程的返回状态,如下面的代码。但这种方式相当于阻塞了主进程，无法执行其他动作。
for pid in $(jobs -p); do
  wait $pid
done
解决办法是新建一个标志文件，并将子进程返回值保存在另外一个临时文件中，主进程通过标志文件和返回值来获取相关信息，执行其余逻辑。
简化代码如下，主进程在等待时会一直打印一个旋转的进度标志。
function doSomething()
{
        local retTmp=$(mktemp)
        local lock="/tmp/do.lock"
        touch $lock
        (
                real script to do something
                echo $? > $retTmp
                rm -f $lock;
        )&
        while [ -f $lock ]; do
                sleep 0.1
                printf "Please wait... %s \r" $f
                let "t=10#$(date +%N) / 100000000 % 4"
                case $t in
                      0) f="/";;
                      1) f="-";;
                      2) f="\\";;
                      3) f="|";;
                esac
        done
        echo

        local retcode=$(cat $retTmp)
        rm -f $retTmp
        return $retcode
}

这种方式类似于通过fork函数实现相关逻辑，感觉上稍微有点笨重，不清楚bash是否有更简便的方式来实现主子进程的信息同步


联系
我们