Inject datadog traces

Author: davidgasquez

bufflog/__init__.py

@@ -8,6 +8,24 @@
 from structlog.stdlib import filter_by_level
 from structlog.stdlib import add_log_level_number
 
+from ddtrace.helpers import get_correlation_ids
+
+
+def tracer_injection(logger, log_method, event_dict):
+    # get correlation ids from current tracer context
+    trace_id, span_id = get_correlation_ids()
+
+    # add ids to structlog event dictionary
+    event_dict["dd.trace_id"] = trace_id or 0
+    event_dict["dd.span_id"] = span_id or 0
+
+    # add the env, service, and version configured for the tracer
+    event_dict["dd.env"] = ddtrace.config.env or ""
+    event_dict["dd.service"] = ddtrace.config.service or ""
+    event_dict["dd.version"] = ddtrace.config.version or ""
+
+    return event_dict
+
 
 def rename_message_key(_, __, event_dict):
     event_dict["message"] = event_dict["event"]
@@ -27,6 +45,7 @@ def increase_level_numbers(_, __, event_dict):
 bufflog = wrap_logger(
     logging.getLogger(__name__),
     processors=[
+        tracer_injection,
         filter_by_level,
         rename_message_key,
         add_log_level_number,

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="python-bufflog",
-    version="0.0.2",
+    version="0.0.3",
     author="David Gasquez",
     author_email="[email protected]",
     description="Python logger for Buffer services",