From 85ca59411868ab84d874ad28e56213f5374e5ca8 Mon Sep 17 00:00:00 2001
From: Naoki INADA <inada-n@gear>
Date: Tue, 9 Jun 2009 13:12:29 +0900
Subject: free buffer when packer deleted.

---
 python/msgpack.pyx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'python/msgpack.pyx')

diff --git a/python/msgpack.pyx b/python/msgpack.pyx
index 8b2006a..f24f403 100644
--- a/python/msgpack.pyx
+++ b/python/msgpack.pyx
@@ -59,6 +59,8 @@ cdef class Packer:
 
         msgpack_packer_init(&self.pk, <void*>self, <msgpack_packer_write>_packer_write)
 
+    def __del__(self):
+        free(self.buff);
 
     def flush(self):
         """Flash local buffer and output stream if it has 'flush()' method."""
@@ -98,7 +100,7 @@ cdef class Packer:
         """
         msgpack_pack_map(&self.pk, len)
 
-    def pack(self, object o):
+    cdef __pack(self, object o):
         cdef long long intval
         cdef double fval
         cdef char* rawval 
@@ -140,6 +142,11 @@ cdef class Packer:
             # TODO: Serialize with defalt() like simplejson.
             raise TypeError, "can't serialize %r" % (o,)
 
+    def pack(self, obj, flush=True):
+        self.__pack(obj)
+        if flush:
+            self.flush()
+
 cdef int _packer_write(Packer packer, const_char_ptr b, unsigned int l):
     if packer.length + l > packer.allocated:
         if packer.length > 0:
-- 
cgit v1.2.1