Fix bug with how parentID is set

DarcyRaynerDD · DarcyRaynerDD · commit 9a89c272989c · 2019-08-22T11:18:47.000-04:00
diff --git a/datadog_lambda/trace_intake.py b/datadog_lambda/trace_intake.py
@@ -46,12 +46,18 @@ def convert_trace_to_protobuf_payload(trace):
         else:
             span_groups[trace_id] = span_group
 
+        parent_id = 0
+
+        if "parent_id" in span and span["parent_id"]:
+            parent_id = int(span["parent_id"])
+
         args = {
             "service": span["service"],
             "name": span["name"],
             "resource": span["resource"],
             "traceID": trace_id,
             "spanID": int(span["span_id"]),
+            "parentID": parent_id,
             "start": span["start"],
             "duration": span["duration"],
             "error": span["error"],
@@ -60,9 +66,6 @@ def convert_trace_to_protobuf_payload(trace):
             "type": "",
         }
 
-        if "parent_id" in span:
-            args["parentID"] = int(span["parent_id"])
-
         span_group.append(Span(**args))
 
     traces = []
@@ -76,7 +79,7 @@ def convert_trace_to_protobuf_payload(trace):
                 endTime=first_span.start + first_span.duration,
             )
         )
-        trace_payload = TracePayload(hostName="none", env="none", traces=traces)
+        trace_payload = TracePayload(hostName=None, env="none", traces=traces)
     return trace_payload
 
 
