如何在python的socket recv方法上设置超时？

Question

我需要在python的socket recv方法上设置超时.怎么做？

Answer 1

典型的方法是使用select()等待数据可用或直到超时发生.仅recv()在数据实际可用时调用.为了安全起见,我们还将套接字设置为非阻塞模式,以确保recv()永远不会无限期地阻塞. select()也可以用来一次等待多个套接字.

import select

mysocket.setblocking(0)

ready = select.select([mysocket], [], [], timeout_in_seconds)
if ready[0]:
    data = mysocket.recv(4096)

如果你有很多打开文件描述符,poll()是一个更有效的替代方法select().

另一种选择是使用套接字上的所有操作设置超时socket.settimeout(),但我看到你在另一个答案中明确拒绝了该解决方案.

Answer 2

有 socket.settimeout()

Answer 3

如前所述都select.select()和socket.settimeout()正常工作.

请注意,您可能需要拨打settimeout两次以满足您的需求,例如

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(("",0))
sock.listen(1)
# accept can throw socket.timeout
sock.settimeout(5.0)
conn, addr = sock.accept()

# recv can throw socket.timeout
conn.settimeout(5.0)
conn.recv(1024)

Answer 4

您可以在收到响应之前设置超时,并在收到响应后将其设置回None:

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

sock.settimeout(5.0)
data = sock.recv(1024)
sock.settimeout(None)

Answer 5

如果您实现服务器端，您正在寻找的超时是连接套接字的超时而不是主套接字的超时。换句话说，连接套接字对象还有另一个超时，它是socket.accept()方法的输出。所以：

sock.listen(1)
connection, client_address = sock.accept()
connection.settimeout(5)    # This is the one that affects recv() method.
connection.gettimeout()     # This should result 5
sock.gettimeout()           # This outputs None when not set previously, if I remember correctly.

如果您实现客户端，那就很简单了。

sock.connect(server_address)
sock.settimeout(3)

Answer 6

试试这个它使用底层C。

timeval = struct.pack('ll', 2, 100)
s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, timeval)

Answer 7

您可以使用socket.settimeout()which 接受表示秒数的整数参数。例如，socket.settimeout(1)将超时设置为 1 秒

Answer 8

对顶级答案有些困惑，所以我写了一个小要点，并附有示例以便更好地理解。

选项1 - socket.settimeout()

如果sock.recv()等待时间超过定义的超时，则会引发异常。

import socket

sock = socket.create_connection(('neverssl.com', 80))
timeout_seconds = 2
sock.settimeout(timeout_seconds)
sock.send(b'GET / HTTP/1.1\r\nHost: neverssl.com\r\n\r\n')
data = sock.recv(4096)
data = sock.recv(4096) # <- will raise a socket.timeout exception here

选项#2 - select.select()

等待数据发送，直到超时。我已经调整了丹尼尔的答案，所以它会引发一个例外

import select
import socket

def recv_timeout(sock, bytes_to_read, timeout_seconds):
    sock.setblocking(0)
    ready = select.select([sock], [], [], timeout_seconds)
    if ready[0]:
        return sock.recv(bytes_to_read)

    raise socket.timeout()

sock = socket.create_connection(('neverssl.com', 80))
timeout_seconds = 2
sock.send(b'GET / HTTP/1.1\r\nHost: neverssl.com\r\n\r\n')
data = recv_timeout(sock, 4096, timeout_seconds)
data = recv_timeout(sock, 4096, timeout_seconds) # <- will raise a socket.timeout exception here