Reuse SqlRoutine interruption variables

trinodb · May 17, 2024 · 279d9d3 · 279d9d3
1 parent dac494c
commit 279d9d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/trino-main/src/main/java/io/trino/sql/routine/SqlRoutineCompiler.java b/core/trino-main/src/main/java/io/trino/sql/routine/SqlRoutineCompiler.java
@@ -424,7 +424,7 @@ private BytecodeNode compileLoop(Scope scope, Optional<IrLabel> label, Function<
         {
             BytecodeBlock block = new BytecodeBlock();
 
-            Variable interruption = scope.createTempVariable(int.class);
+            Variable interruption = scope.getOrCreateTempVariable(int.class);
             block.putVariable(interruption, 0);
 
             BytecodeBlock interruptionBlock = new BytecodeBlock()
@@ -450,6 +450,8 @@ private BytecodeNode compileLoop(Scope scope, Optional<IrLabel> label, Function<
                 block.visitLabel(breakLabel);
             }
 
+            scope.releaseTempVariableForReuse(interruption);
+
             return block;
         }