File Coverage

blib/lib/Mojolicious/Plugin/DirectoryQueue.pm

Criterion	Covered	Total	%
statement	1	3	33.3
branch			n/a
condition			n/a
subroutine	1	1	100.0
pod			n/a
total	2	4	50.0

package Mojolicious::Plugin::DirectoryQueue;

use Mojo::Base 'Mojolicious::Plugin';
use Mojo::JSON qw(decode_json encode_json);
use Directory::Queue;
use POSIX qw(chown);

our $VERSION = '0.01';


sub register {
    my ($plugin, $app, $args) = @_;

    $args->{path}  ||= '/tmp/DirQueue';
    my $dirq = Directory::Queue->new( path => $args->{path} );

    my $config = $app->config;
    if ( $config->{hypnotoad} and $config->{hypnotoad}->{user} ) {
        my ($uid,$gid) = ( getpwnam($config->{hypnotoad}->{user}) )[2,3];
        chown $uid, $gid, $args->{path};
    }

    $app->helper(
        enqueue => sub {
            my ($self, $args) = @_;
            $dirq->add(encode_json($args));
        },
    );

    $app->helper(
        dequeue => sub {
            for (my $name = $dirq->first(); $name; $name = $dirq->next()) {
               next unless $dirq->lock($name);
               my $data = $dirq->get($name);
               $dirq->remove($name);
               return decode_json($data) if $data;
            }
            return;
        },
    );

    $app->helper(
        show => sub {
            my @all;
            for (my $name = $dirq->first(); $name; $name = $dirq->next()) {
               next unless $dirq->lock($name);
               my $data = $dirq->get($name);
               push @all, decode_json($data) if $data;
               $dirq->unlock($name);
            }
            return \@all;
        },
    );

    $app->helper(
        count => sub {
            return $dirq->count();
        },
    );
    $app->helper(
        purge => sub {
            my $self = shift;
            return $dirq->purge(@_);
        },
    );
};

1;
__END__