feat: support custom ctx key to pass in servicei inline

Author: Duslia

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"
 )

server/service_inline.go

@@ -35,6 +35,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)