支持parent span 数据

Author: shynome

cmd/lcode-hub/otel-exporter.go

@@ -3,12 +3,14 @@ package main
 import (
 	"context"
 	"encoding/json"
+	"fmt"
 	"io"
 	"os"
 	"strings"
 	"sync"
 	"text/template"
 
+	"github.com/jellydator/ttlcache/v3"
 	"github.com/lainio/err2"
 	. "github.com/lainio/err2/try"
 	"github.com/vscode-lcode/lcode/v2/util/err0"
@@ -45,6 +47,7 @@ func newTracerProvider(logLevel LogLevel) (tp *sdktrace.TracerProvider) {
 type ConsoleLogExporter struct {
 	Level LogLevel
 	wg    *sync.WaitGroup
+	spans *ttlcache.Cache[string, sdktrace.ReadOnlySpan]
 }
 
 var _ sdktrace.SpanProcessor = (*ConsoleLogExporter)(nil)
@@ -53,17 +56,36 @@ func NewConsoleLogExporter(level LogLevel) *ConsoleLogExporter {
 	return &ConsoleLogExporter{
 		Level: level,
 		wg:    &sync.WaitGroup{},
+		spans: ttlcache.New[string, sdktrace.ReadOnlySpan](),
 	}
 }
 func (log *ConsoleLogExporter) Shutdown(ctx context.Context) error {
 	log.wg.Wait()
 	return nil
 }
 
-func (log *ConsoleLogExporter) OnStart(ctx context.Context, span sdktrace.ReadWriteSpan) {
-	if log.Level == NoneLogLevel {
-		return
+func SpanID(span sdktrace.ReadOnlySpan) string {
+	ctx := span.SpanContext()
+	return ctx.TraceID().String() + ctx.SpanID().String()
+}
+
+func (log *ConsoleLogExporter) getParent(span sdktrace.ReadOnlySpan) sdktrace.ReadOnlySpan {
+	parent := span.Parent()
+	if !parent.HasSpanID() {
+		return nil
+	}
+	v := span.Parent().Equal(span.SpanContext())
+	_ = v
+	id := parent.TraceID().String() + parent.SpanID().String()
+	item := log.spans.Get(id)
+	if item == nil {
+		return nil
 	}
+	return item.Value()
+}
+
+func (log *ConsoleLogExporter) OnStart(ctx context.Context, span sdktrace.ReadWriteSpan) {
+	log.spans.Set(SpanID(span), span, 0)
 	err0.ApplyStatusWithCtx(ctx, span)
 	if code := span.Status().Code; code != codes.Unset {
 		log.wg.Add(1)
@@ -80,6 +102,7 @@ func (log *ConsoleLogExporter) OnEnd(span sdktrace.ReadOnlySpan) {
 	log.wg.Add(1)
 	go func() {
 		defer log.wg.Done()
+		defer log.spans.Delete(SpanID(span))
 		log.PrintlnSpan(span)
 	}()
 }
@@ -98,6 +121,9 @@ const (
 )
 
 func (log ConsoleLogExporter) shouldLog(span sdktrace.ReadOnlySpan) bool {
+	if log.Level == NoneLogLevel {
+		return false
+	}
 	if _, onStart := span.(sdktrace.ReadWriteSpan); !onStart { // onEnd
 		for _, ev := range span.Events() {
 			if ev.Name == err0.TheLogHasBeenOutput {
@@ -122,39 +148,49 @@ func (log ConsoleLogExporter) PrintlnSpan(span sdktrace.ReadOnlySpan) (err error
 		return
 	}
 
-	var assigns = map[string]string{}
+	var status string
 	output := os.Stderr
 	switch s := span.Status(); s.Code {
 	case codes.Unset: //debug
-		assigns["status"] = "desc: " + s.Description
+		status = "desc: " + s.Description
 	case codes.Ok: //info
 		output = os.Stdout
-		assigns["status"] = "msg: " + s.Description
+		status = "msg: " + s.Description
 	case codes.Error: //error
-		assigns["status"] = "err: " + s.Description
+		status = "err: " + s.Description
 	}
 
-	assigns["name"] = "span: " + span.Name()
+	var info = ""
+	for parent := span; parent != nil; parent = log.getParent(parent) {
+		name := parent.Name()
 
-	scope := span.InstrumentationScope()
-	scope.Name = strings.TrimPrefix(scope.Name, "github.com/vscode-lcode/lcode/v2")
-	assigns["scope"] = scope.Name
+		scope := parent.InstrumentationScope()
+		scope.Name = strings.TrimPrefix(scope.Name, "github.com/vscode-lcode/lcode/v2")
+		sn := scope.Name
 
-	var attrs = map[string]any{}
-	for _, kv := range span.Attributes() {
-		k := string(kv.Key)
-		attrs[k] = kv.Value.AsInterface()
+		var attrs = map[string]any{}
+		for _, kv := range parent.Attributes() {
+			k := string(kv.Key)
+			attrs[k] = kv.Value.AsInterface()
+		}
+		attrsBytes := To1(json.Marshal(attrs))
+
+		p := fmt.Sprintf("[%s]-%s-%s", sn, name, string(attrsBytes))
+		if info != "" {
+			info = fmt.Sprintf("%s -> %s", p, info)
+		} else {
+			info = p
+		}
 	}
-	b := To1(json.Marshal(attrs))
-	assigns["attrs"] = "attrs: " + string(b)
+	info = fmt.Sprintf("%s %s", info, status)
 
 	_, onStart := span.(sdktrace.ReadWriteSpan)
 	var endStr = "\n"
 	if onEnd := !onStart; onEnd {
 		endStr = " end\n"
 	}
 
-	To(tpl.Execute(output, assigns))
-	io.WriteString(output, endStr)
+	To1(io.WriteString(output, info))
+	To1(io.WriteString(output, endStr))
 	return
 }

cmd/lcode-hub/otel-exporter_test.go

@@ -16,7 +16,9 @@ func TestExporter(t *testing.T) {
 
 	tracer := otel.Tracer("test")
 
-	_, span := tracer.Start(context.Background(), "err")
+	p, _ := tracer.Start(context.Background(), "parent")
+
+	_, span := tracer.Start(p, "err")
 	span.SetStatus(codes.Error, "some err string")
 	span.End()