fix nereids mv plan

Author: Hastyshell

fe/fe-core/src/main/java/org/apache/doris/nereids/util/RelationUtil.java

@@ -35,6 +35,7 @@
 import org.apache.doris.nereids.trees.plans.logical.LogicalProject;
 import org.apache.doris.qe.ConnectContext;
 import org.apache.doris.qe.OriginStatement;
+import org.apache.doris.qe.SessionVariable;
 
 import com.google.common.base.Strings;
 import com.google.common.collect.ImmutableList;
@@ -50,6 +51,13 @@
  * relation util
  */
 public class RelationUtil {
+    private static final String SYNC_MV_PLANER_DISABLE_RULES = "OLAP_SCAN_PARTITION_PRUNE, PRUNE_EMPTY_PARTITION, "
+            + "ELIMINATE_GROUP_BY_KEY_BY_UNIFORM, HAVING_TO_FILTER, ELIMINATE_GROUP_BY, SIMPLIFY_AGG_GROUP_BY, "
+            + "MERGE_PERCENTILE_TO_ARRAY, VARIANT_SUB_PATH_PRUNING, INFER_PREDICATES, INFER_AGG_NOT_NULL, "
+            + "INFER_SET_OPERATOR_DISTINCT, INFER_FILTER_NOT_NULL, INFER_JOIN_NOT_NULL, MAX_MIN_FILTER_PUSH_DOWN, "
+            + "ELIMINATE_SORT, ELIMINATE_AGGREGATE, ELIMINATE_LIMIT, ELIMINATE_SEMI_JOIN, ELIMINATE_NOT_NULL, "
+            + "ELIMINATE_JOIN_BY_UK, ELIMINATE_JOIN_BY_FK, ELIMINATE_GROUP_BY_KEY, ELIMINATE_GROUP_BY_KEY_BY_UNIFORM, "
+            + "ELIMINATE_FILTER_GROUP_BY_KEY";
 
     /**
      * get table qualifier
@@ -138,15 +146,29 @@ public static Set<String> getMvUsedColumnNames(MaterializedIndexMeta meta) {
             Optional<String> querySql = new NereidsParser().parseForSyncMv(createMvSql);
             if (querySql.isPresent()) {
                 LogicalPlan unboundMvPlan = new NereidsParser().parseSingle(querySql.get());
-                StatementContext statementContext = new StatementContext(ConnectContext.get(),
+                ConnectContext connectContext = ConnectContext.get();
+                StatementContext statementContext = new StatementContext(connectContext,
                         new OriginStatement(querySql.get(), 0));
                 NereidsPlanner planner = new NereidsPlanner(statementContext);
                 if (statementContext.getConnectContext().getStatementContext() == null) {
                     statementContext.getConnectContext().setStatementContext(statementContext);
                 }
-                planner.planWithLock(unboundMvPlan, PhysicalProperties.ANY, ExplainCommand.ExplainLevel.REWRITTEN_PLAN);
-                LogicalPlan logicalPlan = (LogicalPlan) planner.getCascadesContext().getRewritePlan();
-
+                Set<String> tempDisableRules = connectContext.getSessionVariable().getDisableNereidsRuleNames();
+                connectContext.getSessionVariable().setDisableNereidsRules(SYNC_MV_PLANER_DISABLE_RULES);
+                connectContext.getStatementContext().invalidCache(SessionVariable.DISABLE_NEREIDS_RULES);
+                LogicalPlan logicalPlan;
+                try {
+                    // disable rbo sync mv rewrite
+                    connectContext.getSessionVariable().setVarOnce(SessionVariable.ENABLE_SYNC_MV_COST_BASED_REWRITE, "true");
+                    // disable constant fold
+                    connectContext.getSessionVariable().setVarOnce(SessionVariable.DEBUG_SKIP_FOLD_CONSTANT, "true");
+                    planner.planWithLock(unboundMvPlan, PhysicalProperties.ANY, ExplainCommand.ExplainLevel.REWRITTEN_PLAN);
+                    logicalPlan = (LogicalPlan) planner.getCascadesContext().getRewritePlan();
+                } finally {
+                    // after operate, roll back the disable rules
+                    connectContext.getSessionVariable().setDisableNereidsRules(String.join(",", tempDisableRules));
+                    connectContext.getStatementContext().invalidCache(SessionVariable.DISABLE_NEREIDS_RULES);
+                }
                 Map<Boolean, List<Object>> partitionedPlan = logicalPlan
                         .collect(plan -> true)
                         .stream()