Skip to content

Commit d033628

Browse files
committed
unite traceid
Signed-off-by: jyjiangkai <[email protected]>
1 parent 24da846 commit d033628

File tree

3 files changed

+10
-11
lines changed

3 files changed

+10
-11
lines changed

internal/gateway/gateway.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ func (ga *ceGateway) receive(ctx context.Context, event v2.Event) (re *v2.Event,
106106
ctx, span := ga.tracer.Start(ctx, "receive")
107107
defer span.End()
108108

109-
event.SetExtension("spancontext", ctx)
109+
event.SetExtension("traceid", span.SpanContext().TraceID().String())
110110
span.SetName("EventTracing")
111111
span.SetAttributes(attribute.String("event_id", event.ID()))
112112
span.AddEvent("received from source", trace.WithTimestamp(time.Now()))

internal/trigger/reader/reader.go

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -206,12 +206,11 @@ func (elReader *eventlogReader) loop(ctx context.Context, lr api.BusReader) erro
206206
}
207207
for i := range events {
208208
_, span := elReader.newSpan(ctx, *events[i])
209-
// _, span := elReader.config.Tracer.Start(ctx, events[i].ID())
210209
span.SetName("EventTracing")
211210
span.SetAttributes(attribute.String("event_id", events[i].ID()))
212211
span.AddEvent("read from eventbus", trace.WithTimestamp(time.Now()))
213212
span.SetAttributes(attribute.String("eventbus_id", elReader.config.EventbusID.String()))
214-
// span.SetAttributes(attribute.String("eventbus_name", elReader.config.EventbusID.String()))
213+
span.SetAttributes(attribute.String("subscription_id", elReader.config.SubscriptionIDStr))
215214
ec, _ := events[i].Context.(*ce.EventContextV1)
216215
offsetByte, _ := ec.Extensions[eventlog.XVanusLogOffset].([]byte)
217216
offset := binary.BigEndian.Uint64(offsetByte)
@@ -240,14 +239,14 @@ func (elReader *eventlogReader) newSpan(ctx context.Context, event ce.Event) (co
240239
if event.Extensions() == nil {
241240
return elReader.config.Tracer.Start(ctx, event.ID())
242241
}
243-
if _, ok := event.Extensions()["spancontext"]; !ok {
242+
if _, ok := event.Extensions()["traceid"]; !ok {
244243
return elReader.config.Tracer.Start(ctx, event.ID())
245244
}
246-
var spanCtx context.Context
247-
if err := event.ExtensionAs("spancontext", &spanCtx); err != nil {
248-
return elReader.config.Tracer.Start(ctx, event.ID())
249-
}
250-
return elReader.config.Tracer.Start(spanCtx, event.ID())
245+
traceid, _ := trace.TraceIDFromHex(event.Extensions()["traceid"].(string))
246+
_ctx := trace.ContextWithSpanContext(ctx, trace.NewSpanContext(trace.SpanContextConfig{
247+
TraceID: traceid,
248+
}))
249+
return elReader.config.Tracer.Start(_ctx, event.ID())
251250
}
252251

253252
func (elReader *eventlogReader) putEvent(ctx context.Context, event info.EventRecord) error {

observability/tracing/exporter/event_trace_exporter.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,9 @@ func newEvent(span tracesdk.ReadOnlySpan) v2.Event {
124124
data["end_time"] = span.EndTime()
125125
data["status"] = span.Status()
126126
for _, attr := range span.Attributes() {
127-
if attr.Key == attribute.Key("offset") {
127+
if attr.Value.Type() == attribute.INT64 {
128128
data[string(attr.Key)] = attr.Value.AsInt64()
129-
} else {
129+
} else if attr.Value.Type() == attribute.STRING {
130130
data[string(attr.Key)] = attr.Value.AsString()
131131
}
132132
}

0 commit comments

Comments
 (0)