File Coverage

blib/lib/AnyEvent/Semaphore.pm

Criterion	Covered	Total	%
statement	15	59	25.4
branch	0	10	0.0
condition	0	5	0.0
subroutine	5	12	41.6
pod	4	4	100.0
total	24	90	26.6

package AnyEvent::Semaphore;

our $VERSION = '0.01';

use strict;
use warnings;

use AE;
use Scalar::Util ();
use Method::WeakCallback qw(weak_method_callback);

# internal representation of watcher is an array [$semaphore, $cb]

sub new {
    my ($class, $size) = @_;
    my $sem = { size => $size || 1,
                holes => 0,
                running => 0,
                watchers => [] };
    $sem->{schedule_cb} = weak_method_callback($sem, '_schedule'),
    bless $sem, $class;
}

sub size {
    my $sem = shift;
    if (@_) {
        $sem->{size} = shift;
        &AE::postpone($sem->{schedule_cb});
    }
    $sem->{size};
}

sub running { shift->{running} }

sub down {
    return unless defined $_[1];
    my ($sem) = @_;
    my $watchers = $sem->{watchers};
    my $w = [@_];
    bless $w, 'AnyEvent::Semaphore::Watcher';
    push @{$watchers}, $w;
    Scalar::Util::weaken($watchers->[-1]);
    &AE::postpone($sem->{schedule_cb});
    $w;
}

sub _schedule {
    my $sem = shift;
    my $watchers = $sem->{watchers};
    while ($sem->{size} > $sem->{running}) {
        if (defined (my $w = shift @$watchers)) {
            $sem->{running}++;
            my ($cb, @args) = splice @$w, 1;
            bless $w, 'AnyEvent::Semaphore::Down';
            $cb->(@args);
        }
        else {
            @$watchers or return;
        }
    }
}

sub AnyEvent::Semaphore::Watcher::DESTROY {
    local ($!, $@, $SIG{__DIE__});
    eval {
        my $watcher = shift;
        my $sem = $watcher->[0];
        my $holes = ++$sem->{holes};
        my $watchers = $sem->{watchers};
        if ($holes > 100 and $holes * 2 > @$watchers) {
            @{$sem->{watchers}} = grep defined, @$watchers;
            Scalar::Util::weaken $_ for @$watchers;
            $sem->{holes} = 0;
        }
    }
}

sub AnyEvent::Semaphore::Down::DESTROY {
    local ($!, $@, $SIG{__DIE__});
    eval {
        my $sem = shift->[0];
        $sem->{running}--;
        &AE::postpone($sem->{schedule_cb})
    }
}

1;

__END__