Python:非阻塞套接字或异步I/O
我刚开始学习Python,现在需要写一个Python脚本,通过TCP/IP和一个设备(气象站)进行通信。
这个设备充当服务器,负责监听IP和端口,接受连接,接收请求,并传输数据。
我只需要发送一条消息,接收回复,然后优雅地关闭连接。
try:
comSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
except socket.error, msg:
sys.stderr.write("[ERROR] %s\n" % msg[1])
sys.exit(1)
try:
comSocket.connect((''))
except socket.error, msg:
sys.stderr.write("[ERROR] %s\n" % msg[1])
sys.exit(2)
comSocket.send('\r')
comSocket.recv(128)
comSocket.send('\r')
comSocket.recv(128)
comSocket.send('\r\r')
comSocket.recv(128)
comSocket.send('1I\r\r3I\r\r4I\r\r13I\r\r5I\r\r8I\r\r7I\r\r9I\r\r')
rawData = comSocket.recv(512)
comSocket.shutdown(1)
comSocket.close()
我遇到的问题是:
通信通道不太可靠,设备反应比较慢。所以,有时候设备回复的消息长度为0(只是一个确认信号),这时我的代码就会卡住,永远在等待回复。
这段代码涉及到SOCKET的部分,整个代码会在CRON下运行,所以卡住不是我想要的结果。
我的问题是:
在Python中,有什么好的方法可以处理这种情况,让代码不会卡住一直等待,而是能尝试继续发送下一条消息(或者其他操作)呢?
3 个回答
1
在接收数据之前,试着给这个网络连接设置一个超时时间:
comSocket.settimeout(5.0)
try:
rawData = comSocket.recv(512)
except socket.timeout:
print "No response from server"
3
11
你可以尝试使用超时的方法,比如Russel的代码,或者使用非阻塞的套接字,下面的代码就展示了这一点。这样在执行socket.recv
时,它不会让程序停下来,你可以在一个循环里多次尝试。这样的话,你的程序就不会因为超时而卡住。通过这种方式,你可以检查数据是否可用,如果没有数据,你可以做其他事情,稍后再试。
socket.setblocking(0)
while (retry_condition):
try:
data = socket.recv(512)
except socket.error:
'''no data yet..'''