Merge pull request #1 from pcmantz/inatatime

implement inatatime: return n elements at a time from an interable value
ray1729 · Feb 2, 2012 · b3b6f70 · b3b6f70
2 parents e4f18f5 + 8e913f4
commit b3b6f70
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/lib/Iterator/Simple/Util.pm b/lib/Iterator/Simple/Util.pm
@@ -9,6 +9,7 @@ use Sub::Exporter -setup => {
                      iany inone inotall
                      ifirstval ilastval
                      ibefore ibefore_incl iafter iafter_incl
+                     inatatime
                )
              ]
 };
@@ -308,6 +309,23 @@ sub iafter_incl (&$) {
     return ichain iter( [$_] ), $iter;
 }
 
+sub inatatime ($;$) {
+    my ($kicks, $iter) = @_;
+
+    $iter = iter $iter;
+
+    return iterator {
+        my @vals;
+
+        for (1 .. $kicks) {
+            my $val = $iter->next;
+            last unless defined $val;
+            push @vals, $val;
+        }
+        return @vals ? \@vals : undef;
+    };
+}
+
 sub _ensure_coderef {
     unless( ref( shift ) eq 'CODE' ) {
         require Carp;

diff --git a/t/iterator-simple-util.t b/t/iterator-simple-util.t
@@ -13,7 +13,8 @@ BEGIN {
                 imax imin imaxstr iminstr imax_by imin_by imaxstr_by iminstr_by
                 iany inone inotall
                 ifirstval ilastval
-                ibefore ibefore_incl iafter iafter_incl )
+                ibefore ibefore_incl iafter iafter_incl
+                inatatime )
         );
 }
 
@@ -193,6 +194,11 @@ note "Testing iafter_incl";
     is_deeply $it->to_array, [3..10];
 }
 
+note "Testing inatatime";
+{
+    ok my $it = inatatime 3, iter [0..10];
+    isa_ok $it, Iterator::Simple->ITERATOR_CLASS;
+    is_deeply $it->to_array, [ [0..2], [3..5], [6..8], [9,10] ];
+}
 
-
 done_testing();