File Coverage

blib/lib/Plack/Middleware/MemoryUsage.pm

Criterion	Covered	Total	%
statement	36	36	100.0
branch	3	4	75.0
condition	2	2	100.0
subroutine	9	9	100.0
pod	1	2	50.0
total	51	53	96.2

package Plack::Middleware::MemoryUsage;

use strict;
use warnings;

use parent qw( Plack::Middleware );
use Plack::Util::Accessor qw( callback packages);

our $VERSION = '0.03';

use B::Size2::Terse;
use Devel::Symdump;

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

    my $before = $self->memory_usage();
    my $res    = $self->app->($env);
    my $after  = $self->memory_usage();
    my $diff   = {};

    for my $pkg (keys %$after) {
        $diff->{$pkg} = $after->{$pkg} - ($before->{$pkg} || 0);
    }

    $self->response_cb($res, sub {
                           my $res = shift;
                           $self->callback->( $env, $res, $before, $after, $diff )
                               if $self->callback;
                       });
}

sub memory_usage {
    my $self = shift;
    my @packages = $self->packages ? @{$self->packages} : Devel::Symdump->rnew("main")->packages;
    my $size;

    for my $package ("main", @packages) {
        my($subs, $opcount, $opsize) = B::Size2::Terse::package_size($package);
        $size->{$package} = $opsize;
    }
    return $size;
}

1;

__END__