[Truffle] If rand limit is an int, don't return a long.

jruby · Sep 28, 2015 · 17ff676 · 17ff676 · eregon · Sep 28, 2015
1 parent 48a4e24
commit 17ff676
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/RandomizerPrimitiveNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/rubinius/RandomizerPrimitiveNodes.java
@@ -92,6 +92,12 @@ public RandomizerRandIntPrimitiveNode(RubyContext context, SourceSection sourceS
             super(context, sourceSection);
         }
 
+        @Specialization
+        public int randomizerRandInt(DynamicObject randomizer, int limit) {
+            final Random r = Layouts.RANDOMIZER.getRandom(randomizer);
+            return (int) putIntoRange(r, (long) limit);
+        }
+
         @Specialization
         public long randomizerRandInt(DynamicObject randomizer, long limit) {
             final Random r = Layouts.RANDOMIZER.getRandom(randomizer);