def peekInt32(self):
assert self._buffer_size >= INT32_SIZE
need_chunk_index = 0
sum_chunk_size = 0
for chunk in self._buffer:
sum_chunk_size += len(chunk)
if sum_chunk_size >= INT32_SIZE:
break
need_chunk_index += 1
if need_chunk_index > 0:
_merge_prefix(self._buffer, sum_chunk_size)
return unpack('i', self._buffer[0][:INT32_SIZE])[0]
def retrieve(self, len): if len == 0: return b'' _merge_prefix(self._buffer, len) self._buffer_size -= len return self._buffer.popleft()
def consume_from_buffer(self, num):
_merge_prefix(self.handler.read_buf, num)
data = self.handler.read_buf.popleft()
return data
def consume_from_buffer(self, num):
_merge_prefix(self.handler.read_buf, num)
data = self.handler.read_buf.popleft()
return data
def _read_from_buffer(self):
if self._read_delimiter and self._read_delimiter_times:
_merge_prefix(self._read_buffer, sys.maxint)
chunks = self._read_buffer[0].split(self._read_delimiter, self._read_delimiter_times)[:-1]
if chunks:
chunks_str = self._read_delimiter.join(chunks)
callback = self._read_callback
delimiter_len = len(self._read_delimiter)
self._read_callback = None
self._read_delimiter = None
self._read_delimiter_times = None
self._run_callback(callback,self._consume(len(chunks_str)+delimiter_len))
return True
return False
elif self._read_multibulk_answers:
_merge_prefix(self._read_buffer, sys.maxint)
delimiter = '\r\n'
remained_answers = self._read_multibulk_answers * 2
buffer = self._read_buffer[0]
chunks = []
if not buffer:
return False
while True:
splitted_buffer = buffer.split(delimiter, 1)
if len(splitted_buffer) == 2:
chunk, rest_buffer = splitted_buffer[0], splitted_buffer[1]
if chunk in ('+OK', '$-1') or chunk[0] in (':', '-'):
remained_answers -= 2
elif chunk[0]=='*':
remained_answers += int(chunk[1])*2
remained_answers -= 2
else:
remained_answers -= 1
chunks.append(chunk)
buffer = rest_buffer
if remained_answers <=0:
chunks_str = delimiter.join(chunks)
callback = self._read_callback
self._read_multibulk_answers = None
self._read_callback = None
self._run_callback(callback, self._consume(len(chunks_str)+len(delimiter)))
return True
else:
return False
# multiple by two cause 1 default single answer assume to have two '\r\n'
# chunks = self._read_buffer[0].split(delimiter, self._read_multibulk_answers*2)[:-1]
# chunks_str = delimiter.join(chunks)
# if chunks:
# # but if answer is ($-1, +OK, *3, :1, -ERR) then we got one '\r\n'
# # so we must remove unnecessary chunks from next redis responses
# num_chunks = self._read_multibulk_answers*2
# for index, chunk in enumerate(chunks):
# if '$-1' in chunk or '+OK' in chunk or chunk[0] in (':', '-'):
# num_chunks -= 2
# elif chunk[0]=='*':
# num_chunks += int(chunk[1])*2
# num_chunks -= 2
# else:
# num_chunks -= 1
#
# if num_chunks<=0:
# break
#
# else:
# self._consume(len(chunks_str)+len(delimiter))
#
# chunks = chunks[:index]
# callback = self._read_callback
# self._read_multibulk_answers = None
# self._read_callback = None
# self._run_callback(callback, self._consume(len(chunks_str)+len(delimiter)))
# return True
# return False
else:
return super(BrukvaStream, self)._read_from_buffer()