File Coverage

blib/lib/Plack/Middleware/DBIx/DisconnectAll.pm

Criterion	Covered	Total	%
statement	22	28	78.5
branch	1	4	25.0
condition			n/a
subroutine	7	7	100.0
pod	1	1	100.0
total	31	40	77.5

package Plack::Middleware::DBIx::DisconnectAll;

use strict;
use warnings;
use 5.008005;
use parent qw/Plack::Middleware/;
use DBIx::DisconnectAll;

our $VERSION = '0.02';

sub call {
    my ( $self, $env) = @_;
    my $res = $self->app->($env);
    Plack::Util::response_cb($res, sub {
        my $res = shift;
        if ( defined $res->[2] ) {
            dbi_disconnect_all();
            return;
        }
        return sub {
            my $chunk = shift;
            if ( ! defined $chunk ) {
                dbi_disconnect_all();
                return;
            }
            return $chunk;
        };
    });
}

1;
__END__