Implement Range#to_a for RubyLongRange

oracle · Nov 27, 2020 · 83a9e9f · 83a9e9f
1 parent 3483fb7
commit 83a9e9f
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/main/java/org/truffleruby/core/range/RangeNodes.java b/src/main/java/org/truffleruby/core/range/RangeNodes.java
@@ -354,6 +354,30 @@ protected RubyArray toA(RubyIntRange range) {
             }
         }
 
+        @Specialization
+        protected RubyArray longtoA(RubyLongRange range) {
+            final long begin = range.begin;
+            long result;
+            if (range.excludedEnd) {
+                result = range.end;
+            } else {
+                result = range.end + 1;
+            }
+            final int length = (int) (result - begin);
+
+            if (length < 0) {
+                return createEmptyArray();
+            } else {
+                final long[] values = new long[length];
+
+                for (int n = 0; n < length; n++) {
+                    values[n] = begin + n;
+                }
+
+                return createArray(values);
+            }
+        }
+
         @Specialization(guards = "range.isBounded()")
         protected Object boundedToA(RubyObjectRange range) {
             if (toAInternalCall == null) {