File Coverage

blib/lib/CouchDB/View/Server.pm

Criterion	Covered	Total	%
statement	39	51	76.4
branch	6	18	33.3
condition	3	5	60.0
subroutine	11	14	78.5
pod	3	10	30.0
total	62	98	63.2

use strict;
use warnings;

package CouchDB::View::Server;

use JSON::XS;
use IO::Handle;

my $j = JSON::XS->new;

{
  our @d;
  sub dmap { push @d, [@_] }
}

sub new  { bless $_[1] => $_[0] }

sub in   { @_ > 1 ? ($_[0]->{in}   = $_[1]) : $_[0]->{in}   }
sub out  { @_ > 1 ? ($_[0]->{out}  = $_[1]) : $_[0]->{out}  }
sub funs { @_ > 1 ? ($_[0]->{funs} = $_[1]) : $_[0]->{funs} }

my %fun_cache;

sub run {
  my $self = shift;

  $self = $self->new if not ref $self; # autovivify

  $self->in  or $self->in (IO::Handle->new_from_fd(\*STDIN, 'r'));
  $self->out or $self->out(IO::Handle->new_from_fd(\*STDOUT, 'w'));

  $self->out->autoflush(1);

  while (defined(my $line = $self->in->getline)) {
    $self->process($line);
  }
}

sub process {
  my ($self, $line) = @_;
  chomp($line);
  my $input = $j->decode($line);
  my ($cmd, @args) = @$input;
  $self->can($cmd)->($self, @args);
}

sub reset {
  my ($self) = @_;
  delete $self->{funs};
  $self->out->print("true\n");
}

sub add_fun {
  my ($self, $code) = @_;
  my $sub = $fun_cache{$code} ||= eval $code;
  if (my $e = $@) {
    $self->out->print(
      $j->encode({
        error => {
          id => "map_compilation_error",
          reason => $e,
        },
      }), "\n",
    );
  } else {
    push @{ $self->{funs} ||= [] }, $sub;
    $self->out->print("true\n");
  }
}

sub map_doc {
  my ($self, $doc) = @_;
  my @result;
  for my $sub (@{ $self->funs || [] }) {
    our @d;
    local @d;
    eval { $sub->($doc) };
    # we don't have any concept of 'fatal' yet
    if (my $e = $@) {
      warn $e;
    } else {
      push @result, [@d];
    }
  }

  $self->out->print($j->utf8->encode(\@result), "\n");
}


1;
__END__