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

Question

前面的详细信息
我需要在交换机上使用ssh来ping不同的主机。早些时候，我为每个主机启动了一个线程，但结果很容易超过了最大ssh连接数，所以我根据这个创建了一个交互式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