Python教程

【Socket】Python使用非阻塞模式接收完整数据

本文主要是介绍【Socket】Python使用非阻塞模式接收完整数据,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

前言

之前我使用阻塞模式接收数据时必须使用Socket的.settimeout方法设置一个超时时间,用来判断连接对象是否还是数据传入。 但是这种方式有个很明显的弊端,那就是会影响整个的数据传输的效率。 因为需要判断数据传输是否超时来得到数据是否传输完毕,但有时会因为网络波动导致每次接收的数据量都不一样,也就会导致数据完整性无法得到验证。 影响数据接收的效率是因为需要等待.settimeout方法设置的时间来判断是否超时,而短了通常会提前断开,长了又过于耗时。

import socket
s = socket.socket()
s.bind(("0.0.0.0", 8888))
s.listen(3)
s,addr = s.accept()

res = data = b""
try:
	s.settimeout(1)
	while True:
		data = s.recv(4096)
		if not data:
			break
		res += data
except socket.timeout:
	pass

print(res)
s.close()

解决方法

使用Socket中的.setblocking方法,以下是代码

import socket
s = socket.socket()
s.bind(("0.0.0.0", 8888))
s.listen(3)
s,addr = s.accept()

# 需要先接收一次数据
res = s.recv(4096)

# 设置阻塞模式, 0 为非阻塞,1 为阻塞
s.setblocking(0)
while True:
	try:
		res += s.recv(4096)
	except BlockingIOError:
		break

# 将模式重新设置为阻塞模式
s.setblocking(1)
s.close()
print(res)

这样便可以完整接收由对向连接发送过来的数据。

缺点

如果对向连接的Socket端未选择接收由当前Socket端发送回去的数据,则有可能会一直停留在res += s.recv(4096)这行代码中。

这篇关于【Socket】Python使用非阻塞模式接收完整数据的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!