我正在尝试实现一个超时,该超时将在定义的时间间隔内未收到任何连接时终止python脚本。到目前为止,我设法使用以下代码实现超时:
import sys
import socket
# Create a TCP/IP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Bind the socket to the port
server_address = ('192.168.43.112', 5001)
print >>sys.stderr, 'starting up on %s port %s' % server_address
sock.bind(server_address)
# Listen for incoming connections
sock.listen(1)
while True:
try:
# Wait for a connection
print >>sys.stderr, 'waiting for a connection'
connection, client_address = sock.accept()
try:
print >>sys.stderr, 'connection from', client_address
# Receive the data in small chunks and retransmit it
while True:
data = connection.recv(16)
print >>sys.stderr, 'received "%s"' % data
if data:
print >>sys.stderr, 'Do stuff here'
else:
print >>sys.stderr, 'no more data from', client_address
sock.settimeout(5)
break
finally:
# Clean up the connection
connection.close()
except socket.timeout:
break
在建立连接并结束非常相同的连接后,脚本会在5秒钟后终止,从某种意义上说,代码可以正常工作。但是,如果在超时窗口期间尝试建立另一个连接,则会出现以下错误:
starting up on 192.168.43.112 port 5001
waiting for a connection
connection from ('192.168.43.1', 47550)
received "Data 0
"
Do stuff here
received ""
no more data from ('192.168.43.1', 47550)
waiting for a connection
connection from ('192.168.43.1', 39010)
---------------------------------------------------------------------------
error Traceback (most recent call last)
/Users/location/Desktop/sandbox/data_fetcher.py in <module>()
24 # Receive the data in small chunks and retransmit it
25 while True:
---> 26 data = connection.recv(16)
27 print >>sys.stderr, 'received "%s"' % data
28 if data:
error: [Errno 35] Resource temporarily unavailable
我不完全确定您希望这一切如何工作,我感到有些奇怪,因为它现在以这种方式发生(我不希望超时会产生这种影响),但是基于EAGAIN错误(errno) 35),这是由于主套接字上的超时(只有在您建立第一个连接后才设置),也会导致第二个接受的套接字也处于非阻塞模式。这意味着当您打电话时connection.recv
没有立即有数据时,您会得到提示OSError
。
我怀疑某些操作系统之间可能有些不同,但是我能够在FreeBSD上重现此代码(您可能正在Linux上运行)。
围绕它进行的最小更改(我认为这不一定是编写此代码的最佳方法,但它确实起作用了)是将接受的套接字显式设置为阻塞:
# Wait for a connection
print >>sys.stderr, 'waiting for a connection'
connection, client_address = sock.accept()
connection.setblocking(1)
这样,代码的性能会更好(我添加了一个小型测试框架,该框架将您的代码作为一个单独的过程分离出来,然后以不同的延迟进行多个连接)。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句