Handle repeats from inlined ExpandApply calls

init
commit_hash:c6373ff2a9e9562c16afe8d56cbd495790545274

1 files changed, 10 insertions, 0 deletions

diff --git a/yql/essentials/core/yql_expr_optimize.cpp b/yql/essentials/core/yql_expr_optimize.cpp
index 91608eed80..24716a1668 100644
--- a/yql/essentials/core/yql_expr_optimize.cpp
+++ b/yql/essentials/core/yql_expr_optimize.cpp
@@ -862,6 +862,16 @@ IGraphTransformer::TStatus ExpandApply(const TExprNode::TPtr& input, TExprNode::
     return ret;
 }
 
+IGraphTransformer::TStatus ExpandApplyNoRepeat(const TExprNode::TPtr& input, TExprNode::TPtr& output, TExprContext& ctx) {
+    output = input;
+    for (;;) {
+        auto status = ExpandApply(output, output, ctx);
+        if (status.Level != IGraphTransformer::TStatus::Repeat) {
+            return status;
+        }
+    }
+}
+
 TExprNode::TPtr ApplySyncListToWorld(const TExprNode::TPtr& main, const TSyncMap& syncList, TExprContext& ctx) {
     if (syncList.empty()) {
         return main;