From 25efb945dcbd6aca1803b542ae2d5eb16a7d9dd0 Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Mon, 11 Mar 2024 23:53:43 +0300 Subject: [PATCH] fixup logger field and metadata handling Signed-off-by: Vasiliy Tolstov --- requestid.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/requestid.go b/requestid.go index 0f47e01..bb9b533 100644 --- a/requestid.go +++ b/requestid.go @@ -3,6 +3,7 @@ package requestid // import "go.unistack.org/micro-wrapper-requestid/v3" import ( "context" "net/textproto" + "strings" "go.unistack.org/micro/v3/client" "go.unistack.org/micro/v3/logger" @@ -12,9 +13,10 @@ import ( ) func init() { + requestIDLog := strings.ToLower(DefaultMetadataKey) logger.DefaultContextAttrFuncs = append(logger.DefaultContextAttrFuncs, func(ctx context.Context) []interface{} { if v, ok := ctx.Value(XRequestIDKey{}).(string); ok { - return []interface{}{DefaultMetadataKey, v} + return []interface{}{requestIDLog, v} } return nil }) @@ -51,7 +53,7 @@ var DefaultMetadataFunc = func(ctx context.Context) (context.Context, error) { if !ok { omd = metadata.New(1) omd.Set(DefaultMetadataKey, xid) - ctx = metadata.NewOutgoingContext(ctx, imd) + ctx = metadata.NewOutgoingContext(ctx, omd) } else if _, ok = omd.Get(DefaultMetadataKey); !ok { omd.Set(DefaultMetadataKey, xid) }