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

在一些对安全性要求较高的场景下,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是否有更简便的方式来实现主子进程的信息同步.

如果子进程中进行的是非常重要的工作,必须保证即使主进程意外退出子进程也要完成执行,这种情况可以使用Shell的内置命令trap来实现。

...
(
        # Starting ignoring HUP signal
        trap "" HUP
        real script to do something
        echo $? > $retTmp
        rm -f $lock;
)&
...
通过ps命令可以验证,当父进程退出后,子进程继续执行完成后退出。这里需要注意的是,当主进程退出时,标准输出就没有了,需要重定向stdout和stderr到文件中,否则命令会报错。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值