PushLeftStage fix (#7270)

shnikd · web-flow · commit 59d5dd6d7066 · 2024-07-31T18:42:24.000+03:00
diff --git a/ydb/core/kqp/opt/physical/kqp_opt_phy.cpp b/ydb/core/kqp/opt/physical/kqp_opt_phy.cpp
@@ -419,7 +419,7 @@ class TKqpPhysicalOptTransformer : public TOptimizeTransformerBase {
     {
         // TODO: Allow push to left stage for data queries.
         // It is now possible as we don't use datashard transactions for reads in data queries.
-        bool pushLeftStage = !KqpCtx.IsDataQuery() && AllowFuseJoinInputs(node);
+        bool pushLeftStage = (KqpCtx.IsScanQuery() || KqpCtx.Config->EnableKqpDataQueryStreamLookup) && AllowFuseJoinInputs(node);
         TExprBase output = DqBuildJoin(node, ctx, optCtx, *getParents(), IsGlobal,
             pushLeftStage, KqpCtx.Config->GetHashJoinMode(), false
         );

Original file line number	Diff line number	Diff line change
`@@ -419,7 +419,7 @@ class TKqpPhysicalOptTransformer : public TOptimizeTransformerBase {`
`419`	`419`	`{`
`420`	`420`	`// TODO: Allow push to left stage for data queries.`
`421`	`421`	`// It is now possible as we don't use datashard transactions for reads in data queries.`
`422`		`- bool pushLeftStage = !KqpCtx.IsDataQuery() && AllowFuseJoinInputs(node);`
	`422`	`+ bool pushLeftStage = (KqpCtx.IsScanQuery() \|\| KqpCtx.Config->EnableKqpDataQueryStreamLookup) && AllowFuseJoinInputs(node);`
`423`	`423`	`TExprBase output = DqBuildJoin(node, ctx, optCtx, *getParents(), IsGlobal,`
`424`	`424`	`pushLeftStage, KqpCtx.Config->GetHashJoinMode(), false`
`425`	`425`	`);`