之前我使用阻塞模式接收数据时必须使用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)这行代码中。