WIP - reuse thread-local jumps

headius · Dec 5, 2023 · c4c331b · c4c331b
1 parent 8e20695
commit c4c331b
Showing 1 changed file with 20 additions and 7 deletions.
diff --git a/core/src/main/java/org/jruby/ir/runtime/IRReturnJump.java b/core/src/main/java/org/jruby/ir/runtime/IRReturnJump.java
@@ -6,18 +6,31 @@
 import org.jruby.runtime.DynamicScope;
 
 public class IRReturnJump extends IRJump implements Unrescuable {
-    public final StaticScope returnScope;
-    public final DynamicScope methodToReturnFrom;
-    public final Object returnValue;
+    public StaticScope returnScope;
+    public DynamicScope methodToReturnFrom;
+    public Object returnValue;
+
+    private static final ThreadLocal<IRReturnJump> RETURN_JUMP = new ThreadLocal<>();
 
     private IRReturnJump(StaticScope returnScope, DynamicScope scopeToReturnFrom, Object rv) {
-        this.methodToReturnFrom = scopeToReturnFrom;
-        this.returnScope = returnScope;
-        this.returnValue = rv;
+        reset(returnScope, scopeToReturnFrom, rv);
     }
 
     public static IRReturnJump create(StaticScope returnScope, DynamicScope scopeToReturnFrom, Object rv) {
-        return new IRReturnJump(returnScope, scopeToReturnFrom, rv);
+        IRReturnJump jump = RETURN_JUMP.get();
+        if (jump == null) {
+            RETURN_JUMP.set(jump = new IRReturnJump(returnScope, scopeToReturnFrom, rv));
+        } else {
+            jump.reset(returnScope, scopeToReturnFrom, rv);
+        }
+
+        return jump;
+    }
+
+    private void reset(StaticScope returnScope, DynamicScope scopeToReturnFrom, Object rv) {
+        this.methodToReturnFrom = scopeToReturnFrom;
+        this.returnScope = returnScope;
+        this.returnValue = rv;
     }
 
     public boolean isReturnToScope(DynamicScope scope) {