update

Author: JqRrt

cozeloop/internal/trace/model/model.py

@@ -36,7 +36,8 @@ class UploadType(str, Enum):
 
 @dataclass
 class FinishEventInfoExtra:
-    is_root_span: bool
+    is_root_span: bool = False
+    latency_ms: int = 0
 
 
 @dataclass

cozeloop/internal/trace/span_processor.py

@@ -11,7 +11,7 @@
 import threading
 
 from cozeloop.internal.trace.exporter import *
-from cozeloop.internal.trace.model.model import FinishEventInfo, QueueConf
+from cozeloop.internal.trace.model.model import FinishEventInfo, QueueConf, FinishEventInfoExtra
 from cozeloop.internal.trace.queue_manager import BatchQueueManager, BatchQueueManagerOptions, QUEUE_NAME_FILE_RETRY, \
     QUEUE_NAME_FILE, QUEUE_NAME_SPAN_RETRY, QUEUE_NAME_SPAN
 from cozeloop.internal.trace.span import Span
@@ -159,7 +159,9 @@ def export_func(ctx: dict, items: List[Any]):
 
             is_fail = False
             err_msg = ""
+            before = time.perf_counter()
             is_pass, msg = exporter.export_spans(ctx, upload_spans)
+            elapsed_time_ms = (time.perf_counter() - before) * 1000
             if not is_pass:
                 if span_retry_queue:
                     for span in spans:
@@ -178,6 +180,9 @@ def export_func(ctx: dict, items: List[Any]):
                     is_event_fail=is_fail,
                     item_num=len(spans),
                     detail_msg=err_msg,
+                    extra_params=FinishEventInfoExtra(
+                        latency_ms=int(elapsed_time_ms)
+                    )
                 ))
 
         return export_func
@@ -193,7 +198,9 @@ def export_func(ctx: dict, items: List[Any]):
 
             is_fail = False
             err_msg = ""
+            before = time.perf_counter()
             is_pass, msg = exporter.export_files(ctx, files)
+            elapsed_time_ms = (time.perf_counter() - before) * 1000
             if not is_pass:
                 if file_retry_queue:
                     for file in files:
@@ -208,6 +215,9 @@ def export_func(ctx: dict, items: List[Any]):
                     is_event_fail=is_fail,
                     item_num=len(files),
                     detail_msg=err_msg,
+                    extra_params=FinishEventInfoExtra(
+                        latency_ms=int(elapsed_time_ms)
+                    )
                 ))
 
         return export_func