From f3e268a7ddb85ef4111067724caa93060117d209 Mon Sep 17 00:00:00 2001
From: Anthony Mahanna <anthony.mahanna@arangodb.com>
Date: Sat, 23 Dec 2023 13:29:00 -0500
Subject: [PATCH] update `benchmark/write.py`

---
 benchmark/write.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/benchmark/write.py b/benchmark/write.py
index 11b7f95..6bf1753 100644
--- a/benchmark/write.py
+++ b/benchmark/write.py
@@ -38,13 +38,15 @@
 class JaegerSpan:
     def __init__(
         self,
-        spanID: str,
-        operationName: str,
+        span_id: str,
+        operation_name: str,
+        start_time: int,
         duration: int,
         tags: list[dict[str, str]],
     ):
-        self.span_id = spanID
-        self.operation_name = operationName
+        self.span_id = span_id
+        self.operation_name = operation_name
+        self.start_time = start_time
         self.duration = duration
         self.tags = {
             tag["key"]: tag["value"]
@@ -65,6 +67,7 @@ def to_dict(self) -> dict[str, Any]:
         return {
             "spanID": self.span_id,
             "operationName": self.operation_name,
+            "startTime": self.start_time,
             "duration": self.duration,
             "tags": self.tags,
             "children": [child.to_dict() for child in self.children.values()],
@@ -110,6 +113,7 @@ def __build_span_tree(self) -> None:
             span_object = JaegerSpan(
                 span_id,
                 operation_name,
+                span["startTime"],
                 span["duration"],
                 span["tags"],
             )