File Coverage

blib/lib/Command/Runner/Timeout.pm

Criterion	Covered	Total	%
statement	26	26	100.0
branch	5	6	83.3
condition	5	6	83.3
subroutine	6	6	100.0
pod	0	3	0.0
total	42	47	89.3

package Command::Runner::Timeout;
use strict;
use warnings;
use Time::HiRes ();

our $_USE_CLOCK_MONOTONIC = 0;

my $time;
{
    local $SIG{__DIE__} = 'DEFAULT';
    local $@;
    if (eval 'Time::HiRes::clock_gettime( Time::HiRes::CLOCK_MONOTONIC() )') {
        $time = sub { Time::HiRes::clock_gettime(Time::HiRes::CLOCK_MONOTONIC()) };
        $_USE_CLOCK_MONOTONIC = 1;
    } else {
        $time = \&Time::HiRes::time;
    }
}

sub new {
    my ($class, $at, $kill) = @_;
    my $now = $time->();
    bless { signaled => 0, at => $now + $at, at_kill => $now + $at + $kill }, $class;
}

sub signal {
    my $self = shift;
    return if !$self->{at} && !$self->{at_kill};
    my $now = $time->();
    if ($self->{at} and $now >= $self->{at}) {
        $self->{at} = undef;
        $self->{signaled} = 1;
        return 'TERM';
    }
    if ($now >= $self->{at_kill}) {
        $self->{at_kill} = undef;
        $self->{signaled} = 1;
        return 'KILL';
    }
    return;
}

sub signaled {
    my $self = shift;
    $self->{signaled};
}

1;