File Coverage

blib/lib/Test/mysqld/Pool.pm

Criterion	Covered	Total	%
statement	15	61	24.5
branch	0	16	0.0
condition	0	3	0.0
subroutine	5	13	38.4
pod	0	3	0.0
total	20	96	20.8

package Test::mysqld::Pool;
use strict;
use warnings;
use Mouse;
use Test::mysqld;
use Cache::FastMmap;

has jobs         => ( is => 'rw', isa => 'Int', );
has share_file   => ( is => 'rw', isa => 'Str', required => 1 );
has cache        => ( is => 'rw', lazy => 1,
                      default => sub {
                          my ($self) = @_;

                          # only need this for atomical get_and_set
                          # is there anything better?

                          # dont let Cache::FastMmap delete the share_file,
                          # File::Temp does that
                          return Cache::FastMmap->new(
                              share_file     => $self->share_file,
                              init_file      => 0,
                              empty_on_exit  => 0,
                              unlink_on_exit => 0,
                              cache_size     => '1k',
                          );
                      });
has preparer     => ( is => 'rw', isa => 'Maybe[CodeRef]' );
has my_cnf       => ( is => 'rw', isa => 'HashRef',
                      default => sub {
                          {
                              'skip-networking' => '', # no TCP socket
                          };
                      } );
has instances    => ( is => 'rw', isa => 'ArrayRef' );
has _owner_pid   => ( is => 'ro', isa => 'Int', default => sub { $$ } );

sub prepare {
    my ($self) = @_;

    my @instances = Test::mysqld->start_mysqlds($self->jobs, my_cnf => $self->my_cnf);
    $self->instances( \@instances );

    my $orig = $SIG{INT};
    $SIG{INT} = sub {
        Test::mysqld->stop_mysqlds(grep {defined($_)} @instances);
        $self->instances([]) if $self;
        if ($orig) {
            $orig->();
        } else {
            $SIG{INT} = 'DEFAULT';
            kill INT => $$;
        }
    };

    if ($self->preparer) {
        $self->preparer->($_) for @instances;
    }

    $self->cache->clear;
    $self->cache->set( dsns => {
        map { $_->dsn => 0 } @instances
    });
}

sub alloc {
    my ($self) = @_;

    my $ret_dsn;
    do {
        $self->cache->get_and_set( dsns => sub {
            my ($key, $val) = @_;

            for my $dsn (keys %$val) {
                if ( $val->{ $dsn } == 0 ) {
                    # alloc one from unused
                    $ret_dsn = $dsn;
                    $val->{ $dsn } = $$; # record pid
                    return $val;
                }
            }

            return $val;
        });

        return $ret_dsn if $ret_dsn;

        sleep 1;

    } while ( ! $ret_dsn );
}

sub dealloc_unused {
    my ($self) = @_;

    $self->cache->get_and_set( dsns => sub {
        my ($key, $val) = @_;
        for my $dsn (keys %$val) {

            my $pid = $val->{ $dsn }
                or next;

            if ( ! $self->_pid_lives( $pid ) ) {
                $val->{ $dsn } = 0; # dealloc
            }
        }

        return $val;
    });
}

sub _pid_lives {
    my ($self, $pid) = @_;

    my $command = "ps -o pid -p $pid | grep $pid";
    my @lines   = qx{$command};
    return scalar @lines;
}

sub DESTROY {
    my $self = shift;
    Test::mysqld->stop_mysqlds(grep { defined($_) } @{$self->instances})
            if $self->instances && $$ == $self->_owner_pid;
}

1;

__END__