在Python中创建非阻塞式套接字

Question

我在尝试理解非阻塞套接字的工作原理，于是我用Python编写了这个简单的服务器。

import socket

s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)

s.bind(('127.0.0.1',1000))
s.listen(5)
s.setblocking(0)
while True:
      try:
            conn, addr = s.accept()
            print ('connection from',addr)
            data=conn.recv(100)
            print ('recived: ',data,len(data))

      except:
          pass  

然后我尝试从多个客户端实例连接到此服务器

import socket

s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1',1000))
while True:
    continue

但是无论将阻塞设置为0或1似乎都没有效果，服务器的recv方法总是会阻塞执行。因此，在Python中创建非阻塞套接字是否需要更多操作而不仅仅是将阻塞标志设置为0。

- Gray

Answer 1

setblocking 只影响你在其上使用的套接字。因此，您必须添加 conn.setblocking(0) 才能看到效果：如果没有可用数据，则 recv 立即返回。

- Phillip

---

setblocking(0)会影响socket.accept()，它将不会阻塞等待传入的连接，而是会引发异常。 - mhawke

文档在这里似乎不完整，你当然是对的。我已经从我的答案中删除了那个评论。但是有一个例外：如果使用了 socket.settimeout，Python 仍然会在 accept(2) 之前调用 select(2)，即至少等待超时时间到达。 - Phillip

Answer 2

您只需要在已连接的套接字上调用setblocking(0)，即conn。

import socket

s = socket.socket()
s.bind(('127.0.0.1', 12345))
s.listen(5)
s.setblocking(0)
>>> conn, addr = s.accept()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib64/python2.7/socket.py", line 202, in accept
    sock, addr = self._sock.accept()
socket.error: [Errno 11] Resource temporarily unavailable

# start your client...
>>> conn, addr = s.accept()
>>> conn.recv()            # this will hang until the client sends some data....
'hi there\n'
>>> conn.setblocking(0)    # set non-blocking on the connected socket "conn"
>>> conn.recv()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
socket.error: [Errno 11] Resource temporarily unavailable

- mhawke

Answer 3

https://docs.python.org/3/library/socket.html#socket.setdefaulttimeout

您可以使用s.setdefaulttimeout(1.0)将所有连接套接字应用为默认值。

- Ömer Feyyaz Selçuk