Add simple aritmetic methods to the Bar class

Pere Urbon-Bayes · Pere Urbon-Bayes · commit d44fd20e60a3 · 2015-08-27T10:00:01.000+02:00
diff --git a/extensions/basic/basic.rb b/extensions/basic/basic.rb
@@ -16,6 +16,8 @@
 puts bar.say
 puts bar.shout
 
+puts bar.add(4, 4)
+
 ##
 # Using a module defined in the extensione.
 # Foo is a module defined in jruby extension point.
diff --git a/extensions/basic/jruby-ext/src/main/java/com/purbon/Bar.java b/extensions/basic/jruby-ext/src/main/java/com/purbon/Bar.java
@@ -1,9 +1,6 @@
 package com.purbon;
 
-import org.jruby.Ruby;
-import org.jruby.RubyClass;
-import org.jruby.RubyObject;
-import org.jruby.RubyString;
+import org.jruby.*;
 import org.jruby.anno.JRubyClass;
 import org.jruby.anno.JRubyMethod;
 import org.jruby.runtime.ThreadContext;
@@ -26,4 +23,24 @@ public static RubyString shout(ThreadContext context, IRubyObject self) {
          return context.runtime.newString("Hello World!");
     }
 
+    @JRubyMethod(module = true, name = "add")
+    public IRubyObject add(ThreadContext context, IRubyObject a, IRubyObject b) {
+        return a.callMethod(context, "+", b);
+    }
+
+    @JRubyMethod(module = true, name = "sub")
+    public IRubyObject sub(ThreadContext context, IRubyObject a, IRubyObject b) {
+        return a.callMethod(context, "-", b);
+    }
+
+    @JRubyMethod(module = true, name = "div")
+    public IRubyObject div(ThreadContext context, IRubyObject a, IRubyObject b) {
+        return a.callMethod(context, "/", b);
+    }
+
+    @JRubyMethod(module = true, name = "plus")
+    public IRubyObject plus(ThreadContext context, IRubyObject a, IRubyObject b) {
+        return a.callMethod(context, "*", b);
+    }
+
 }
