翻了下源码,目测没有看到相关的功能。
大概的思路是自己构造socket,然后传递给paramiko,通过判断socket的状态来做连接健康检查。
主要代码如下:
client = paramiko.SSHClient()
client.load_system_host_keys()
client.set_missing_host_key_policy(paramiko.WarningPolicy())
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
from paramiko.util import retry_on_signal
retry_on_signal(lambda: sock.connect((ssh_host, ssh_port)))
client.connect(
ssh_host,
ssh_port,
username=ssh_user,
password=ssh_pass,
sock=sock,
timeout=10.0 # 连接超时时间
)
while True:
try:
# 值正常为1,网络不通之后,值会增长,每次心跳失败等待时间会翻倍,最大超时时间120s,大约12次后则报OSError
# 网络恢复,下次心跳后,值恢复为1
# 假设心跳时间为3秒,则为3、6、12、24、48、96、120、120、120、120、120、120
# 超时加上每次判断超时都需要一点时间,所以网络中断后,大概约16分钟才会退出,这段时间网络恢复可有望重新连接回去
# 如若此时先收到了服务器的请求,则值回降低,请求需要等待下次心跳恢复后,获取返回结果
ret = sock.getsockopt(socket.IPPROTO_TCP, socket.TCP_INFO)
# print(f"{datetime.now()}: {ret} {type(ret)} {dir(ret)}") # int
if ret != 1:
print("{}: network maybe not work well, tcp info {}".format(datetime.now(), ret))
time.sleep(3)
continue
except OSError as err:
print(f"{datetime.now()}: socket dead, {err}")
exit(1)