Merge pull request #1273 from RumbleDB/ExecutionModeWarning

Remove execution mode warning.
RumbleDB · Oct 28, 2024 · 1f52989 · 1f52989
2 parents 90a7faa + abd0770
commit 1f52989
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/org/rumbledb/compiler/ExecutionModeVisitor.java b/src/main/java/org/rumbledb/compiler/ExecutionModeVisitor.java
@@ -656,7 +656,11 @@ public StaticContext visitProlog(Prolog prolog, StaticContext argument) {
     @Override
     public StaticContext visitProgram(Program program, StaticContext argument) {
         visitDescendants(program, argument);
-        ExecutionMode mergedExecutionMode = getHighestExecutionModeFromStatements(exitStatementChildren);
+        ExecutionMode mergedExecutionMode = program.getStatementsAndOptionalExpr().getHighestExecutionMode();
+        for (Statement statement : exitStatementChildren) {
+            ExecutionMode statementExecMode = statement.getHighestExecutionMode(this.visitorConfig);
+            mergedExecutionMode = getHighestExecutionMode(mergedExecutionMode, statementExecMode);
+        }
         program.setHighestExecutionMode(mergedExecutionMode);
         return argument;
     }