【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)这行代码中。