update condition in pushDerived to check for single shard routing

ChaitanyaD48 · ChaitanyaD48 · commit 2ccbd53dc15e · 2025-12-03T02:50:50.000+05:30
Signed-off-by: ChaitanyaD48 &lt;chaitanya.d48@gmail.com&gt;
diff --git a/go/vt/vtgate/planbuilder/operators/route_planning.go b/go/vt/vtgate/planbuilder/operators/route_planning.go
@@ -44,7 +44,7 @@ func pushDerived(ctx *plancontext.PlanningContext, op *Horizon) (Operator, *Appl
 		return op, NoRewrite
 	}
 
-	if !(innerRoute.Routing.OpCode() == engine.EqualUnique) && !op.IsMergeable(ctx) {
+	if !innerRoute.Routing.OpCode().IsSingleShard() && !op.IsMergeable(ctx) {
 		// no need to check anything if we are sure that we will only hit a single shard
 		return op, NoRewrite
 	}

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ func pushDerived(ctx plancontext.PlanningContext, op Horizon) (Operator, *Appl`
`44`	`44`	`return op, NoRewrite`
`45`	`45`	`}`
`46`	`46`
`47`		`- if !(innerRoute.Routing.OpCode() == engine.EqualUnique) && !op.IsMergeable(ctx) {`
	`47`	`+ if !innerRoute.Routing.OpCode().IsSingleShard() && !op.IsMergeable(ctx) {`
`48`	`48`	`// no need to check anything if we are sure that we will only hit a single shard`
`49`	`49`	`return op, NoRewrite`
`50`	`50`	`}`