Configure FluentSender with msgpack.packb() kwargs

Josh Wilson · Josh Wilson · commit 322a89e6d2fe · 2017-08-07T20:59:47.000-07:00
This adds an attribute to the `FluentSender` class to enable
specialization of the internal call to `msgpack.packb()`. For
instance, one can use this provide the `default` parameter to allow for
serialization of non-standard types (like `DateTime`).
diff --git a/fluent/sender.py b/fluent/sender.py
@@ -53,6 +53,7 @@ def __init__(self,
                  verbose=False,
                  buffer_overflow_handler=None,
                  nanosecond_precision=False,
+                 msgpack_kwargs=None,
                  **kwargs): # This kwargs argument is not used in __init__. This will be removed in the next major version.
 
         self.tag = tag
@@ -63,6 +64,7 @@ def __init__(self,
         self.verbose = verbose
         self.buffer_overflow_handler = buffer_overflow_handler
         self.nanosecond_precision = nanosecond_precision
+        self.msgpack_kwargs = {} if msgpack_kwargs is None else msgpack_kwargs
 
         self.socket = None
         self.pendings = None
@@ -109,7 +111,6 @@ def close(self):
         finally:
             self.lock.release()
 
-
     def _make_packet(self, label, timestamp, data):
         if label:
             tag = '.'.join((self.tag, label))
@@ -118,7 +119,7 @@ def _make_packet(self, label, timestamp, data):
         packet = (tag, timestamp, data)
         if self.verbose:
             print(packet)
-        return msgpack.packb(packet)
+        return msgpack.packb(packet, **self.msgpack_kwargs)
 
     def _send(self, bytes_):
         self.lock.acquire()
