fix: Fixed a bug where 'sender.new_buffer()' would return None.

Author: amunra

src/questdb/ingress.pyx

@@ -306,6 +306,8 @@ cdef class Buffer:
       * To see the contents, call ``str(buffer)``.
     """
     cdef line_sender_buffer* _impl
+    cdef size_t _init_capacity
+    cdef size_t _max_name_len
     cdef object _row_complete_sender
 
     def __cinit__(self, init_capacity: int=65536, max_name_len: int=127):
@@ -319,12 +321,33 @@ cdef class Buffer:
     cdef inline _cinit_impl(self, size_t init_capacity, size_t max_name_len):
         self._impl = line_sender_buffer_with_max_name_len(max_name_len)
         line_sender_buffer_reserve(self._impl, init_capacity)
+        self._init_capacity = init_capacity
+        self._max_name_len = max_name_len
         self._row_complete_sender = None
 
     def __dealloc__(self):
         self._row_complete_sender = None
         line_sender_buffer_free(self._impl)
 
+    @property
+    def init_capacity(self) -> int:
+        """
+        The initial capacity of the buffer when first created.
+
+        This may grow over time, see ``capacity()``.
+        """
+        return self._init_capacity
+
+    @property
+    def max_name_len(self) -> int:
+        """Maximum length of a table or column name."""
+        return self._max_name_len
+
+    @property
+    def max_name_len(self) -> int:
+        """Maximum length of a table or column name."""
+        return self._max_name_len
+
     def reserve(self, additional: int):
         """
         Ensure the buffer has at least `additional` bytes of future capacity.
@@ -770,8 +793,8 @@ cdef class Sender:
     cdef Buffer _buffer
     cdef bint _auto_flush_enabled
     cdef ssize_t _auto_flush_watermark
-    cdef object _init_capacity
-    cdef object _max_name_len
+    cdef size_t _init_capacity
+    cdef size_t _max_name_len
 
     def __cinit__(
             self,
@@ -896,10 +919,20 @@ cdef class Sender:
         The buffer is set up with the configured `init_capacity` and
         `max_name_len`.
         """
-        self._buffer = Buffer(
+        return Buffer(
             init_capacity=self._init_capacity,
             max_name_len=self._max_name_len)
 
+    @property
+    def init_capacity(self) -> int:
+        """The initial capacity of the sender's internal buffer."""
+        return self._init_capacity
+
+    @property
+    def max_name_len(self) -> int:
+        """Maximum length of a table or column name."""
+        return self._max_name_len
+
     def connect(self):
         cdef line_sender_error* err = NULL
         if self._opts == NULL:

test/test.py

@@ -254,6 +254,18 @@ def test_dont_flush_on_exception(self):
             msgs = server.recv()
             self.assertEqual(msgs, [])
 
+    def test_new_buffer(self):
+        sender = qi.Sender(
+            host='localhost',
+            port=9009,
+            init_capacity=1024,
+            max_name_len=10)
+        buffer = sender.new_buffer()
+        self.assertEqual(buffer.init_capacity, 1024)
+        self.assertEqual(buffer.max_name_len, 10)
+        self.assertEqual(buffer.init_capacity, sender.init_capacity)
+        self.assertEqual(buffer.max_name_len, sender.max_name_len)
+
 
 if __name__ == '__main__':
     unittest.main()