File Coverage

blib/lib/DBIx/TransactionManager/EndHook.pm

Criterion	Covered	Total	%
statement	42	42	100.0
branch	4	4	100.0
condition	4	5	80.0
subroutine	12	12	100.0
pod	0	3	0.0
total	62	66	93.9

package DBIx::TransactionManager::EndHook;
use 5.008_001;
use strict;
use warnings;

our $VERSION = '0.02';
use DBIx::TransactionManager;

use Try::Tiny;
use Carp;

sub DBIx::TransactionManager::add_end_hook {
    my ($self, $sub) = @_;

    unless ( $self->in_transaction ) {
        croak "only can call add_end_hook in transaction";
    }

    my $arr = $self->{_end_hooks} ||= [];
    push @$arr, $sub;
}

my $orig_txn_commit = \&DBIx::TransactionManager::txn_commit;
my $orig_txn_rollback = \&DBIx::TransactionManager::txn_rollback;

{
    no warnings 'redefine';
    *DBIx::TransactionManager::txn_commit = *txn_commit;
    *DBIx::TransactionManager::txn_rollback = *txn_rollback;
}

sub txn_commit {
    my $self = shift;
    my $is_last_txn = @{ $self->active_transactions } == 1;

    my $ret = $orig_txn_commit->($self);

    if ( $is_last_txn && defined $self->{_end_hooks} ) {
        try {
            while ( my $end_hook = shift @{ $self->{_end_hooks} } ) {
                $end_hook->();
            }
        }
        catch {
            $self->{_end_hooks} = [];
            croak $_;
        };
    }

    $ret;
}

sub txn_rollback {
    my $self = shift;
    my $ret = $orig_txn_rollback->($self);
    $self->{_end_hooks} = [];
    $ret;
}

1;
__END__