feat: implement query splitting in the query-tee (#20039)

Author: trevorwhitney

pkg/querier/queryrange/engine_router.go

@@ -59,9 +59,9 @@ type engineRouter struct {
 	clock quartz.Clock
 }
 
-// newEngineRouterMiddleware creates a middleware that splits and routes part of the query
+// NewEngineRouterMiddleware creates a middleware that splits and routes part of the query
 // to v2 engine if the query is supported by it.
-func newEngineRouterMiddleware(
+func NewEngineRouterMiddleware(
 	v2RouterConfig RouterConfig,
 	v1Chain []queryrangebase.Middleware,
 	merger queryrangebase.Merger,

pkg/querier/queryrange/engine_router_test.go

@@ -309,7 +309,7 @@ func Test_engineRouter_Do(t *testing.T) {
 		Handler:  v2EngineHandler,
 	}
 
-	router := newEngineRouterMiddleware(
+	router := NewEngineRouterMiddleware(
 		routerConfig,
 		[]queryrangebase.Middleware{newEntrySuffixTestMiddleware(" [v1-chain-processed]")},
 		DefaultCodec,

pkg/querier/queryrange/roundtrip.go

@@ -732,7 +732,7 @@ func NewLogFilterTripperware(cfg Config, engineOpts logql.EngineOpts, routerConf
 
 		// route query range supported by v2 engine to the new engine handler.
 		if routerConfig.Enabled {
-			engineRouterMiddleware := newEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, false, log)
+			engineRouterMiddleware := NewEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, false, log)
 			queryRangeMiddleware = append(
 				queryRangeMiddleware,
 				base.InstrumentMiddleware("v2_engine_router", metrics.InstrumentMiddlewareMetrics),
@@ -797,7 +797,7 @@ func NewLimitedTripperware(cfg Config, engineOpts logql.EngineOpts, routerConfig
 
 		// route query range supported by v2 engine to the new engine handler.
 		if routerConfig.Enabled {
-			engineRouterMiddleware := newEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, false, log)
+			engineRouterMiddleware := NewEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, false, log)
 			queryRangeMiddleware = append(
 				queryRangeMiddleware,
 				base.InstrumentMiddleware("v2_engine_router", metrics.InstrumentMiddlewareMetrics),
@@ -1084,7 +1084,7 @@ func NewMetricTripperware(cfg Config, engineOpts logql.EngineOpts, routerConfig
 
 		// route query range supported by v2 engine to the new engine handler.
 		if routerConfig.Enabled && !disableEngineRouter {
-			engineRouterMiddleware := newEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, true, log)
+			engineRouterMiddleware := NewEngineRouterMiddleware(routerConfig, chunksEngineMWs, merger, true, log)
 			queryRangeMiddleware = append(
 				queryRangeMiddleware,
 				base.InstrumentMiddleware("v2_engine_router", metrics.InstrumentMiddlewareMetrics),