fix(confluent-kafka): changed function order

open-telemetry · Aug 19, 2023 · 51167dd · 51167dd
1 parent 1815057
commit 51167dd
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/...nstrumentation-confluent-kafka/src/opentelemetry/instrumentation/confluent_kafka/utils.py b/...nstrumentation-confluent-kafka/src/opentelemetry/instrumentation/confluent_kafka/utils.py
@@ -99,6 +99,18 @@ def _create_new_consume_span(instance, tracer, records):
     )
 
 
+def _get_links_from_records(records):
+    links = []
+    for record in records:
+        ctx = propagate.extract(record.headers(), getter=_kafka_getter)
+        if ctx:
+            for item in ctx.values():
+                if hasattr(item, "get_span_context"):
+                    links.append(Link(context=item.get_span_context()))
+
+    return links
+
+
 def _enrich_span(
     span,
     topic,
@@ -134,18 +146,6 @@ def _enrich_span(
         )
 
 
-def _get_links_from_records(records):
-    links = []
-    for record in records:
-        ctx = propagate.extract(record.headers(), getter=_kafka_getter)
-        if ctx:
-            for item in ctx.values():
-                if hasattr(item, "get_span_context"):
-                    links.append(Link(context=item.get_span_context()))
-
-    return links
-
-
 _kafka_setter = KafkaContextSetter()