From d94ec2219be77f0197038b438c5e23b7fb6555b2 Mon Sep 17 00:00:00 2001 From: Ashwanth Goli Date: Tue, 26 Mar 2024 11:55:43 +0530 Subject: [PATCH] httpgrpc: inject disable wrappers in ctx --- pkg/querier/queryrange/codec.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pkg/querier/queryrange/codec.go b/pkg/querier/queryrange/codec.go index 44de02408b4d..8a72bb985c53 100644 --- a/pkg/querier/queryrange/codec.go +++ b/pkg/querier/queryrange/codec.go @@ -413,6 +413,11 @@ func (Codec) DecodeHTTPGrpcRequest(ctx context.Context, r *httpgrpc.HTTPRequest) ctx = httpreq.InjectQueryTags(ctx, queryTags) } + // Add disable pipleine wrappers + if disableWrappers := httpReq.Header.Get(httpreq.LokiDisablePipelineWrappersHeader); disableWrappers != "" { + httpreq.InjectHeader(ctx, httpreq.LokiDisablePipelineWrappersHeader, disableWrappers) + } + // Add query metrics if queueTimeHeader := httpReq.Header.Get(string(httpreq.QueryQueueTimeHTTPHeader)); queueTimeHeader != "" { queueTime, err := time.ParseDuration(queueTimeHeader)