mulithread或multiprocess中的python socket.connect超时错误

时间:2010-11-17 14:34:34

标签: python multithreading sockets timeout multiprocessing

如下所示,我想与特定IP范围内的许多PC进行通信。

My PC ---+------> Client A PC
         +------> Client B PC
         +------> Client C PC
         .................
         +------> Client Z PC

因为有太多客户端要沟通,我通过多线程尝试了。 socket.connect()不断产生超时错误。 如果我在单线程中尝试它,那就没问题了。

我用Google搜索并找到了以下内容:

Python Interpreter blocks Multithreaded DNS requests?

说在某些平台上,套接字模块可能是线程不安全的。

所以我将代码更改为多处理。但是它仍会产生相同的错误。

在以下代码示例中,test_single()完成正常。 test_mp()和test_mt()都会导致超时错误。

您是否曾经历过这种异常行为? 测试环境是Windows XP SP3,python 2.5.4。 还尝试了python 2.6.6和2.7.0,同样的错误。

import multiprocessing
import Queue
import socket
import threading

PROCESS_NUM = 5
PORT = 8888

def search_proc(ip):
    try:
        csock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        csock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        csock.settimeout(5.0)
        csock.connect((ip, PORT))
        csock.shutdown(socket.SHUT_RDWR)
        csock.close()
        return ip, "ok"
    except socket.error, msg:
        return ip, "fail", msg

def mp_connect(ip_range):
    pool = multiprocessing.Pool( PROCESS_NUM )
    for output in pool.imap_unordered(search_proc, ip_range):
        print output

def test_mp():
    ip_range = []
    for i in range(256):
        ip_range.append("192.168.123.%d"%(i,))

    mp_connect(ip_range)

def test_mt():
    def search_thread(ip_queue):
        while True:
            ip = ip_queue.get()
            print search_proc(ip)
            ip_queue.task_done()
    ip_queue = Queue.Queue()

    for i in range(256):
        ip_queue.put("192.168.123.%d"%(i,))

    for i in range(PROCESS_NUM):
        th = threading.Thread(target=search_thread, args=(ip_queue,))
        th.setDaemon(True)
        th.start()

    ip_queue.join()

def test_single():
    ip_range = []
    for i in range(256):
        print search_proc("192.168.123.%d"%(i,))

if __name__ == "__main__":
    multiprocessing.freeze_support()
    test_mp()
    #test_single()
    #test_mt()

1 个答案:

答案 0 :(得分:5)

David Beazley围绕Python GIL做了一些很好的研究,以及它如何影响IO和多线程。您可以找到有关其研究的信息herehere