feat: support custom ctx key to pass in servicei inline

cloudwego · Feb 19, 2025 · 14655b3 · 14655b3
1 parent 8bde660
commit 14655b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/pkg/consts/ctx.go b/pkg/consts/ctx.go
@@ -23,7 +23,8 @@ const (
 )
 
 const (
-	SERVICE_INLINE_RPCINFO_KEY  = "SERVICE_INLINE_RPCINFO_KEY"
-	SERVICE_INLINE_SERVICE_NAME = "SERVICE_INLINE_SERVICE_NAME"
-	SERVICE_INLINE_DATA_KEY     = "SERVICE_INLINE_DATA_KEY"
+	SERVICE_INLINE_RPCINFO_KEY    = "SERVICE_INLINE_RPCINFO_KEY"
+	SERVICE_INLINE_SERVICE_NAME   = "SERVICE_INLINE_SERVICE_NAME"
+	SERVICE_INLINE_DATA_KEY       = "SERVICE_INLINE_DATA_KEY"
+	SERVICE_INLINE_CUSTOM_CTX_KEY = "SERVICE_INLINE_CUSTOM_CTX_KEY"
 )
diff --git a/server/service_inline.go b/server/service_inline.go
@@ -37,6 +37,10 @@ var invocationType = reflect.TypeOf(rpcinfo.NewServerInvocation()).Elem()
 
 func constructServerCtxWithMetadata(cliCtx context.Context) (serverCtx context.Context) {
 	serverCtx = context.Background()
+	customVal := cliCtx.Value(consts.SERVICE_INLINE_CUSTOM_CTX_KEY)
+	if customVal != nil {
+		serverCtx = context.WithValue(serverCtx, consts.SERVICE_INLINE_CUSTOM_CTX_KEY, customVal)
+	}
 	// metainfo
 	// forward transmission
 	kvs := make(map[string]string, 16)