Add U.countBy(iterator) method.

Author: javadev

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

@@ -1020,11 +1020,28 @@ public static <K, E> Map<K, Integer> countBy(final Iterable<E> iterable, Functio
         return retVal;
     }
 
+    public static <K> Map<K, Integer> countBy(final Iterable<K> iterable) {
+        final Map<K, Integer> retVal = newLinkedHashMap();
+        for (K key : iterable) {
+            if (retVal.containsKey(key)) {
+                retVal.put(key, 1 + retVal.get(key));
+            } else {
+                retVal.put(key, 1);
+            }
+        }
+        return retVal;
+    }
+
     @SuppressWarnings("unchecked")
     public <K, E> Map<K, Integer> countBy(Function<E, K> func) {
         return countBy((Iterable<E>) iterable, func);
     }
 
+    @SuppressWarnings("unchecked")
+    public <K> Map<K, Integer> countBy() {
+        return countBy((Iterable<K>) iterable);
+    }
+
     /*
      * Documented, #toArray
      */
@@ -2885,6 +2902,10 @@ public <F> Chain<Map<F, Integer>> countBy(final Function<T, F> func) {
             return new Chain<Map<F, Integer>>(U.countBy(list, func));
         }
 
+        public Chain<Map<T, Integer>> countBy() {
+            return new Chain<Map<T, Integer>>(U.countBy(list));
+        }
+
         public Chain<T> shuffle() {
             return new Chain<T>(U.shuffle(list));
         }

src/main/java/com/github/underscore/lodash/U.java

@@ -269,6 +269,10 @@ public <F> Chain<Map<F, Integer>> countBy(final Function<T, F> func) {
             return new Chain<Map<F, Integer>>(U.countBy(value(), func));
         }
 
+        public Chain<Map<T, Integer>> countBy() {
+            return new Chain<Map<T, Integer>>(U.countBy(value()));
+        }
+
         public Chain<T> shuffle() {
             return new Chain<T>(U.shuffle(value()));
         }

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

@@ -1712,6 +1712,9 @@ public String apply(Person person) {
             }
         }).item();
         assertEquals("{moe=2, curly=1}", resultChain.toString());
+        U.countBy(asList(1, 2, 3));
+        new U<Integer>(asList(1, 2, 3)).countBy();
+        U.chain(asList(1, 2, 2, 3)).countBy().item();
     }
 
 /*

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

@@ -843,6 +843,7 @@ public Integer apply(Integer a, Integer b) {
         U.chain(new Integer[] {0}).indexBy("");
         U.chain(new Integer[] {0}).countBy(new Function<Integer, Integer>() {
             public Integer apply(Integer value) { return value; } });
+        U.chain(new Integer[] {0}).countBy();
         U.chain(new Integer[] {0}).shuffle();
         U.chain(new Integer[] {0}).sample();
         U.chain(new Integer[] {0}).sample(1);
@@ -1017,4 +1018,16 @@ public void sqlru1() {
 
         assertEquals("[memory, cpu, hdd]", U.keys((Map<String, Object>) U.fromJson(json)).toString());
     }
+
+    @SuppressWarnings("unchecked")
+    @Test
+    public void sqlru2() {
+        System.out.println(U.countBy(U.words("Маша ищет Мишу а Миша ищет Машу"),
+            new Function<String, String>() {
+            public String apply(String item) {
+                return item;
+            }
+            }
+        ).get("ищет"));
+    }
 }