File Coverage

blib/lib/Dancer2/Plugin/SessionDatabase.pm

Criterion	Covered	Total	%
statement	16	39	41.0
branch	0	10	0.0
condition	0	3	0.0
subroutine	5	11	45.4
pod	0	4	0.0
total	21	67	31.3

package Dancer2::Plugin::SessionDatabase;

use Modern::Perl;
use Dancer2::Plugin;
use Carp qw(croak);
use Data::Dumper;

=head1 NAME

Dancer2::Plugin::SessionDatabase - Hook Loader For Dancer2::Session::DatabasePlugin

=head1 DESCRIPTION

Hook loader for Dancer2::Session::DatabasePlugin.

=cut

our $DBH;

sub reset_session {
  %{$Dancer2::Session::DatabasePlugin::CACHE}=();
  $DBH=undef;
}

sub DBC {
    my ($self,$conn)=@_;
    my $db=$self->find_plugin('Dancer2::Plugin::Database');
    my $dbh=$db->database($conn);

    if(defined($DBH) && $dbh ne $DBH) {
      %{$Dancer2::Session::DatabasePlugin::CACHE}=();
    }

    return $DBH=$dbh;
}

sub db_check {
  my ($self,$dbh)=@_;
  return 0 unless defined($DBH);
  return 0 unless $DBH eq $dbh;
  return 1;
}

# This method runs after the new constructor
sub BUILD {
  my ($self)=@_;

  $self->app->add_hook(
    Dancer2::Core::Hook->new(
      name=>"engine.session.before_db",
      code=>sub {
          my ($session)=@_;
          my $dbh=$self->DBC($session->connection);
          $session->dbh($dbh);
        }
      )
  );
  $self->app->add_hook(
    Dancer2::Core::Hook->new(
      name=>"database_connection_lost",
      code=>sub {
        	my ($dbh)=@_;
          return unless $self->db_check($dbh);
          $self->reset_session;
        }
      )
  );
  $self->app->add_hook(
    Dancer2::Core::Hook->new(
      name=>"database_error",
      code=>sub {
	        my ($err,$dbh)=@_;
          return unless $self->db_check($dbh);
          $self->reset_session;
      }
    )
  );
}

=head1 AUTHOR

Michael Shipper [e-mail]

=cut

1;