一个完整的socket recv()案例,包括解决粘包、服务器主动推数据的问题
前言:
本文是针对socket长连接(涉及到服务器主动推数据),每个包头的拼接算法和长度都不一样,具体的包头小伙伴们问自己公司的开发吧,本文只是提供思路。再啰嗦一句:recv到的包头中数字进行某种运算后得到的就是开发定义的返回code。
SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。粘包问题解决思路:
python中的socket recv()是阻塞接收的,所以不存在同时发送多个数据接收错乱的情况。网上对于粘包有各种各样的解决办法,什么每次recv前sleep几秒,保证缓存中有足够数据在接收。还有说在recv()中加个wait的参数,保证每次都接受满1024个数据长度。我觉得网上唯一正确的办法是:先知道自己想要接收的数据的大小,然后用一个死循环接收完所有数据。
作者一开始是使用静态拆包的方式,就是每次send后都recv一次,然后把每次recv的结果都拼在一起,然后再根据包头中的包体长度拆出一个个完整包。这种方式有个缺点:如果后面的请求需要用到前面请求返回值做参数,那么就凉凉了。
这时候就需要用到动态拆包,send一个请求recv一个返回结果,然后解析出完整数据。但是实现的过程中发现几个问题:
(1)返回数据的长度跟我从包头里得到的长度不一致,我以为是粘包导致的接收不全,因为我后来加了个一样的请求,就能接收全了。
(2)后来我在每个请求发送前都加了sleep(10),第16个请求接受全了,但是后面的数据直接报错了。
(3)后来偶然一次,我发现能运行成功不报错,但是在大部分时运行会报错。
(4)然后发现send一个请求,返回了两个完整的包。
结合上面的四个问题,我开始了漫长的寻求解答的过程,后来得高人解答,才发现了问题——服务器主动推了数据。现在想想上面4点可不就是服务器主动推数据的表现嘛。
下面直接上方法,同时解决粘包、服务器主动推数据的问题。
思路:send之后,先recv(8)包头,得到包体大小,然后用死循环recv全。然后要用包头算法对得到的返回数据进行判断,进一步保证了这个包就是我需要的数据包。
def sendData(self, data, respHeader): self.data = data self.t = t self.respHeader=respHeader self.s.send(data) # 传输包装好的请求 total_data = bytes() header = 1 for i in range(5): #每次send,我规定它最多去recv5次,以防浪费时间 print("i:",i) header_bytes = self.s.recv(8) # 先接收包头 body_len = struct.unpack("!2I", header_bytes)[0] #这是包体长度 header = struct.unpack("!2I", header_bytes)[1] #开发定义的用来做运算 print("应该接收的长度:", body_len) while len(total_data) < body_len: total_data += self.s.recv(body_len) print("实际接收的长度:", len(total_data)) print("") header_code = header & 0x3xxxxxxx #做运算得到返回请求的code print("header_code: " + str(header_code)) #如果算出来等于这个返回请求的头或者返回error,跳出for循环,不再继续recv if header_code == int(respHeader) or header_code == 1: print("over") break else: #如果接收的不是我需要的包头,清空total_data,重新for循环从缓存中recv数据 total_data = bytes() if i >= 5: print("返回数据有误")
