Simplify TraceParent constructor

prisma · Sep 26, 2024 · 4a7d9d4 · 4a7d9d4
1 parent 7162560
commit 4a7d9d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/libs/telemetry/src/helpers.rs b/libs/telemetry/src/helpers.rs
@@ -33,15 +33,11 @@ pub struct TraceParent {
 }
 
 impl TraceParent {
-    pub fn new_unsafe(trace_id: TraceId, span_id: SpanId, sampled: bool) -> Self {
+    pub fn new_unsafe(trace_id: TraceId, span_id: SpanId, flags: TraceFlags) -> Self {
         Self {
             trace_id,
             span_id,
-            flags: if sampled {
-                TraceFlags::SAMPLED
-            } else {
-                TraceFlags::default()
-            },
+            flags,
         }
     }
 

diff --git a/query-engine/core/src/interactive_transactions/mod.rs b/query-engine/core/src/interactive_transactions/mod.rs
@@ -1,5 +1,5 @@
 use derive_more::Display;
-use opentelemetry::trace::{SpanId, TraceId};
+use opentelemetry::trace::{SpanId, TraceFlags, TraceId};
 use serde::Deserialize;
 
 use telemetry::helpers::TraceParent;
@@ -51,7 +51,7 @@ impl TxId {
     /// emitted. Same transaction id is guaranteed to have traceparent with the same trace_id and
     /// span_id.
     pub fn as_traceparent(&self) -> TraceParent {
-        TraceParent::new_unsafe(self.as_trace_id(), self.as_span_id(), false)
+        TraceParent::new_unsafe(self.as_trace_id(), self.as_span_id(), TraceFlags::default())
     }
 }