From 2557a24b758697307098211ecc75cb36707a54cb Mon Sep 17 00:00:00 2001
From: Niklas Gehlen <mail@niklasgehlen.de>
Date: Wed, 26 Jun 2024 16:23:53 +0200
Subject: [PATCH] Do not create spans from batch queries if parent span is not
 recording. (#36)

---
 tracer.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tracer.go b/tracer.go
index 0a6527c..091770e 100644
--- a/tracer.go
+++ b/tracer.go
@@ -248,6 +248,10 @@ func (t *Tracer) TraceBatchStart(ctx context.Context, conn *pgx.Conn, data pgx.T
 
 // TraceBatchQuery is called at the after each query in a batch.
 func (t *Tracer) TraceBatchQuery(ctx context.Context, conn *pgx.Conn, data pgx.TraceBatchQueryData) {
+	if !trace.SpanFromContext(ctx).IsRecording() {
+		return
+	}
+
 	opts := []trace.SpanStartOption{
 		trace.WithSpanKind(trace.SpanKindClient),
 		trace.WithAttributes(t.attrs...),