fixup logger field and metadata handling
Some checks failed
build / test (push) Failing after 1m35s
codeql / analyze (go) (push) Failing after 2m6s
build / lint (push) Successful in 9m35s

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2024-03-11 23:53:43 +03:00
parent c4b4fc187e
commit 25efb945dc

View File

@ -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)
}