Add support for the method $.chain(iterable, size).

Author: javadev

lodash-plugin/src/main/java/com/github/underscore/lodash/$.java

@@ -632,6 +632,10 @@ public static <T> Chain<T> chain(final Iterable<T> iterable) {
         return new $.Chain<T>(newArrayList(iterable));
     }
 
+    public static <T> Chain<T> chain(final Iterable<T> iterable, int size) {
+        return new $.Chain<T>(newArrayList(iterable, size));
+    }
+
     public static <T> Chain<T> chain(final T ... list) {
         return new $.Chain<T>(Arrays.asList(list));
     }

lodash-plugin/src/test/java/com/github/underscore/lodash/LodashTest.java

@@ -575,6 +575,7 @@ public void main() {
         new $("");
         new $(asList()).chain();
         $.chain(new ArrayList<String>());
+        $.chain(new ArrayList<String>(), 1);
         $.chain(new HashSet<String>());
         $.chain(new String[] {});
         $.chain("");

math-plugin/src/main/java/com/github/underscore/math/$.java

@@ -362,6 +362,10 @@ public static <T> Chain<T> chain(final Iterable<T> iterable) {
         return new $.Chain<T>(newArrayList(iterable));
     }
 
+    public static <T> Chain<T> chain(final Iterable<T> iterable, int size) {
+        return new $.Chain<T>(newArrayList(iterable, size));
+    }
+
     public static <T> Chain<T> chain(final T ... array) {
         return new $.Chain<T>(Arrays.asList(array));
     }

math-plugin/src/test/java/com/github/underscore/math/MathTest.java

@@ -235,6 +235,7 @@ public void main() {
         $.main(new String[] {});
         new $("");
         new $(asList()).chain();
+        $.chain(new ArrayList<String>(), 1);
         $.chain(new HashSet<String>());
         $.chain(new String[] {});
     }

src/main/java/com/github/underscore/$.java

@@ -1887,6 +1887,10 @@ public static <T> Chain<T> chain(final Iterable<T> iterable) {
         return new $.Chain<T>(newArrayList(iterable));
     }
 
+    public static <T> Chain<T> chain(final Iterable<T> iterable, int size) {
+        return new $.Chain<T>(newArrayList(iterable, size));
+    }
+
     @SuppressWarnings("unchecked")
     public static <T> Chain<T> chain(final T ... array) {
         return new $.Chain<T>(Arrays.asList(array));
@@ -2567,7 +2571,7 @@ protected static <T> List<T> newArrayList() {
     }
 
     @SuppressWarnings("unchecked")
-    protected static <T> List<T> newArrayList(Iterable<T> iterable) {
+    protected static <T> List<T> newArrayList(final Iterable<T> iterable) {
         final List<T> result;
         if (iterable instanceof Collection) {
             result = new ArrayList<T>((Collection) iterable);
@@ -2580,6 +2584,14 @@ protected static <T> List<T> newArrayList(Iterable<T> iterable) {
         return result;
     }
 
+    protected static <T> List<T> newArrayList(final Iterable<T> iterable, final int size) {
+        final List<T> result = new ArrayList<T>();
+        for (int index = 0; iterable.iterator().hasNext() && index < size; index += 1) {
+            result.add(iterable.iterator().next());
+        }
+        return result;
+    }
+
     protected static List<Integer> newIntegerList(int ... array) {
         final List<Integer> result = new ArrayList<Integer>(array.length);
         for (final int item : array) {

src/test/java/com/github/underscore/UnderscoreTest.java

@@ -299,6 +299,34 @@ public long[] apply(long[] arg) {
         assertArrayEquals(new long[] {1, 2}, iterable.iterator().next());
     }
 
+    @Test
+    public void iterateChain() {
+        Iterable<long[]> iterable = $.<long[]>iterate(new long[] {1, 1}, new UnaryOperator<long[]>() {
+                public long[] apply(long[] arg) {
+                    return new long[] {arg[1], arg[0] + arg[1]};
+                }
+            });
+        assertEquals(1L, $.chain(iterable, 5).first().item()[0]);
+        class MyIterable<T> implements Iterable<T> {
+            public Iterator<T> iterator() {
+                return new Iterator<T>() {
+                    @Override
+                    public boolean hasNext() {
+                        return false;
+                    }
+                    @Override
+                    public T next() {
+                        return null;
+                    }
+                    @Override
+                    public void remove() {
+                    }
+                };
+            }
+        }
+        assertTrue($.chain(new MyIterable<Integer>(), 5).isEmpty());
+    }
+
     @Test
     public void optional() {
         assertTrue(Optional.absent().equals(Optional.absent()));

string-plugin/src/main/java/com/github/underscore/string/$.java

@@ -519,6 +519,10 @@ public static <T> Chain<T> chain(final Iterable<T> iterable) {
         return new $.Chain<T>(newArrayList(iterable));
     }
 
+    public static <T> Chain<T> chain(final Iterable<T> iterable, int size) {
+        return new $.Chain<T>(newArrayList(iterable, size));
+    }
+
     @SuppressWarnings("unchecked")
     public static <T> Chain<T> chain(final T ... list) {
         return new $.Chain<T>(Arrays.asList(list));

string-plugin/src/test/java/com/github/underscore/string/StringTest.java

@@ -1950,6 +1950,7 @@ public void main() throws Exception {
         new $.JsonJavaValue();
         new $.JsonJavaObject();
         $.chain(new ArrayList<String>());
+        $.chain(new ArrayList<String>(), 1);
         $.chain(new HashSet<String>());
         $.chain(new String[] {});
     }