File Coverage

blib/lib/WWW/Kickstarter/Iterator.pm

Criterion	Covered	Total	%
statement	9	39	23.0
branch	0	4	0.0
condition	0	8	0.0
subroutine	3	8	37.5
pod	4	4	100.0
total	16	63	25.4

package WWW::Kickstarter::Iterator;

use strict;
use warnings;
no autovivification;


sub _done { () }


sub new {
   my ($class, $generator, $buf) = @_;
   my $self = bless({}, $class);
   $self->{generator} = $generator || \&_done;
   $self->{buf      } = $buf       || [];
   return $self;
}


sub get {
   my ($self, $n) = @_;
   $n ||= 1;

   my $buf = $self->{buf};
   while (@$buf < $n) {
      my @add = $self->{generator}->()
         or do {
            $self->{generator} = \&_done;
            last;
         };

      push @$buf, @add;
   }

   return splice(@$buf, 0, $n);
}


sub get_rest {
   my ($self) = @_;

   my $buf = $self->{buf};
   while (my @add = $self->{generator}->()) {
      push @$buf, @add;
   }

   $self->{generator} = \&_done;
   return splice(@$buf, 0, $#$buf);
}


sub get_chunk {
   my ($self, $min) = @_;
   $min ||= 1;

   my $buf = $self->{buf};
   while (@$buf < $min) {
      my @add = $self->{generator}->()
         or do {
            $self->{generator} = \&_done;
            last;
         };

      push @$buf, @add;
   }

   return splice(@$buf, 0, $#$buf);
}


1;


__END__