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

Posted 2021-04-10

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

答案

典型的方法是使用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()为套接字上的所有操作设置超时，但我看到你在另一个答案中明确拒绝了该解决方案。

另一答案

有socket.settimeout()

另一答案

如上所述，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)

另一答案

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

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

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

另一答案

您正在寻找的超时是连接套接字的超时而不是主套接字，如果您实现服务器端。换句话说，连接套接字对象有另一个超时，它是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)

另一答案

如前面的回复中所述，您可以使用以下内容：.settimeout()例如：

import socket

s = socket.socket()

s.settimeout(1) # Sets the socket to timeout after 1 second of no activity

host, port = "somehost", 4444
s.connect((host, port))

s.send("Hello World!
")

try:
    rec = s.recv(100) # try to receive 100 bytes
except socket.timeout: # fail after 1 second of no activity
    print("Didn't receive data! [Timeout]")
finally:
    s.close()

我希望这有帮助！！

另一答案

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

另一答案

试试这个它使用底层C.

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