[GR-19220] Truffle::Graal.bailout (#1808).

PullRequest: truffleruby/1141

Author: eregon

spec/truffle/graal/assert_constant_spec.rb

@@ -14,7 +14,7 @@
     -> { Truffle::Graal.send(:assert_constant, 14 + 2) }.should raise_error(RuntimeError)
   end
 
-  unless TruffleRuby.jit?
+  guard -> { !TruffleRuby.jit? } do
     it "returns the value of the argument" do
       Truffle::Graal.assert_constant(14 + 2).should == 16
     end

spec/truffle/graal/assert_not_compiled_spec.rb

@@ -14,7 +14,7 @@
     -> { Truffle::Graal.send(:assert_not_compiled) }.should raise_error(RuntimeError)
   end
 
-  unless TruffleRuby.jit?
+  guard -> { !TruffleRuby.jit? } do
     it "returns nil" do
       Truffle::Graal.assert_not_compiled.should be_nil
     end

spec/truffle/graal/bailout_spec.rb

@@ -0,0 +1,23 @@
+# Copyright (c) 2019 Oracle and/or its affiliates. All rights reserved. This
+# code is released under a tri EPL/GPL/LGPL license. You can use it,
+# redistribute it and/or modify it under the terms of the:
+#
+# Eclipse Public License version 2.0, or
+# GNU General Public License version 2, or
+# GNU Lesser General Public License version 2.1.
+
+require_relative '../../ruby/spec_helper'
+
+describe "Truffle::Graal.bailout" do
+
+  it "raises a RuntimeError when called dynamically" do
+    -> { Truffle::Graal.send(:bailout, "message") }.should raise_error(RuntimeError)
+  end
+
+  guard -> { !TruffleRuby.jit? } do
+    it "returns nil" do
+      Truffle::Graal.bailout("message").should be_nil
+    end
+  end
+
+end

src/main/java/org/truffleruby/core/exception/CoreExceptions.java

@@ -251,6 +251,10 @@ public DynamicObject runtimeErrorCompiled(Node currentNode) {
         return runtimeError("Truffle::Graal.assert_not_compiled can only be called lexically", currentNode);
     }
 
+    public DynamicObject runtimeErrorBailout(Node currentNode) {
+        return runtimeError("Truffle::Graal.bailout can only be called lexically", currentNode);
+    }
+
     public DynamicObject runtimeErrorCoverageNotEnabled(Node currentNode) {
         return runtimeError("coverage measurement is not enabled", currentNode);
     }

src/main/java/org/truffleruby/extra/TruffleGraalNodes.java

@@ -37,6 +37,7 @@ public abstract class TruffleGraalNodes {
     @CoreMethod(names = "assert_constant", onSingleton = true, required = 1)
     public abstract static class AssertConstantNode extends CoreMethodArrayArgumentsNode {
 
+        @TruffleBoundary
         @Specialization
         protected DynamicObject assertConstant(Object value) {
             throw new RaiseException(getContext(), coreExceptions().runtimeErrorNotConstant(this));
@@ -47,13 +48,25 @@ protected DynamicObject assertConstant(Object value) {
     @CoreMethod(names = "assert_not_compiled", onSingleton = true)
     public abstract static class AssertNotCompiledNode extends CoreMethodArrayArgumentsNode {
 
+        @TruffleBoundary
         @Specialization
         protected DynamicObject assertNotCompiled() {
             throw new RaiseException(getContext(), coreExceptions().runtimeErrorCompiled(this));
         }
 
     }
 
+    @CoreMethod(names = "bailout", onSingleton = true, required = 1)
+    public abstract static class BailoutNode extends CoreMethodArrayArgumentsNode {
+
+        @TruffleBoundary
+        @Specialization(guards = "isRubyString(message)")
+        protected DynamicObject bailout(DynamicObject message) {
+            throw new RaiseException(getContext(), coreExceptions().runtimeErrorBailout(this));
+        }
+
+    }
+
     @CoreMethod(names = "always_split", onSingleton = true, required = 1, argumentNames = "method_or_proc")
     public abstract static class AlwaysSplitNode extends CoreMethodArrayArgumentsNode {
 

src/main/java/org/truffleruby/parser/BodyTranslator.java

@@ -254,6 +254,7 @@
 import org.truffleruby.parser.parser.ParserSupport;
 import org.truffleruby.parser.scope.StaticScope;
 import org.truffleruby.platform.AssertConstantNodeGen;
+import org.truffleruby.platform.BailoutNode;
 import org.truffleruby.platform.AssertNotCompiledNodeGen;
 
 import com.oracle.truffle.api.RootCallTarget;
@@ -549,6 +550,10 @@ public RubyNode visitCallNode(CallParseNode node) {
                 final RubyNode ret = AssertNotCompiledNodeGen.create();
                 ret.unsafeSetSourceSection(sourceSection);
                 return addNewlineIfNeeded(node, ret);
+            } else if (methodName.equals("bailout")) {
+                final RubyNode ret = BailoutNode.create(((ArrayParseNode) node.getArgsNode()).get(0).accept(this));
+                ret.unsafeSetSourceSection(sourceSection);
+                return addNewlineIfNeeded(node, ret);
             }
         } else if (receiver instanceof VCallParseNode // undefined.equal?(obj)
                 && ((VCallParseNode) receiver).getName().equals("undefined") && inCore() &&

src/main/java/org/truffleruby/platform/BailoutNode.java

@@ -0,0 +1,34 @@
+/*
+ * Copyright (c) 2019 Oracle and/or its affiliates. All rights reserved. This
+ * code is released under a tri EPL/GPL/LGPL license. You can use it,
+ * redistribute it and/or modify it under the terms of the:
+ *
+ * Eclipse Public License version 2.0, or
+ * GNU General Public License version 2, or
+ * GNU Lesser General Public License version 2.1.
+ */
+package org.truffleruby.platform;
+
+import com.oracle.truffle.api.CompilerDirectives;
+import com.oracle.truffle.api.dsl.Cached;
+import com.oracle.truffle.api.dsl.NodeChild;
+import com.oracle.truffle.api.dsl.Specialization;
+import com.oracle.truffle.api.object.DynamicObject;
+import org.truffleruby.interop.ToJavaStringNode;
+import org.truffleruby.language.RubyNode;
+
+@NodeChild
+public abstract class BailoutNode extends RubyNode {
+
+    @Specialization(guards = "isRubyString(message)")
+    protected DynamicObject bailout(DynamicObject message,
+            @Cached ToJavaStringNode toJavaStringNode) {
+        CompilerDirectives.bailout(toJavaStringNode.executeToJavaString(message));
+        return nil();
+    }
+
+    public static BailoutNode create(RubyNode messageNode) {
+        return BailoutNodeGen.create(messageNode);
+    }
+
+}