Python使用Semaphore实现多进程并发SSH登录

Python多进程并发SSH登录与Semaphore控制

利用Semaphore 可以限制对共享资源的访问数量，从而精确控制同一时刻并发的进程数。

#/usr/bin/python

# _*_ coding: utf-8 _*_

import multiprocessing

import time

import paramiko

def ssh(s,i,host):

代码逻辑包含以下核心部分：

s.acquire()

print(time.strftime('%H:%M:%S'),multiprocessing.current_process().name + " 获得锁运行");

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

ssh.connect(hostname=host, port=22, username="root", password="yankefei")

print (host+" is login success")

stdin, stdout, stderr = ssh.exec_command("echo 
d
a
t
e
 && df -hl")

print(stdout.read().decode('utf-8'))

returncode = stdout.channel.recv_exit_status()

print("returncode:",returncode)

异常处理块：

ssh.close()

# time.sleep(i)

print(time.strftime('%H:%M:%S'),multiprocessing.current_process().name + " 释放锁结束");

s.release()

print (host+" is unreachable")

资源清理块：

ssh.close()

s.release()

if __name__ == "__main__":

s = multiprocessing.Semaphore(200) #同时并发200个进程

for n in range(111):

p = multiprocessing.Process(target = ssh, args=(s,2,"192.168.0."+str(n)))

p.start()

具体运行效果如下图所示：

关于Python多进程登录远端服务器的介绍到此结束，更多相关多进程Python内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站！