fallback: _fb_read: add fast-path

msgpack · Jan 29, 2013 · 4cde7f080c16fd396d758930faf9ebf15b138047 · 4cde7f0
1 parent d91a0d3
commit 4cde7f080c16fd396d758930faf9ebf15b138047
Unified Split

Showing with 5 additions and 0 deletions.

+5 −0 msgpack/fallback.py
diff --git a/msgpack/fallback.py b/msgpack/fallback.py
@@ -222,6 +222,11 @@ def _fb_get_extradata(self):
        return ''.join(bufs)

    def _fb_read(self, n, write_bytes=None):
+        if (write_bytes is None and self._fb_buf_i < len(self._fb_buffers)
+                and self._fb_buf_o + n < len(self._fb_buffers[self._fb_buf_i])):
+            self._fb_buf_o += n
+            return self._fb_buffers[self._fb_buf_i][
+                    self._fb_buf_o-n:self._fb_buf_o]
        ret = ''
        while len(ret) != n:
            if self._fb_buf_i == len(self._fb_buffers):