如何使用 paramiko 与一个 ssh 会话并行运行

Question

Previous details
我需要在交换机上 ssh 以 ping 不同的主机。早些时候我为每个主机启动了一个线程，但结果很容易超过最大 ssh 连接数，所以我根据this 创建了一个交互式 shell 会话。但是当我并行运行时，它在发送第一个命令后一直挂在那里。我不知道如何解决这个问题。
下面列出的简化代码：

import paramiko
import time
from paramiko import SSHClient
from multiprocessing.dummy import Pool

def rping(src, user, passwd, dst):

    def command(des):
        chan.send('ping -s 64 -t 1500 %s\r\n' % des)
        time.sleep(3)
        resp = chan.recv(9999)
        print resp

    ssh = SSHClient()
    ssh.load_system_host_keys()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(src, 22, user, passwd, timeout=3)

    chan = ssh.invoke_shell()

    pool = Pool()
    pool.map(command, dst)
    pool.close()
    pool.join()


if __name__ == '__main__':
    print time.ctime()

    src = '10.130.1.121'
    user = 'user'
    passwd = 'password'
    dst = ['10.130.1.122', '10.130.1.123', 10.130.1.124'']
    rping(src, user, passwd, dst)

    print time.ctime()

Answer 1

请记住，使用multiprocessing.dummy 实际上是在使用threading 模块。 鉴于此，您的 command 函数不会返回任何内容。

def rping(src, user, passwd, dst):
    def command(des):
        chan = ssh.invoke_shell()  # new channel 
        chan.send('ping -t 64 -s 1500 -c 3 %s\r\n' % des)
        time.sleep(3)
        resp = chan.recv(9999)
        chan.close()  # close channel
        return resp

    ssh = SSHClient()
    ssh.load_system_host_keys()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(src, 22, user, passwd, timeout=3)

    pool = Pool(len(dst))
    result = pool.map(command, dst)
    pool.close()
    pool.join()
    ssh.close()  # close SSH client
    return result