paramiko執行命令超時的問題

問題：paramiko遠程執行命令，需要等到命令返回信息，如果命令執行時間比較長，返回信息就需要等很久

方案：1、使用nohup + 待執行命令 + & ，使用后台執行的方式，應該可以快速返回

         2、設置paramiko的執行命令等待超時時間

stdin, stdout, stderr = self.client.exec_command(cmd,timeout=10,get_pty=True)

 其實上面的兩種方案都不可行：方案1，需要優化，下面這種直接調用的方式會導致test.sh啟動不起來

stdin, stdout, stderr = self.client.exec_command(‘sh ~/test.sh &’,get_pty=True)

方案2，對於需要等待很久的命令，如果timeout小於程序的執行時間，還是會失敗，命令執行失敗

 

最終的解決方案來自參考1

把執行很久的sh文件A放入另一個sh文件B中，然后paramiko執行文件B。摘錄原文：

paramiko遠程執行后台腳本“阻塞”問題

我寫的遠程命令通道上線之后，發現在遠程腳本中后台再執行另一個腳本，通道會一直等待后台腳本執行完成才會返回，有時甚至會僵死。

1、復現過程如下：

①、編寫測試腳本

腳本1：test.sh

#!/bin/bash
sleep 30
echo test end
exit 0

腳本2：run.sh

#!/bin/bash
bash /tmp/test.sh &
echo run ok!
exit 0

腳本3：test.py

import paramiko
client = paramiko.SSHClient()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(hostname='192.168.1.10', port=22, username='root', password='123456', timeout=300,allow_agent=False,look_for_keys=False)                           
stdin,stdout,stderr=client.exec_command("bash /tmp/run.sh") 
                                             
result_info = ""

for line in  stdout.readlines():
    result_info += line

print result_info

將test.sh和run.sh傳到遠程服務器上，比如放到192.168.1.10:/tmp/下。

②、發起遠程執行

在本地執行 python test.py，會發現整個腳本不會立即打印run ok，而是等30s之后才打印包括test.sh的所有輸出信息。

2、解決辦法

將遠程腳本的標准輸出stdout重定向到錯誤輸出stderr即可，test.py 修改如下：

import paramiko
client = paramiko.SSHClient()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(hostname='192.168.1.10', port=22, username='root', password='123456', timeout=300,allow_agent=False,look_for_keys=False)                           
stdin,stdout,stderr=client.exec_command("bash /tmp/run.sh 1>&2") 
                                             
result_info = ""

for line in  stderr.readlines():
    result_info += line

print result_info

現在執行，就能立即得到結果了。其實原因很簡單，因為bash /tmp/test.sh & 雖然是后台執行，但是依然會產生標准輸出，一旦產生標准輸出，paramiko就會認為命令還未執行完成，且stdout的buffer大於stderr，因此產生等待問題。

這里只要將腳本執行的標准輸出重定向到錯誤輸出（1>&2），然后paramiko就可以使用stderr快速讀取遠程打屏信息了。

 

基於上面paramiko的原理：有stdout輸出，就認為命令沒有執行完成。得出下面的解決方案，對於需要執行很久test.sh，我們首先把標准輸出都發給標准錯誤輸出(1>&2)，然后后台啟動(&)

stdin, stdout, stderr = self.client.exec_command(‘bash ~/test.sh 1>&2 &’,get_pty=True)

項目實踐，還有下面的方案：去掉參數get_pty,這樣就不會回傳標准輸出信息和標准錯誤信息

self.client.exec_command(‘bash ~/test.sh’)

 

 

 

參考：

1、https://zhangge.net/5122.html