Skip to content

Commit 90ad415

Browse files
committed
Add Write.buffer() reset method
The new method allows reusing the buffer instance to generate multiple messages in a loop.
1 parent 126bcd2 commit 90ad415

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

buffer.pxd

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ cdef class WriteBuffer:
3434
cdef inline _check_readonly(self)
3535
cdef inline _ensure_alloced(self, ssize_t extra_length)
3636
cdef _reallocate(self, ssize_t new_size)
37+
cdef inline reset(self)
3738
cdef inline start_message(self, char type)
3839
cdef inline end_message(self)
3940
cdef write_buffer(self, WriteBuffer buf)

buffer.pyx

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,10 @@ cdef class WriteBuffer:
113113
hton.pack_int32(&self._buf[1], <int32_t>mlen)
114114
return self
115115

116+
cdef inline reset(self):
117+
self._length = 0
118+
self._message_mode = 0
119+
116120
cdef write_buffer(self, WriteBuffer buf):
117121
self._check_readonly()
118122

0 commit comments

Comments
 (0)