check ctxStore in intcp

gotomicro · Jan 3, 2024 · 97106f7 · 97106f7
1 parent ce7467d
commit 97106f7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/grpc/direct/server/main.go b/examples/grpc/direct/server/main.go
@@ -38,6 +38,7 @@ func (g Greeter) SayHello(ctx context.Context, request *helloworld.HelloRequest)
 		return nil, status.Error(codes.Unavailable, "error")
 	}
 	egrpc.CtxStoreSet(ctx, "x-biz-guid", "123")
+	egrpc.CtxStoreSet(ctx, "x-biz-uid", "100")
 	// header := metadata.Pairs("x-header-key", "val")
 	// err := grpc.SendHeader(context, header)
 	// if err != nil {

diff --git a/server/egrpc/interceptor.go b/server/egrpc/interceptor.go
@@ -288,8 +288,10 @@ func (c *Container) defaultUnaryServerInterceptor() grpc.UnaryServerInterceptor
 
 			skv, skvOk := ctx.Value(ctxStoreStruct{}).(*ctxStore)
 			for _, key := range loggerKeys {
-				if v, ok := skv.kvs[key]; skvOk && ok {
-					fields = append(fields, elog.Any(strings.ToLower(key), v))
+				if skvOk {
+					if v, ok := skv.kvs[key]; ok {
+						fields = append(fields, elog.Any(strings.ToLower(key), v))
+					}
 				}
 				if value := tools.ContextValue(ctx, key); value != "" {
 					fields = append(fields, elog.FieldCustomKeyValue(strings.ToLower(key), value))