File Coverage

blib/lib/Aspect/Library/Memoize.pm

Criterion	Covered	Total	%
statement	25	29	86.2
branch	2	4	50.0
condition	2	3	66.6
subroutine	8	8	100.0
pod	0	1	0.0
total	37	45	82.2

package Aspect::Library::Memoize;

use 5.008002;
use strict;
use warnings;
use Memoize           1.01 ();
use Aspect::Modular   1.00 ();
use Aspect::Advice::Before ();

our $VERSION = '1.00';
our @ISA     = 'Aspect::Modular';

sub get_advice {
	my %WRAPPER = ();
	Aspect::Advice::Before->new(
		lexical  => $_[0]->lexical,
		pointcut => $_[1],
		code     => sub {
			my $name = $_->sub_name;

			# Would be difficult if Memoize did not have INSTALL => undef option
			$WRAPPER{$name} ||= Memoize::memoize(
				$_->original,
				INSTALL => undef,
			);

			# Pass through to the memoised function, using the
			# same wantarray context as the original call.
			if ( $_->wantarray ) {
				my @rv = $WRAPPER{$name}->($_->args);
				$_->return_value(@rv);
			} elsif ( defined $_->wantarray ) {
				my $rv = $WRAPPER{$name}->($_->args);
				$_->return_value($rv);
			} else {
				$WRAPPER{$name}->($_->args);
				$_->return_value;
			}
		},
	);
}

1;

__END__