File Coverage

blib/lib/HTTP/LoadGen/Logger.pm

Criterion	Covered	Total	%
statement	4	6	66.6
branch			n/a
condition			n/a
subroutine	2	2	100.0
pod			n/a
total	6	8	75.0

package HTTP::LoadGen::Logger;

use strict;
use Coro;
use Coro qw/:prio/;
use Coro::Channel;
use Coro::Handle;

our $VERSION = '0.02';

sub get {
  my ($fh, $fmt)=@_;
  my $queue=Coro::Channel->new;
  $fh=\*STDOUT unless $fh;
  unless( ref $fh ) {
    my $name=$fh;
    undef $fh;
    open $fh, '>>', $name or die "Cannot open logfile $name: $!\n";
  }
  $fh=unblock $fh;
  my $thr=async {
    my ($fh)=@_;
    $Coro::current->prio(PRIO_MIN);
    while(defined(my $l=$queue->get)) {
      $fh->syswrite($l);
      cede;
    }
  } $fh;

  if($fmt) {
    return sub {
      if(@_) {
	$queue->put(scalar $fmt->(@_));
      } else {
	$queue->shutdown;
	$thr->join;
      }
    };
  } else {
    return sub {
      if(@_) {
	$queue->put(join("\t", @_)."\n");
      } else {
	$queue->shutdown;
	$thr->join;
      }
    };
  }
}

1;
__END__