diff --git a/go/logger/logger.go b/go/logger/logger.go new file mode 100644 index 0000000000..e13de5d580 --- /dev/null +++ b/go/logger/logger.go @@ -0,0 +1,20 @@ +package logger + +import ( + "context" + + "github.com/vmware/alb-sdk/go/models" +) + +func GetTraceID(ctx context.Context) string { + traceIDType := ctx.Value(models.TraceID) + traceID, ok := traceIDType.(string) + if !ok { + return "" + } + return traceID +} + +func SetTraceID(ctx context.Context, traceID string) context.Context { + return context.WithValue(ctx, models.TraceID, traceID) +} diff --git a/go/models/logger.go b/go/models/logger.go new file mode 100644 index 0000000000..030d5a0a92 --- /dev/null +++ b/go/models/logger.go @@ -0,0 +1,5 @@ +package models + +type TraceIDType string + +const TraceID TraceIDType = "trace-id"