Replace Condense on SqueezeToList

1 files changed, 3 insertions, 19 deletions

diff --git a/ydb/core/kqp/opt/physical/effects/kqp_opt_phy_effects.cpp b/ydb/core/kqp/opt/physical/effects/kqp_opt_phy_effects.cpp
index eb31d985fb0..b86b07530c1 100644
--- a/ydb/core/kqp/opt/physical/effects/kqp_opt_phy_effects.cpp
+++ b/ydb/core/kqp/opt/physical/effects/kqp_opt_phy_effects.cpp
@@ -102,28 +102,12 @@ TMaybe<TCondenseInputResult> CondenseInput(const TExprBase& input, TExprContext&
     stageInputs.push_back(input);
     stageArguments.push_back(arg);
 
-    auto condense = Build<TCoCondense>(ctx, input.Pos())
-        .Input(arg)
-        .State<TCoList>()
-            .ListType<TCoTypeOf>()
-                .Value(input)
-                .Build()
-            .Build()
-        .SwitchHandler()
-            .Args({"item", "state"})
-            .Body(MakeBool<false>(input.Pos(), ctx))
-            .Build()
-        .UpdateHandler()
-            .Args({"item", "state"})
-            .Body<TCoAppend>()
-                .List("state")
-                .Item("item")
-                .Build()
-            .Build()
+    auto squeeze = Build<TCoSqueezeToList>(ctx, input.Pos())
+        .Stream(arg)
         .Done();
 
     return TCondenseInputResult {
-        .Stream = condense,
+        .Stream = squeeze,
         .StageInputs = stageInputs,
         .StageArgs = stageArguments
     };