Reuse temp variables in SwitchCodeGenerator

trinodb · May 17, 2024 · dac494c · dac494c
1 parent 34b8303
commit dac494c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/trino-main/src/main/java/io/trino/sql/gen/SwitchCodeGenerator.java b/core/trino-main/src/main/java/io/trino/sql/gen/SwitchCodeGenerator.java
@@ -126,7 +126,7 @@ else if (<value> == <non-constant2>) {
 
         // evaluate the value and store it in a variable
         LabelNode nullValue = new LabelNode("nullCondition");
-        Variable tempVariable = scope.createTempVariable(valueType);
+        Variable tempVariable = scope.getOrCreateTempVariable(valueType);
         BytecodeBlock block = new BytecodeBlock()
                 .append(valueBytecode)
                 .append(BytecodeUtils.ifWasNullClearPopAndGoto(scope, nullValue, void.class, valueType))
@@ -163,6 +163,8 @@ else if (<value> == <non-constant2>) {
                     .ifFalse(elseValue);
         }
 
-        return block.append(elseValue);
+        block.append(elseValue);
+        scope.releaseTempVariableForReuse(tempVariable);
+        return block;
     }
 }