如何设置 python 的套接字恢复方法的超时？

我需要在 python 的套接字恢复方法上设置超时？

322325 次浏览
小开

这是 socket.settimeout()

小开

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

import select


mysocket.setblocking(0)


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

如果您有许多打开的文件描述符，那么 民意调查()是比 select()更有效的替代品。

另一种选择是使用 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)
小开

您可以在收到响应之前设置超时，并在收到响应之后将其设置为无:

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


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

尝试这个它使用底层的 C。

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

如果您实现了服务器端，那么您要查找的超时是连接套接字的超时，而不是主套接字的超时。换句话说，连接套接字对象还有另一个超时，即 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!\r\n")


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秒

小开 
#! /usr/bin/python3.6


# -*- coding: utf-8 -*-
import socket
import time
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
s.settimeout(5)
PORT = 10801


s.bind(('', PORT))
print('Listening for broadcast at ', s.getsockname())
BUFFER_SIZE = 4096
while True:
try:
data, address = s.recvfrom(BUFFER_SIZE)
except socket.timeout:
print("Didn't receive data! [Timeout 5s]")
continue
小开

呼叫: https://boltons.readthedocs.io/en/latest/socketutils.html

它提供了一个缓冲套接字，这提供了很多非常有用的功能，例如:

.recv_until()    #recv until occurrence of bytes
.recv_closed()   #recv until close
.peek()          #peek at buffer but don't pop values
.settimeout()    #configure timeout (including recv timeout)
小开

从顶部答案有点困惑，所以 我写了一个小要点与例子，以便更好地理解。

选项 # 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

提交答案