ping ip多进程处理小程序

时间:2023-03-09 21:17:59
ping ip多进程处理小程序

最近,环境维护需要经常需要判断某些服务器上的IP是否可达,由于服务器数量较多,逐一手工ping检查太过繁琐。写个小程序使用。

实现和说明

1、使用配置文件ip.txt实现可配置指定ip列表。

2、利用multiprocessing多进程模块实现并发处理。同时利用Pool类实现进程数量控制。

3、通过ping_result.append()先收集ping执行结果,再打印的方式避免出现输出时错行的现象。

具体代码实现如下:

import multiprocessing
import os, sys ip_file = "ip.txt"
cur_dir = os.getcwd() def ping_ip(ip):
ret = os.system('ping -n 2 -w 3 %s > nul' % (ip,))
return (ret, ip) if __name__== '__main__':
try:
fp = open(ip_file, 'r')
except:
print "%s have not %s file,please check." % (cur_dir, ip_file)
sys.exit(1)
else:
ip_lst = fp.readlines()
fp.close()
p = multiprocessing.Pool(multiprocessing.cpu_count()*2)
ping_result = []
for ip in ip_lst:
ping_result.append(p.apply_async(ping_ip, args=(ip.strip(),)))
p.close()
p.join() for each in ping_result:
ret, ip = each.get()
if ret == 0:
print "%s success" % (ip,)
else:
print "%s fail" % (ip,)

关键代码描述:

1、multiprocessing.Pool()

函数原型:Pool(processes=None, initializer=None, initargs=(), maxtasksperchild=None)。返回进程池对象(Returns a process pool object)

Pool 可以提供指定数量的进程供用户使用,默认是 CPU 核数。当有新的请求提交到 Poll 的时候,如果池子没有满,会创建一个进程来执行,否则就会让该请求等待。特别说明:

(1)Pool 对象调用 join 方法会等待所有的子进程执行完毕。

(2)调用 join 方法之前,必须调用 close 方法

(3)调用 close 之后就不能继续添加新的 Process 了。

2、multiprocessing.cpu_count()。Returns the number of CPUs in the system

3、apply_async()方法用来同步执行进程,允许多个进程同时进入池子。

4、multiprocessing模块更详细的使用方法可以查看官方说明文档:multiprocessing