Add a method to transform the TraceContext into a map[string]interface{}

interfaces_test.go

@@ -11,7 +11,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/fatih/structs"
 	"github.com/google/go-cmp/cmp"
 )
 
@@ -142,13 +141,13 @@ func TestStructSnapshots(t *testing.T) {
 				StartTime: time.Unix(3, 0).UTC(),
 				Timestamp: time.Unix(5, 0).UTC(),
 				Contexts: map[string]Context{
-					"trace": structs.Map(TraceContext{
+					"trace": TraceContext{
 						TraceID:     TraceIDFromHex("90d57511038845dcb4164a70fc3a7fdb"),
 						SpanID:      SpanIDFromHex("f7f3fd754a9040eb"),
 						Op:          "http.GET",
 						Description: "description",
 						Status:      SpanStatusOK,
-					}),
+					}.Map(),
 				},
 			},
 		},

tracing.go

@@ -150,7 +150,7 @@ func StartSpan(ctx context.Context, operation string, options ...SpanOption) *Sp
 
 	// Update scope so that all events include a trace context, allowing
 	// Sentry to correlate errors to transactions/spans.
-	hubFromContext(ctx).Scope().SetContext("trace", structs.Map(span.traceContext()))
+	hubFromContext(ctx).Scope().SetContext("trace", span.traceContext().Map())
 
 	return &span
 }
@@ -501,6 +501,31 @@ func (tc *TraceContext) MarshalJSON() ([]byte, error) {
 	})
 }
 
+func (tc TraceContext) Map() map[string]interface{} {
+	m := map[string]interface{}{
+		"trace_id": tc.TraceID,
+		"span_id":  tc.SpanID,
+	}
+
+	if tc.ParentSpanID != [8]byte{} {
+		m["parent_span_id"] = tc.ParentSpanID
+	}
+
+	if tc.Op != "" {
+		m["op"] = tc.Op
+	}
+
+	if tc.Description != "" {
+		m["description"] = tc.Description
+	}
+
+	if tc.Status > 0 && tc.Status < maxSpanStatus {
+		m["status"] = tc.Status
+	}
+
+	return m
+}
+
 // Sampled signifies a sampling decision.
 type Sampled int8
 

tracing_test.go

@@ -13,7 +13,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/fatih/structs"
 	"github.com/google/go-cmp/cmp"
 	"github.com/google/go-cmp/cmp/cmpopts"
 )
@@ -132,14 +131,14 @@ func TestStartSpan(t *testing.T) {
 		Type:        transactionType,
 		Transaction: transaction,
 		Contexts: map[string]Context{
-			"trace": structs.Map(TraceContext{
+			"trace": TraceContext{
 				TraceID:      span.TraceID,
 				SpanID:       span.SpanID,
 				ParentSpanID: parentSpanID,
 				Op:           op,
 				Description:  description,
 				Status:       status,
-			}),
+			}.Map(),
 		},
 		Tags: nil,
 		// TODO(tracing): the root span / transaction data field is
@@ -192,11 +191,11 @@ func TestStartChild(t *testing.T) {
 		Type:        transactionType,
 		Transaction: "Test Transaction",
 		Contexts: map[string]Context{
-			"trace": structs.Map(TraceContext{
+			"trace": TraceContext{
 				TraceID: span.TraceID,
 				SpanID:  span.SpanID,
 				Op:      span.Op,
-			}),
+			}.Map(),
 		},
 		Spans: []*Span{
 			{