File Coverage

blib/lib/Plack/Middleware/TimeStats.pm

Criterion	Covered	Total	%
statement	34	34	100.0
branch	8	8	100.0
condition			n/a
subroutine	9	9	100.0
pod	2	2	100.0
total	53	53	100.0

package Plack::Middleware::TimeStats;
use strict;
use warnings;
use parent 'Plack::Middleware';
use Plack::Util::Accessor qw/callback psgix option action/;
use Devel::TimeStats;

our $VERSION = '0.03';

sub prepare_app {
    my $self = shift;

    if ($self->psgix) {
        $self->psgix('psgix.'. $self->psgix);
    }
    else {
        $self->psgix('psgix.timestats');
    }

    $self->option(+{
        percentage_decimal_precision => 2,
    }) unless $self->option;

    $self->callback(
        sub {
            my ($stats, $env, $res) = @_;
            warn scalar($stats->report);
        }
    ) unless $self->callback;
}

sub call {
    my($self, $env) = @_;

    $env->{$self->psgix} = Devel::TimeStats->new($self->option);

    my $action = $self->action ? $self->action->($env) : $env->{PATH_INFO};

    $env->{$self->psgix}->profile(
        begin   => $action,
        comment => '',
    );

    my $res = $self->app->($env);

    $self->response_cb($res, sub {
        my $res = shift;

        $env->{$self->psgix}->profile('end app');

        $env->{$self->psgix}->profile(
            end => $action,
        );

        $self->callback->($env->{$self->psgix}, $env, $res);
        return;
    });
}


1;

__END__