add disable rule when calc relations

apache · Jan 8, 2024 · 1ee2e7a · 1ee2e7a
1 parent 3de64e2
commit 1ee2e7a
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/job/extensions/mtmv/MTMVTask.java b/fe/fe-core/src/main/java/org/apache/doris/job/extensions/mtmv/MTMVTask.java
@@ -39,9 +39,11 @@
 import org.apache.doris.mtmv.MTMVUtil;
 import org.apache.doris.nereids.glue.LogicalPlanAdapter;
 import org.apache.doris.nereids.trees.plans.commands.UpdateMvByPartitionCommand;
+import org.apache.doris.nereids.trees.plans.commands.info.CreateMTMVInfo;
 import org.apache.doris.nereids.trees.plans.commands.info.TableNameInfo;
 import org.apache.doris.qe.ConnectContext;
 import org.apache.doris.qe.QueryState.MysqlStateType;
+import org.apache.doris.qe.SessionVariable;
 import org.apache.doris.qe.StmtExecutor;
 import org.apache.doris.thrift.TCell;
 import org.apache.doris.thrift.TRow;
@@ -144,7 +146,18 @@ public void run() throws JobException {
             ConnectContext ctx = MTMVPlanUtil.createMTMVContext(mtmv);
             // Every time a task is run, the relation is regenerated because baseTables and baseViews may change,
             // such as deleting a table and creating a view with the same name
-            this.relation = MTMVPlanUtil.generateMTMVRelation(mtmv, ctx);
+            // Should not make table without data to empty relation when analyze the related table,
+            // so add disable rules
+            SessionVariable sessionVariable = ctx.getSessionVariable();
+            Set<String> tempDisableRules = sessionVariable.getDisableNereidsRuleNames();
+            sessionVariable.setDisableNereidsRules(CreateMTMVInfo.MTMV_PLANER_DISABLE_RULES);
+            ctx.getStatementContext().invalidCache(SessionVariable.DISABLE_NEREIDS_RULES);
+            try {
+                this.relation = MTMVPlanUtil.generateMTMVRelation(mtmv, ctx);
+            } finally {
+                sessionVariable.setDisableNereidsRules(String.join(",", tempDisableRules));
+                ctx.getStatementContext().invalidCache(SessionVariable.DISABLE_NEREIDS_RULES);
+            }
             List<Long> needRefreshPartitionIds = calculateNeedRefreshPartitions();
             this.needRefreshPartitions = MTMVUtil.getPartitionNamesByIds(mtmv, needRefreshPartitionIds);
             this.refreshMode = generateRefreshMode(needRefreshPartitionIds);