Python socket.accept nonblocking?

前端 未结 2 1476
盖世英雄少女心
盖世英雄少女心 2020-12-01 05:38

Is there a way I can use python\'s socket.accept() in a non-blocking way that simply runs it and lets me just check if it got any new connections? I really don\'t w

相关标签:
2条回答
  • 2020-12-01 06:18

    You can invoke the setblocking(0) method on the Socket to make it non-blocking. Look into the asyncore module or a framework like Twisted.

    0 讨论(0)
  • 2020-12-01 06:19

    You probably want something like select.select() (see documentation). You supply select() with three lists of sockets: sockets you want to monitor for readability, writability, and error states. The server socket will be readable when a new client is waiting.

    The select() function will block until one of the socket states has changed. You can specify an optional fourth parameter, timeout, if you don't want to block forever.

    Here is a dumb echo server example:

    import select
    import socket
    
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    server_socket.bind(('', 8888))
    server_socket.listen(5)
    print "Listening on port 8888"
    
    read_list = [server_socket]
    while True:
        readable, writable, errored = select.select(read_list, [], [])
        for s in readable:
            if s is server_socket:
                client_socket, address = server_socket.accept()
                read_list.append(client_socket)
                print "Connection from", address
            else:
                data = s.recv(1024)
                if data:
                    s.send(data)
                else:
                    s.close()
                    read_list.remove(s)
    

    Python also has epoll, poll, and kqueue implementations for platforms that support them. They are more efficient versions of select.

    0 讨论(0)
提交回复
热议问题