首先,发送所有字节不需要循环,因为python套接字提供了一种简单的方法:socket.sendall()
。
现在对您的问题:
是的,即使只接收4个字节,您也应该有一个recv()
在套接字上调用的接收循环,直到读取4个字节为止。
如果 可以保证这些字符不会出现在消息本身中, 则 可以。但是,您仍然需要搜索读入的每个字符以寻找魔术分隔符,因此仅给消息正文加上一个长度就显得不那么理想了。
当您调用时recv(n)
,只能保证 返回n个字节,而不完全是n个字节。
以下是三种不同的recvall()
比较方法:
def recvall(sock, size):
received_chunks = []
buf_size = 4096
remaining = size
while remaining > 0:
received = sock.recv(min(remaining, buf_size))
if not received:
raise Exception('unexpected EOF')
received_chunks.append(received)
remaining -= len(received)
return b''.join(received_chunks)
而且更短
def recvall2(sock, size):
return sock.recv(size, socket.MSG_WAITALL)
最后是另一个版本,该版本比第一个版本短一些,但缺少几个功能:
def recvall3(sock, size):
result = b''
remaining = size
while remaining > 0:
data = sock.recv(remaining)
result += data
remaining -= len(data)
return result
第二个不错,很简短,但是它依赖于套接字选项socket.MSG_WAITALL
,我认为不能保证每个平台上都存在。第一个和第三个应该在任何地方工作。我还没有真正进行基准比较和对比。