File Coverage

blib/lib/Memory/Stats.pm

Criterion	Covered	Total	%
statement	56	56	100.0
branch	10	12	83.3
condition	10	11	90.9
subroutine	12	12	100.0
pod	6	6	100.0
total	94	97	96.9

#
# This file is part of Memory-Stats
#
# This software is copyright (c) 2014 by celogeek .
#
# This is free software; you can redistribute it and/or modify it under
# the same terms as the Perl 5 programming language system itself.
#
package Memory::Stats;

# ABSTRACT: Memory Usage Consumption of your process

use strict;
use warnings;
our $VERSION = '0.04';    # VERSION
use Proc::ProcessTable;
use Carp qw/croak/;
use Moo;

my $pt = Proc::ProcessTable->new;

has '_memory_usage' => ( is => 'rw', default => sub { [] } );

sub _get_current_memory_usage {
    my ($info) = grep { $_->pid eq $$ } @{ $pt->table };
    my $memory_usage;
    return -1 if !defined eval { $memory_usage = $info->rss };
    return $memory_usage;
}

sub start {
    my $self = shift;
    $self->_memory_usage( [ [ "start", _get_current_memory_usage() ] ] );
    return;
}

sub checkpoint {
    my $self = shift;
    my $title = shift // 'checkpoint';
    croak "Please start first !"
        if !scalar @{ $self->_memory_usage }
        || $self->_memory_usage->[-1][0] eq 'stop';
    push @{ $self->_memory_usage }, [ $title, _get_current_memory_usage() ];
    return;
}

sub stop {
    my $self = shift;
    croak "Please start first !"
        if !scalar @{ $self->_memory_usage }
        || $self->_memory_usage->[-1][0] eq 'stop';
    push @{ $self->_memory_usage }, [ 'stop', _get_current_memory_usage() ];
    return;
}

sub delta_usage {
    my $self              = shift;
    my $last_memory_usage = $self->_memory_usage->[-1][1]
        or return;
    return ( _get_current_memory_usage() - $last_memory_usage );
}

sub usage {
    my $self = shift;
    croak "Please start and stop before !"
        if scalar @{ $self->_memory_usage } < 2
        || $self->_memory_usage->[-1][0] ne 'stop';
    return $self->_memory_usage->[-1][1] - $self->_memory_usage->[0][1];

}

sub report {
    my $self = shift;
    print "--- Memory Usage ---\n";
    my $prev;
    for my $row ( @{ $self->_memory_usage } ) {
        if ($prev) {
            printf(
                "%s: %d - delta: %d - total: %d\n",
                @$row,
                $row->[1] - $prev,
                $row->[1] - $self->_memory_usage->[0][1]
            );
        }
        else {
            printf( "%s: %d\n", @$row );
        }
        $prev = $row->[1];
    }
    print "--- Memory Usage ---\n";
    return;
}

1;

__END__