File Coverage

blib/lib/XAS/Lib/Pidfile.pm

Criterion	Covered	Total	%
statement	11	53	20.7
branch	1	18	5.5
condition			n/a
subroutine	4	9	44.4
pod	3	3	100.0
total	19	83	22.8

package XAS::Lib::Pidfile;

our $VERSION = '0.01';
my $mixin;

BEGIN {
    $mixin = 'XAS::Lib::Pidfile::Unix';
    $mixin = 'XAS::Lib::Pidfile::Win32' if ($^O eq 'MSWin32');
}

use XAS::Factory;
use File::Basename;

use XAS::Class
  debug      => 0,
  version    => $VERSION,
  base       => 'XAS::Base',
  mixin      => $mixin,
  utils      => 'trim dotid',
  accessors  => 'lockmgr lock',
  filesystem => 'Dir',
  vars => {
    PARAMS => {
      pid  => 1,
      file => { optional => 1, default => undef, isa => 'Badger::Filesystem::File' },
    }
  }
;

#use Data::Dumper;

# ----------------------------------------------------------------------
# Public Methods
# ----------------------------------------------------------------------

sub write {
    my $self = shift;

    my $stat = 0;
    my $lock = $self->lock;

    my $output = sub {

        my $fh = $self->file->open('w');
        $fh->printf("%s\n", $self->pid);
        $fh->close;

    };

    if ($self->lockmgr->lock($lock)) {

        if ($self->file->exists) {

            $output->();

        } else {

            $self->file->create();
            $output->();

        }

        $stat = 1;
        $self->lockmgr->unlock($lock);

    }

    return $stat;

}

sub remove {
    my $self = shift;

    my $lock = $self->lock;
    my $pid = $self->_get_pid;

    if ($pid == $$) {

        if ($self->lockmgr->lock($lock)) {

            $self->file->delete() if ($self->file->exists);
            $self->lockmgr->unlock($lock);

        }

    }

}

# ----------------------------------------------------------------------
# Private Methods
# ----------------------------------------------------------------------

sub _get_pid {
    my $self = shift;

    my $pid = undef;
    my $lock = $self->lock;

    if ($self->lockmgr->lock($lock)) {

        if ($self->file->exists) {

            my $fh = $self->file->open();
            $pid = $fh->getline();
            $pid = trim($pid);
            $fh->close();

        }

        $self->lockmgr->unlock($lock);

    }

    return $pid

}

sub init {
    my $class = shift;

    my $self = $class->SUPER::init(@_);

    unless (defined($self->{'file'})) {

        $self->{'file'} = $self->env->pid_file;

    }

    my $basename = fileparse($self->env->script, qr/\.[^.]*/);

    $self->{'lock'} = Dir($self->file->volume, $self->file->directory, $basename)->path;
    $self->{'lockmgr'} = XAS::Factory->module('lockmgr');

    $self->lockmgr->add(
        -key    => $self->lock,
        -driver => 'Filesystem',
    );

    return $self;

}

1;

__END__