Output traceback to log if msgpack raise exception

Yuki Nishiwaki · Yuki Nishiwaki · commit eed914a56dde · 2016-02-08T10:05:54.000+09:00
This patch change so that it can log traceback without abnormal termination, if msgpack raise exception. Fixed: fluent#49
diff --git a/fluent/sender.py b/fluent/sender.py
@@ -4,6 +4,7 @@
 import socket
 import threading
 import time
+import traceback
 
 import msgpack
 
@@ -54,7 +55,13 @@ def emit(self, label, data):
         self.emit_with_time(label, cur_time, data)
 
     def emit_with_time(self, label, timestamp, data):
-        bytes_ = self._make_packet(label, timestamp, data)
+        try:
+            bytes_ = self._make_packet(label, timestamp, data)
+        except Exception:
+            bytes_ = self._make_packet(label, timestamp,
+                                       {"level": "CRITICAL",
+                                        "message": "Can't output to log",
+                                        "traceback": traceback.format_exc()})
         self._send(bytes_)
 
     def _make_packet(self, label, timestamp, data):
