set status and tracestate

lightstep · Jun 3, 2023 · 2718ebe · 2718ebe
1 parent c8fa020
commit 2718ebe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/lightstep/sdk/trace/exporters/otlp/otelcol/client_test.go b/lightstep/sdk/trace/exporters/otlp/otelcol/client_test.go
@@ -29,6 +29,7 @@ import (
 	"go.opentelemetry.io/collector/config/confignet"
 	"go.opentelemetry.io/collector/consumer/consumertest"
 	"go.opentelemetry.io/collector/pdata/pcommon"
+	"go.opentelemetry.io/collector/pdata/ptrace"
 	"go.opentelemetry.io/collector/pdata/ptrace/ptraceotlp"
 	"go.opentelemetry.io/collector/receiver"
 	"go.opentelemetry.io/collector/receiver/receivertest"
@@ -299,6 +300,9 @@ func (t *clientTestSuite) TestD2PD() {
 	t.Equal(uint32(roSpan.DroppedLinks()), actualSpan.DroppedLinksCount())
 	t.Equal(uint32(roSpan.DroppedEvents()), actualSpan.DroppedEventsCount())
 	t.Equal(uint32(roSpan.SpanKind()), uint32(actualSpan.Kind()))
+	t.Equal(roSpan.SpanContext().TraceState().String(), actualSpan.TraceState().AsRaw())
+	t.Equal(ptrace.StatusCode(roSpan.Status().Code), actualSpan.Status().Code())
+	t.Equal(roSpan.Status().Description, actualSpan.Status().Message())
 
 	for _, attr := range roSpan.Attributes() {
 		actualVal, ok := actualSpan.Attributes().Get(string(attr.Key))

diff --git a/lightstep/sdk/trace/exporters/otlp/otelcol/otlp.go b/lightstep/sdk/trace/exporters/otlp/otelcol/otlp.go
@@ -127,6 +127,9 @@ func (c *client) d2pd(in []trace.ReadOnlySpan) ptrace.Traces {
 		s.SetStartTimestamp(pcommon.NewTimestampFromTime(tr.StartTime()))
 		s.SetEndTimestamp(pcommon.NewTimestampFromTime(tr.EndTime()))
 		s.SetTraceID(pcommon.TraceID(tr.SpanContext().TraceID()))
+		s.TraceState().FromRaw(tr.SpanContext().TraceState().String())
+		s.Status().SetCode(ptrace.StatusCode(tr.Status().Code))
+		s.Status().SetMessage(tr.Status().Description)
 
 		copyAttributes(s.Attributes(), attribute.NewSet(tr.Attributes()...))
 		copyEvents(s.Events(), tr.Events())