Drop unused flag in IsYtCompleteIsolatedLambda (#7587)

rvu1024 · web-flow · commit 403d37f5db96 · 2024-08-09T11:22:51.000+03:00
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_join.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_join.cpp
@@ -31,14 +31,14 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node
         }
         if (auto maybeFlatMap = list.Maybe<TCoFlatMapBase>()) {
             TSyncMap syncList;
-            if (!IsYtCompleteIsolatedLambda(maybeFlatMap.Cast().Lambda().Ref(), syncList, usedCluster, true, false)) {
+            if (!IsYtCompleteIsolatedLambda(maybeFlatMap.Cast().Lambda().Ref(), syncList, usedCluster, false)) {
                 return node;
             }
             list = maybeFlatMap.Cast().Input();
         }
         if (!IsYtProviderInput(list)) {
             TSyncMap syncList;
-            if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, true, false)) {
+            if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, false)) {
                 return node;
             }
             continue;
@@ -84,7 +84,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node
         if (auto maybeFlatMap = listStepForward.Maybe<TCoFlatMapBase>()) {
             auto flatMap = maybeFlatMap.Cast();
             if (IsLambdaSuitableForPullingIntoEquiJoin(flatMap, joinInput.Scope().Ref(), tableKeysMap, extractedMembers.Get())) {
-                if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), worldList, usedCluster, true, false)) {
+                if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), worldList, usedCluster, false)) {
                     return node;
                 }
 
@@ -134,7 +134,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node
             }
             else {
                 TSyncMap syncList;
-                if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, true, false)) {
+                if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, false)) {
                     return node;
                 }
 
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_map.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_map.cpp
@@ -62,7 +62,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::FlatMap(TExprBase node,
 
     auto cluster = TString{GetClusterName(input)};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), syncList, cluster, false)) {
         return node;
     }
 
@@ -150,7 +150,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::LMap(TExprBase node, TE
 
     auto cluster = TString{GetClusterName(lmap.Input())};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(lmap.Lambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(lmap.Lambda().Ref(), syncList, cluster, false)) {
         return node;
     }
 
@@ -228,19 +228,19 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::CombineByKey(TExprBase
 
     auto cluster = TString{GetClusterName(input)};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(combineByKey.PreMapLambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(combineByKey.PreMapLambda().Ref(), syncList, cluster, false)) {
         return node;
     }
-    if (!IsYtCompleteIsolatedLambda(combineByKey.KeySelectorLambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(combineByKey.KeySelectorLambda().Ref(), syncList, cluster, false)) {
         return node;
     }
-    if (!IsYtCompleteIsolatedLambda(combineByKey.InitHandlerLambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(combineByKey.InitHandlerLambda().Ref(), syncList, cluster, false)) {
         return node;
     }
-    if (!IsYtCompleteIsolatedLambda(combineByKey.UpdateHandlerLambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(combineByKey.UpdateHandlerLambda().Ref(), syncList, cluster, false)) {
         return node;
     }
-    if (!IsYtCompleteIsolatedLambda(combineByKey.FinishHandlerLambda().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(combineByKey.FinishHandlerLambda().Ref(), syncList, cluster, false)) {
         return node;
     }
 
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_misc.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_misc.cpp
@@ -244,7 +244,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Mux(TExprBase node, TEx
                 return node;
             }
             TSyncMap syncList;
-            if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, true, false)) {
+            if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, false)) {
                 return node;
             }
 
@@ -319,7 +319,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::TakeOrSkip(TExprBase no
 
     auto cluster = TString{GetClusterName(input)};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(countBase.Count().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(countBase.Count().Ref(), syncList, cluster, false)) {
         return node;
     }
 
@@ -558,7 +558,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Extend(TExprBase node,
                 return node;
             }
             TSyncMap syncList;
-            if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, true, false)) {
+            if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, false)) {
                 return node;
             }
 
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_partition.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_partition.cpp
@@ -35,8 +35,8 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PartitionByKey(TExprBas
 
     auto cluster = TString{GetClusterName(input)};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, true, false)
-        || !IsYtCompleteIsolatedLambda(handlerLambda.Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, false)
+        || !IsYtCompleteIsolatedLambda(handlerLambda.Ref(), syncList, cluster, false)) {
         return node;
     }
 
@@ -94,7 +94,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PartitionByKey(TExprBas
         }
 
         TCoLambda sortKeySelectorLambda = partByKey.SortKeySelectorLambda().Cast<TCoLambda>();
-        if (!IsYtCompleteIsolatedLambda(sortKeySelectorLambda.Ref(), syncList, cluster, true, false)) {
+        if (!IsYtCompleteIsolatedLambda(sortKeySelectorLambda.Ref(), syncList, cluster, false)) {
             return node;
         }
 
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_push.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_push.cpp
@@ -162,7 +162,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PushDownKeyExtract(TExp
 
             TSyncMap syncList;
             for (auto filter: keyFilters) {
-                if (!IsYtCompleteIsolatedLambda(*filter, syncList, true, false)) {
+                if (!IsYtCompleteIsolatedLambda(*filter, syncList, false)) {
                     return node;
                 }
             }
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_sort.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_sort.cpp
@@ -73,7 +73,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Sort(TExprBase node, TE
     auto keySelectorLambda = sort.KeySelectorLambda();
     auto cluster = TString{GetClusterName(sort.Input())};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, false)) {
         return node;
     }
 
diff --git a/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_write.cpp b/ydb/library/yql/providers/yt/provider/phy_opt/yql_yt_phy_opt_write.cpp
@@ -43,7 +43,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::DqWrite(TExprBase node,
     }
 
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, true, true)) {
+    if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, true)) {
         return node;
     }
 
@@ -554,7 +554,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::ReplaceStatWriteTable(T
 
         TString cluster;
         TSyncMap syncList;
-        if (!IsYtCompleteIsolatedLambda(input.Ref(), syncList, cluster, true, false)) {
+        if (!IsYtCompleteIsolatedLambda(input.Ref(), syncList, cluster, false)) {
             return node;
         }
 
@@ -710,7 +710,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Fill(TExprBase node, TE
 
     auto cluster = TString{write.DataSink().Cluster().Value()};
     TSyncMap syncList;
-    if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, cluster, true, false)) {
+    if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, cluster, false)) {
         return node;
     }
 
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_datasource.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_datasource.cpp
@@ -270,7 +270,7 @@ class TYtDataSource : public TDataProviderBase {
 
     bool CanBuildResult(const TExprNode& node, TSyncMap& syncList) override {
         TString usedCluster;
-        return IsYtCompleteIsolatedLambda(node, syncList, usedCluster, true, false);
+        return IsYtCompleteIsolatedLambda(node, syncList, usedCluster, false);
     }
 
     bool GetExecWorld(const TExprNode::TPtr& node, TExprNode::TPtr& root) override {
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_datasource_exec.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_datasource_exec.cpp
@@ -114,7 +114,7 @@ class TYtDataSourceExecTransformer : public TExecTransformerBase {
 
         TString usedCluster;
         TSyncMap syncList;
-        if (!IsYtIsolatedLambda(data.Ref(), syncList, usedCluster, true, false)) {
+        if (!IsYtIsolatedLambda(data.Ref(), syncList, usedCluster, false)) {
             ctx.AddError(TIssue(ctx.GetPosition(data.Pos()), TStringBuilder() << "Failed to execute node due to bad graph: " << input->Content()));
             return SyncError();
         }
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_dq_optimize.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_dq_optimize.cpp
@@ -97,7 +97,7 @@ class TYtDqOptimizers: public TDqOptimizationBase {
         }
 
         TSyncMap syncList;
-        if (!IsYtCompleteIsolatedLambda(count.Count().Ref(), syncList, cluster, true, false)) {
+        if (!IsYtCompleteIsolatedLambda(count.Count().Ref(), syncList, cluster, false)) {
             return read;
         }
 
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_helpers.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_helpers.cpp
@@ -38,7 +38,7 @@ using namespace NNodes;
 
 namespace {
 
-bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TString* usedCluster, bool supportsReads, bool supportsDq, TNodeSet& visited) {
+bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TString* usedCluster, bool supportsDq, TNodeSet& visited) {
     if (!visited.insert(&lambdaBody).second) {
         return true;
     }
@@ -50,14 +50,14 @@ bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TSt
     if (auto maybeLength = TMaybeNode<TYtLength>(&lambdaBody)) {
         if (auto maybeRead = maybeLength.Input().Maybe<TYtReadTable>()) {
             auto read = maybeRead.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
                 return false;
             }
             syncList.emplace(read.Ptr(), syncList.size());
         }
         if (auto maybeOutput = maybeLength.Input().Maybe<TYtOutput>()) {
             auto output = maybeOutput.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
                 return false;
             }
             syncList.emplace(output.Operation().Ptr(), syncList.size());
@@ -68,14 +68,14 @@ bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TSt
     if (auto maybeContent = TMaybeNode<TYtTableContent>(&lambdaBody)) {
         if (auto maybeRead = maybeContent.Input().Maybe<TYtReadTable>()) {
             auto read = maybeRead.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
                return false;
             }
             syncList.emplace(read.Ptr(), syncList.size());
         }
         if (auto maybeOutput = maybeContent.Input().Maybe<TYtOutput>()) {
             auto output = maybeOutput.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
                 return false;
             }
             syncList.emplace(output.Operation().Ptr(), syncList.size());
@@ -89,14 +89,14 @@ bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TSt
         }
         if (auto maybeRead = maybeContent.Input().Maybe<TYtReadTable>()) {
             auto read = maybeRead.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
                 return false;
             }
             syncList.emplace(read.Ptr(), syncList.size());
         }
         if (auto maybeOutput = maybeContent.Input().Maybe<TYtOutput>()) {
             auto output = maybeOutput.Cast();
-            if (supportsReads && usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
+            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{GetOutputOp(output).DataSink().Cluster().Value()})) {
                 return false;
             }
             syncList.emplace(output.Operation().Ptr(), syncList.size());
@@ -108,28 +108,26 @@ bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TSt
         return false;
     }
 
-    if (supportsReads) {
-        if (auto maybeRead = TMaybeNode<TCoRight>(&lambdaBody).Input().Maybe<TYtReadTable>()) {
-            auto read = maybeRead.Cast();
-            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
-                return false;
-            }
-            syncList.emplace(read.Ptr(), syncList.size());
-            return true;
-        } else if (auto out = TMaybeNode<TYtOutput>(&lambdaBody)) {
-            auto op = GetOutputOp(out.Cast());
-            if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{op.DataSink().Cluster().Value()})) {
-                return false;
-            }
-            syncList.emplace(out.Cast().Operation().Ptr(), syncList.size());
-            return true;
+    if (auto maybeRead = TMaybeNode<TCoRight>(&lambdaBody).Input().Maybe<TYtReadTable>()) {
+        auto read = maybeRead.Cast();
+        if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{read.DataSource().Cluster().Value()})) {
+            return false;
         }
+        syncList.emplace(read.Ptr(), syncList.size());
+        return true;
+    } else if (auto out = TMaybeNode<TYtOutput>(&lambdaBody)) {
+        auto op = GetOutputOp(out.Cast());
+        if (usedCluster && !UpdateUsedCluster(*usedCluster, TString{op.DataSink().Cluster().Value()})) {
+            return false;
+        }
+        syncList.emplace(out.Cast().Operation().Ptr(), syncList.size());
+        return true;
     }
 
     if (auto right = TMaybeNode<TCoRight>(&lambdaBody).Input()) {
         if (auto maybeCons = right.Maybe<TCoCons>()) {
             syncList.emplace(maybeCons.Cast().World().Ptr(), syncList.size());
-            return IsYtIsolatedLambdaImpl(maybeCons.Cast().Input().Ref(), syncList, usedCluster, supportsReads, supportsDq, visited);
+            return IsYtIsolatedLambdaImpl(maybeCons.Cast().Input().Ref(), syncList, usedCluster, supportsDq, visited);
         }
 
         if (right.Cast().Raw()->IsCallable("PgReadTable!")) {
@@ -148,7 +146,7 @@ bool IsYtIsolatedLambdaImpl(const TExprNode& lambdaBody, TSyncMap& syncList, TSt
     }
 
     for (auto& child : lambdaBody.Children()) {
-        if (!IsYtIsolatedLambdaImpl(*child, syncList, usedCluster, supportsReads, supportsDq, visited)) {
+        if (!IsYtIsolatedLambdaImpl(*child, syncList, usedCluster, supportsDq, visited)) {
             return false;
         }
     }
@@ -293,22 +291,22 @@ bool UpdateUsedCluster(TString& usedCluster, const TString& newCluster) {
     return true;
 }
 
-bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, bool supportsReads, bool supportsDq) {
+bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, bool supportsDq) {
     TNodeSet visited;
-    return IsYtIsolatedLambdaImpl(lambdaBody, syncList, nullptr, supportsReads, supportsDq, visited);
+    return IsYtIsolatedLambdaImpl(lambdaBody, syncList, nullptr, supportsDq, visited);
 }
 
-bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsReads, bool supportsDq) {
+bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsDq) {
     TNodeSet visited;
-    return IsYtIsolatedLambdaImpl(lambdaBody, syncList, &usedCluster, supportsReads, supportsDq, visited);
+    return IsYtIsolatedLambdaImpl(lambdaBody, syncList, &usedCluster, supportsDq, visited);
 }
 
-bool IsYtCompleteIsolatedLambda(const TExprNode& lambda, TSyncMap& syncList, bool supportsReads, bool supportsDq) {
-    return lambda.IsComplete() && IsYtIsolatedLambda(lambda, syncList, supportsReads, supportsDq);
+bool IsYtCompleteIsolatedLambda(const TExprNode& lambda, TSyncMap& syncList, bool supportsDq) {
+    return lambda.IsComplete() && IsYtIsolatedLambda(lambda, syncList, supportsDq);
 }
 
-bool IsYtCompleteIsolatedLambda(const TExprNode& lambda, TSyncMap& syncList, TString& usedCluster, bool supportsReads, bool supportsDq) {
-    return lambda.IsComplete() && IsYtIsolatedLambda(lambda, syncList, usedCluster, supportsReads, supportsDq);
+bool IsYtCompleteIsolatedLambda(const TExprNode& lambda, TSyncMap& syncList, TString& usedCluster, bool supportsDq) {
+    return lambda.IsComplete() && IsYtIsolatedLambda(lambda, syncList, usedCluster, supportsDq);
 }
 
 TExprNode::TPtr YtCleanupWorld(const TExprNode::TPtr& input, TExprContext& ctx, TYtState::TPtr state) {
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_helpers.h b/ydb/library/yql/providers/yt/provider/yql_yt_helpers.h
@@ -21,9 +21,9 @@
 namespace NYql {
 
 bool UpdateUsedCluster(TString& usedCluster, const TString& newCluster);
-bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsReads, bool supportsDq);
-bool IsYtCompleteIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, bool supportsReads, bool supportsDq);
-bool IsYtCompleteIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsReads, bool supportsDq);
+bool IsYtIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsDq);
+bool IsYtCompleteIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, bool supportsDq);
+bool IsYtCompleteIsolatedLambda(const TExprNode& lambdaBody, TSyncMap& syncList, TString& usedCluster, bool supportsDq);
 TExprNode::TPtr YtCleanupWorld(const TExprNode::TPtr& input, TExprContext& ctx, TYtState::TPtr state);
 TVector<TYtTableBaseInfo::TPtr> GetInputTableInfos(NNodes::TExprBase input);
 TVector<TYtPathInfo::TPtr> GetInputPaths(NNodes::TExprBase input);
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_physical_finalizing.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_physical_finalizing.cpp

Original file line number	Diff line number	Diff line change
`@@ -31,14 +31,14 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node`
`31`	`31`	`}`
`32`	`32`	`if (auto maybeFlatMap = list.Maybe<TCoFlatMapBase>()) {`
`33`	`33`	`TSyncMap syncList;`
`34`		`- if (!IsYtCompleteIsolatedLambda(maybeFlatMap.Cast().Lambda().Ref(), syncList, usedCluster, true, false)) {`
	`34`	`+ if (!IsYtCompleteIsolatedLambda(maybeFlatMap.Cast().Lambda().Ref(), syncList, usedCluster, false)) {`
`35`	`35`	`return node;`
`36`	`36`	`}`
`37`	`37`	`list = maybeFlatMap.Cast().Input();`
`38`	`38`	`}`
`39`	`39`	`if (!IsYtProviderInput(list)) {`
`40`	`40`	`TSyncMap syncList;`
`41`		`- if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, true, false)) {`
	`41`	`+ if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, false)) {`
`42`	`42`	`return node;`
`43`	`43`	`}`
`44`	`44`	`continue;`
`@@ -84,7 +84,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node`
`84`	`84`	`if (auto maybeFlatMap = listStepForward.Maybe<TCoFlatMapBase>()) {`
`85`	`85`	`auto flatMap = maybeFlatMap.Cast();`
`86`	`86`	`if (IsLambdaSuitableForPullingIntoEquiJoin(flatMap, joinInput.Scope().Ref(), tableKeysMap, extractedMembers.Get())) {`
`87`		`- if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), worldList, usedCluster, true, false)) {`
	`87`	`+ if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), worldList, usedCluster, false)) {`
`88`	`88`	`return node;`
`89`	`89`	`}`
`90`	`90`
`@@ -134,7 +134,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::EquiJoin(TExprBase node`
`134`	`134`	`}`
`135`	`135`	`else {`
`136`	`136`	`TSyncMap syncList;`
`137`		`- if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, true, false)) {`
	`137`	`+ if (!IsYtCompleteIsolatedLambda(list.Ref(), syncList, usedCluster, false)) {`
`138`	`138`	`return node;`
`139`	`139`	`}`
`140`	`140`
Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::FlatMap(TExprBase node,`
`62`	`62`
`63`	`63`	`auto cluster = TString{GetClusterName(input)};`
`64`	`64`	`TSyncMap syncList;`
`65`		`- if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), syncList, cluster, true, false)) {`
	`65`	`+ if (!IsYtCompleteIsolatedLambda(flatMap.Lambda().Ref(), syncList, cluster, false)) {`
`66`	`66`	`return node;`
`67`	`67`	`}`
`68`	`68`
`@@ -150,7 +150,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::LMap(TExprBase node, TE`
`150`	`150`
`151`	`151`	`auto cluster = TString{GetClusterName(lmap.Input())};`
`152`	`152`	`TSyncMap syncList;`
`153`		`- if (!IsYtCompleteIsolatedLambda(lmap.Lambda().Ref(), syncList, cluster, true, false)) {`
	`153`	`+ if (!IsYtCompleteIsolatedLambda(lmap.Lambda().Ref(), syncList, cluster, false)) {`
`154`	`154`	`return node;`
`155`	`155`	`}`
`156`	`156`
`@@ -228,19 +228,19 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::CombineByKey(TExprBase`
`228`	`228`
`229`	`229`	`auto cluster = TString{GetClusterName(input)};`
`230`	`230`	`TSyncMap syncList;`
`231`		`- if (!IsYtCompleteIsolatedLambda(combineByKey.PreMapLambda().Ref(), syncList, cluster, true, false)) {`
	`231`	`+ if (!IsYtCompleteIsolatedLambda(combineByKey.PreMapLambda().Ref(), syncList, cluster, false)) {`
`232`	`232`	`return node;`
`233`	`233`	`}`
`234`		`- if (!IsYtCompleteIsolatedLambda(combineByKey.KeySelectorLambda().Ref(), syncList, cluster, true, false)) {`
	`234`	`+ if (!IsYtCompleteIsolatedLambda(combineByKey.KeySelectorLambda().Ref(), syncList, cluster, false)) {`
`235`	`235`	`return node;`
`236`	`236`	`}`
`237`		`- if (!IsYtCompleteIsolatedLambda(combineByKey.InitHandlerLambda().Ref(), syncList, cluster, true, false)) {`
	`237`	`+ if (!IsYtCompleteIsolatedLambda(combineByKey.InitHandlerLambda().Ref(), syncList, cluster, false)) {`
`238`	`238`	`return node;`
`239`	`239`	`}`
`240`		`- if (!IsYtCompleteIsolatedLambda(combineByKey.UpdateHandlerLambda().Ref(), syncList, cluster, true, false)) {`
	`240`	`+ if (!IsYtCompleteIsolatedLambda(combineByKey.UpdateHandlerLambda().Ref(), syncList, cluster, false)) {`
`241`	`241`	`return node;`
`242`	`242`	`}`
`243`		`- if (!IsYtCompleteIsolatedLambda(combineByKey.FinishHandlerLambda().Ref(), syncList, cluster, true, false)) {`
	`243`	`+ if (!IsYtCompleteIsolatedLambda(combineByKey.FinishHandlerLambda().Ref(), syncList, cluster, false)) {`
`244`	`244`	`return node;`
`245`	`245`	`}`
`246`	`246`
Original file line number	Diff line number	Diff line change
`@@ -244,7 +244,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Mux(TExprBase node, TEx`
`244`	`244`	`return node;`
`245`	`245`	`}`
`246`	`246`	`TSyncMap syncList;`
`247`		`- if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, true, false)) {`
	`247`	`+ if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, false)) {`
`248`	`248`	`return node;`
`249`	`249`	`}`
`250`	`250`
`@@ -319,7 +319,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::TakeOrSkip(TExprBase no`
`319`	`319`
`320`	`320`	`auto cluster = TString{GetClusterName(input)};`
`321`	`321`	`TSyncMap syncList;`
`322`		`- if (!IsYtCompleteIsolatedLambda(countBase.Count().Ref(), syncList, cluster, true, false)) {`
	`322`	`+ if (!IsYtCompleteIsolatedLambda(countBase.Count().Ref(), syncList, cluster, false)) {`
`323`	`323`	`return node;`
`324`	`324`	`}`
`325`	`325`
`@@ -558,7 +558,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Extend(TExprBase node,`
`558`	`558`	`return node;`
`559`	`559`	`}`
`560`	`560`	`TSyncMap syncList;`
`561`		`- if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, true, false)) {`
	`561`	`+ if (!IsYtCompleteIsolatedLambda(child.Ref(), syncList, resultCluster, false)) {`
`562`	`562`	`return node;`
`563`	`563`	`}`
`564`	`564`
Original file line number	Diff line number	Diff line change
`@@ -35,8 +35,8 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PartitionByKey(TExprBas`
`35`	`35`
`36`	`36`	`auto cluster = TString{GetClusterName(input)};`
`37`	`37`	`TSyncMap syncList;`
`38`		`- if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, true, false)`
`39`		`- \|\| !IsYtCompleteIsolatedLambda(handlerLambda.Ref(), syncList, cluster, true, false)) {`
	`38`	`+ if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, false)`
	`39`	`+ \|\| !IsYtCompleteIsolatedLambda(handlerLambda.Ref(), syncList, cluster, false)) {`
`40`	`40`	`return node;`
`41`	`41`	`}`
`42`	`42`
`@@ -94,7 +94,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PartitionByKey(TExprBas`
`94`	`94`	`}`
`95`	`95`
`96`	`96`	`TCoLambda sortKeySelectorLambda = partByKey.SortKeySelectorLambda().Cast<TCoLambda>();`
`97`		`- if (!IsYtCompleteIsolatedLambda(sortKeySelectorLambda.Ref(), syncList, cluster, true, false)) {`
	`97`	`+ if (!IsYtCompleteIsolatedLambda(sortKeySelectorLambda.Ref(), syncList, cluster, false)) {`
`98`	`98`	`return node;`
`99`	`99`	`}`
`100`	`100`
Original file line number	Diff line number	Diff line change
`@@ -162,7 +162,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::PushDownKeyExtract(TExp`
`162`	`162`
`163`	`163`	`TSyncMap syncList;`
`164`	`164`	`for (auto filter: keyFilters) {`
`165`		`- if (!IsYtCompleteIsolatedLambda(*filter, syncList, true, false)) {`
	`165`	`+ if (!IsYtCompleteIsolatedLambda(*filter, syncList, false)) {`
`166`	`166`	`return node;`
`167`	`167`	`}`
`168`	`168`	`}`
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Sort(TExprBase node, TE`
`73`	`73`	`auto keySelectorLambda = sort.KeySelectorLambda();`
`74`	`74`	`auto cluster = TString{GetClusterName(sort.Input())};`
`75`	`75`	`TSyncMap syncList;`
`76`		`- if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, true, false)) {`
	`76`	`+ if (!IsYtCompleteIsolatedLambda(keySelectorLambda.Ref(), syncList, cluster, false)) {`
`77`	`77`	`return node;`
`78`	`78`	`}`
`79`	`79`
Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::DqWrite(TExprBase node,`
`43`	`43`	`}`
`44`	`44`
`45`	`45`	`TSyncMap syncList;`
`46`		`- if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, true, true)) {`
	`46`	`+ if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, true)) {`
`47`	`47`	`return node;`
`48`	`48`	`}`
`49`	`49`
`@@ -554,7 +554,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::ReplaceStatWriteTable(T`
`554`	`554`
`555`	`555`	`TString cluster;`
`556`	`556`	`TSyncMap syncList;`
`557`		`- if (!IsYtCompleteIsolatedLambda(input.Ref(), syncList, cluster, true, false)) {`
	`557`	`+ if (!IsYtCompleteIsolatedLambda(input.Ref(), syncList, cluster, false)) {`
`558`	`558`	`return node;`
`559`	`559`	`}`
`560`	`560`
`@@ -710,7 +710,7 @@ TMaybeNode<TExprBase> TYtPhysicalOptProposalTransformer::Fill(TExprBase node, TE`
`710`	`710`
`711`	`711`	`auto cluster = TString{write.DataSink().Cluster().Value()};`
`712`	`712`	`TSyncMap syncList;`
`713`		`- if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, cluster, true, false)) {`
	`713`	`+ if (!IsYtCompleteIsolatedLambda(write.Content().Ref(), syncList, cluster, false)) {`
`714`	`714`	`return node;`
`715`	`715`	`}`
`716`	`716`
Original file line number	Diff line number	Diff line change
`@@ -270,7 +270,7 @@ class TYtDataSource : public TDataProviderBase {`
`270`	`270`
`271`	`271`	`bool CanBuildResult(const TExprNode& node, TSyncMap& syncList) override {`
`272`	`272`	`TString usedCluster;`
`273`		`- return IsYtCompleteIsolatedLambda(node, syncList, usedCluster, true, false);`
	`273`	`+ return IsYtCompleteIsolatedLambda(node, syncList, usedCluster, false);`
`274`	`274`	`}`
`275`	`275`
`276`	`276`	`bool GetExecWorld(const TExprNode::TPtr& node, TExprNode::TPtr& root) override {`
Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ class TYtDataSourceExecTransformer : public TExecTransformerBase {`
`114`	`114`
`115`	`115`	`TString usedCluster;`
`116`	`116`	`TSyncMap syncList;`
`117`		`- if (!IsYtIsolatedLambda(data.Ref(), syncList, usedCluster, true, false)) {`
	`117`	`+ if (!IsYtIsolatedLambda(data.Ref(), syncList, usedCluster, false)) {`
`118`	`118`	`ctx.AddError(TIssue(ctx.GetPosition(data.Pos()), TStringBuilder() << "Failed to execute node due to bad graph: " << input->Content()));`
`119`	`119`	`return SyncError();`
`120`	`120`	`}`
Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ class TYtDqOptimizers: public TDqOptimizationBase {`
`97`	`97`	`}`
`98`	`98`
`99`	`99`	`TSyncMap syncList;`
`100`		`- if (!IsYtCompleteIsolatedLambda(count.Count().Ref(), syncList, cluster, true, false)) {`
	`100`	`+ if (!IsYtCompleteIsolatedLambda(count.Count().Ref(), syncList, cluster, false)) {`
`101`	`101`	`return read;`
`102`	`102`	`}`
`103`	`103`