最近在使用python进行网络编程开发一个通用的tcpclient测试小工具。在使用socket进行网络编程中，如何判定对端发送一条报文是否接收完成，是进行socket网络开发必须要考虑的一个问题。这里，简要介绍一下判别数据接收接收完成常用的三种方法：

1.基础数据接收法：

使用基础数据接收法时，当与服务socket断开连接时，会接收到空字符串。因此，可以根据此特点，在程序中加入循环，一直接收数据，直到数据发送端关闭socket连接。适用场景：客户端和服务器的链接为短链接(即一次socket通讯后连接就会关闭)。代码如下：

importsocket,struct,sys,time

Port=22220

#assumeasocketdisconnect(datareturnedisemptystring)meansalldatawas#donebeingsent.

defrecv_basic(the_socket):

total_data=[]

whileTrue:

data=the_socket.recv(20480)

ifnotdata:break

total_data.append(data)

return''.join(total_data)

2.尾标识方法

通过寻找接收的协议数据中的尾标识字符串，获取完整的数据的数据报文。适用场景：接收的协议数据中包含相关的尾标识。代码如下：

End='somethinguseableasanendmarker'

defrecv_end(the_socket):

total_data=[];data=''

whileTrue:

data=the_socket.recv(8192)

ifEndindata:

total_data.append(data[:data.find(End)])

break

total_data.append(data)

iflen(total_data)>1:

#checkifend_of_datawassplit

last_pair=total_data[-2]+total_data[-1]

ifEndinlast_pair:

total_data[-2]=last_pair[:last_pair.find(End)]

total_data.pop()

break

return''.join(total_data)

3.负载长度方法

即通过协议数据中的负载长度值，确定有效报文的长度，适用场景：协议数据中包含负载协议字段。这种方法也是比较常用和通用的一种方法，但是需要一边接收数据一边解析数据。代码如下：

defrecv_size(the_socket):

#datalengthispackedinto4bytes

total_len=0;total_data=[];size=sys.maxint

size_data=sock_data='';recv_size=8192

whiletotal_len

sock_data=the_socket.recv(recv_size)

ifnottotal_data:

iflen(sock_data)>4:

size_data+=sock_data

size=struct.unpack('>i',size_data[:4])[0]

recv_size=size

ifrecv_size>524288:recv_size=524288

total_data.append(size_data[4:])

else:

size_data+=sock_data

else:

total_data.append(sock_data)

total_len=sum([len(i)foriintotal_data])

return''.join(total_data)

以上内容为大家介绍了Python网络编程调用接收数据的三种方法，希望对大家有所帮助，如果想要了解更多Python相关知识，请关注IT培训机构:千锋教育。http://www.mobiletrain.org/

上一篇

Python的技术攻略

下一篇

如何让Python代码更易维护