Merge pull request #4 from zhangxiaoyang/master

fix recv function
change recv buffer
This commit is contained in:
Bookug Lobert 2016-05-06 21:30:54 +08:00
commit 28ae308d72
1 changed files with 4 additions and 4 deletions

View File

@ -35,15 +35,15 @@ class GstoreConnector:
head = self._sock.recv(4)
context_len = 0
for i in range(4):
context_len |= (ord(head[i]) << i * 8)
context_len |= (ord(head[i]) & 0xFF) << i * 8
data = bytearray(context_len)
data = bytearray()
recv_len = 0
while recv_len < context_len:
chunk = self._sock.recv(1024)
chunk = self._sock.recv(context_len - recv_len)
data.extend(chunk)
recv_len += len(chunk)
return str(data)
return data.rstrip('\x00').decode('utf-8')
def _pack(self, msg):
data_context = bytearray()