Tạo nhiều kết nối SSH tại một thời gian sử dụng paramiko

Question

Đoạn code dưới đây chạy grep trong cùng một máy thông qua SSH và in kết quả:

import sys, os, string 
import paramiko 

cmd = "grep -h 'king' /opt/data/horror_20100810*" 

ssh = paramiko.SSHClient() 
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
ssh.connect('10.10.3.10', username='xy', password='xy') 
stdin, stdout, stderr = ssh.exec_command(cmd) 
stdin.write('xy\n') 
stdin.flush() 

print stdout.readlines() 

Làm thế nào tôi có thể grep lăm máy cùng một lúc (vì vậy mà tôi không có sự chậm trễ lớn), hơn là đặt tất cả trong năm biến và in tất cả chúng ra.

Answer 1

Bạn sẽ cần phải đặt các cuộc gọi thành các chuỗi riêng biệt (hoặc các quá trình, nhưng điều đó sẽ quá mức cần thiết), do đó yêu cầu mã phải ở trong một hàm (đó là một ý tưởng hay: không có mã đáng kể ở cấp cao nhất của mô-đun).

Ví dụ:

import sys, os, string, threading 
import paramiko 

cmd = "grep -h 'king' /opt/data/horror_20100810*" 

outlock = threading.Lock() 

def workon(host): 

    ssh = paramiko.SSHClient() 
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
    ssh.connect(host, username='xy', password='xy') 
    stdin, stdout, stderr = ssh.exec_command(cmd) 
    stdin.write('xy\n') 
    stdin.flush() 

    with outlock: 
     print stdout.readlines() 

def main(): 
    hosts = ['10.10.3.10', '10.10.4.12', '10.10.2.15', ] # etc 
    threads = [] 
    for h in hosts: 
     t = threading.Thread(target=workon, args=(h,)) 
     t.start() 
     threads.append(t) 
    for t in threads: 
     t.join() 

main() 

Nếu bạn có nhiều hơn năm host, tôi sẽ khuyên bạn sử dụng thay vì một "hồ bơi thread" kiến ​​trúc và một danh sách các đơn vị làm việc. Nhưng, chỉ trong năm, nó đơn giản hơn để dính vào mô hình "chuyên dụng thread" (đặc biệt là kể từ khi không có hồ bơi thread trong thư viện chuẩn, vì vậy bạn sẽ cần một gói bên thứ ba như threadpool ... hoặc rất nhiều tùy chỉnh tinh tế mã của riêng bạn tất nhiên ;-).

Answer 2

Chỉ cần chạy mọi thứ trong vòng lặp for và đừng quên đóng stdin trước khi chuyển sang lần lặp tiếp theo. Tức là, sau dòng stdin.flush(), hãy thêm dòng stdin.close()