Nanosecond precision timestamps

dimaqq · dimaqq · commit c8e2ca43364a · 2016-09-09T14:52:22.000+02:00
diff --git a/fluent/event.py b/fluent/event.py
@@ -10,4 +10,6 @@ def __init__(self, label, data, **kwargs):
         assert isinstance(data, dict), 'data must be a dict'
         sender_ = kwargs.get('sender', sender.get_global_sender())
         timestamp = kwargs.get('time', int(time.time()))
+        if isinstance(timestamp, float):
+            timestamp = sender.EventTime(timestamp)
         sender_.emit_with_time(label, timestamp, data)
diff --git a/fluent/sender.py b/fluent/sender.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 from __future__ import print_function
+import struct
 import socket
 import threading
 import time
@@ -30,6 +31,14 @@ def get_global_sender():
 def close():
     get_global_sender().close()
 
+
+class EventTime(msgpack.ExtType):
+    def __new__(cls, timestamp):
+        seconds = int(timestamp)
+        nanoseconds = int(timestamp % 1 * 10 ** 9)
+        return super(EventTime, cls).__new__(cls, code=0, data=struct.pack(">II", seconds, nanoseconds))
+
+
 class FluentSender(object):
     def __init__(self,
                  tag,
@@ -61,7 +70,7 @@ def __init__(self,
             self._close()
 
     def emit(self, label, data):
-        cur_time = int(time.time())
+        cur_time = EventTime(time.time())
         return self.emit_with_time(label, cur_time, data)
 
     def emit_with_time(self, label, timestamp, data):
