Send down the labels into the dispatch requests from the lazy router

leonlnj · Nov 3, 2022 · 0aa265c · 0aa265c
1 parent d2dc9aa
commit 0aa265c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/interceptor.go b/interceptor.go
@@ -12,6 +12,8 @@ var (
 	CtxComponentIDKey CtxKey = "CTX_COMPONENT_ID"
 	// CtxComponentKindKey is used to denote the component's kind in the request context
 	CtxComponentKindKey CtxKey = "CTX_COMPONENT_KIND"
+	// CtxComponentLabelsKey is used to denote the component's labels in the request context
+	CtxComponentLabelsKey CtxKey = "CTX_COMPONENT_LABELS"
 )
 
 // Interceptor is the interface for a structural interceptor

diff --git a/lazy_router.go b/lazy_router.go
@@ -75,7 +75,7 @@ func (r *LazyRouter) Dispatch(ctx context.Context, req Request) ResponseQueue {
 			for _, route := range routes {
 				copyReq, _ := req.Clone()
 				responses := make([]Response, 0)
-				responseCh := route.Dispatch(ctx, copyReq).Iter()
+				responseCh := route.Dispatch(context.WithValue(ctx, CtxComponentLabelsKey, labels), copyReq).Iter()
 				ok := true
 				for ok {
 					select {