最近在使用python進(jìn)行網(wǎng)絡(luò)編程開(kāi)發(fā)一個(gè)通用的tcpclient測(cè)試小工具。在使用socket進(jìn)行網(wǎng)絡(luò)編程中，如何判定對(duì)端發(fā)送一條報(bào)文是否接收完成，是進(jìn)行socket網(wǎng)絡(luò)開(kāi)發(fā)必須要考慮的一個(gè)問(wèn)題。這里，簡(jiǎn)要介紹一下判別數(shù)據(jù)接收接收完成常用的三種方法：

1.基礎(chǔ)數(shù)據(jù)接收法：

使用基礎(chǔ)數(shù)據(jù)接收法時(shí)，當(dāng)與服務(wù)socket斷開(kāi)連接時(shí)，會(huì)接收到空字符串。因此，可以根據(jù)此特點(diǎn)，在程序中加入循環(huán)，一直接收數(shù)據(jù)，直到數(shù)據(jù)發(fā)送端關(guān)閉socket連接。適用場(chǎng)景：客戶(hù)端和服務(wù)器的鏈接為短鏈接(即一次socket通訊后連接就會(huì)關(guān)閉)。代碼如下：

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.尾標(biāo)識(shí)方法

通過(guò)尋找接收的協(xié)議數(shù)據(jù)中的尾標(biāo)識(shí)字符串，獲取完整的數(shù)據(jù)的數(shù)據(jù)報(bào)文。適用場(chǎng)景：接收的協(xié)議數(shù)據(jù)中包含相關(guān)的尾標(biāo)識(shí)。代碼如下：

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.負(fù)載長(zhǎng)度方法

即通過(guò)協(xié)議數(shù)據(jù)中的負(fù)載長(zhǎng)度值，確定有效報(bào)文的長(zhǎng)度，適用場(chǎng)景：協(xié)議數(shù)據(jù)中包含負(fù)載協(xié)議字段。這種方法也是比較常用和通用的一種方法，但是需要一邊接收數(shù)據(jù)一邊解析數(shù)據(jù)。代碼如下：

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)

以上內(nèi)容為大家介紹了Python網(wǎng)絡(luò)編程調(diào)用接收數(shù)據(jù)的三種方法，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://www.parentadvocate.org/