From 59d12315c70c0bf516d65004cd0faff85780c315 Mon Sep 17 00:00:00 2001
From: Mikhail Cheshkov <mcheshkov@gmail.com>
Date: Wed, 30 Oct 2024 13:57:29 +0200
Subject: [PATCH] refactor(cubesql): Tighten rule for non-trivial wrapper
 pullup

---
 .../src/compile/rewrite/rules/wrapper/wrapper_pull_up.rs     | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/rust/cubesql/cubesql/src/compile/rewrite/rules/wrapper/wrapper_pull_up.rs b/rust/cubesql/cubesql/src/compile/rewrite/rules/wrapper/wrapper_pull_up.rs
index b90bc4604ad0b..be7c228d321b7 100644
--- a/rust/cubesql/cubesql/src/compile/rewrite/rules/wrapper/wrapper_pull_up.rs
+++ b/rust/cubesql/cubesql/src/compile/rewrite/rules/wrapper/wrapper_pull_up.rs
@@ -204,7 +204,8 @@ impl WrapperRules {
                         ),
                         "?select_alias",
                         "?select_distinct",
-                        "?select_push_to_cube",
+                        // This node has a WrappedSelect in from, so it's not allowed to use push to Cube
+                        "WrappedSelectPushToCube:false",
                         "?select_ungrouped_scan",
                     ),
                     "CubeScanWrapperFinalized:false",
@@ -245,7 +246,7 @@ impl WrapperRules {
                             "?order_expr",
                             "?select_alias",
                             "?select_distinct",
-                            "?select_push_to_cube",
+                            "WrappedSelectPushToCube:false",
                             "?select_ungrouped_scan",
                         ),
                         "?alias_to_cube",