support subclass

lib/Scope/Container/DBI.pm

@@ -10,6 +10,7 @@ use Carp;
 use DBI;
 
 our $VERSION = '0.01';
+our $DBI_CLASS = 'DBI';
 
 sub connect {
     my $class = shift;
@@ -49,9 +50,9 @@ sub connect {
     my $dbh = do {
         if ($INC{'Apache/DBI.pm'} && $ENV{MOD_PERL}) {
             local $DBI::connect_via = 'connect'; # Disable Apache::DBI.
-            DBI->connect( $dsn, $user, $pass, $attr );
+            $DBI_CLASS->connect( $dsn, $user, $pass, $attr );
         } else {
-            DBI->connect( $dsn, $user, $pass, $attr );
+            $DBI_CLASS->connect( $dsn, $user, $pass, $attr );
         }
     };
     croak($DBI::errstr) if !$dbh;
@@ -194,6 +195,21 @@ Scope::Container::DBI doesn't have callback function, but you can set callbacks
       },
   });
 
+=item USING DBI SUBCLASSES
+
+There is two way of using DBI subclass with Scope::Container::DBI. One is DBI's RootClass attribute, other is $Scope::Container::DBI::DBI_CLASS.
+
+  # use RootClass
+
+  my $dbh = Scope::Container::DBI->connect($dsn, $username, $password, {
+      RootClass => 'MySubDBI',
+  });
+
+  # use $Scope::Container::DBI::DBI_CLASS
+
+  local $Scope::Container::DBI::DBI_CLASS = 'MySubDBI';
+  my $dbh = Scope::Container::DBI->connect($dsn, $username, $password);
+
 =back
 
 =head1 AUTHOR

t/02_subclass.t

@@ -0,0 +1,23 @@
+use strict;
+
+package MySubDBI;
+
+use base qw/DBI/;
+
+package MySubDBI::db;
+our @ISA = qw(DBI::db);
+
+package MySubDBI::st;
+our @ISA = qw(DBI::st);
+
+package main;
+
+use Test::More;
+use Scope::Container::DBI;
+
+local $Scope::Container::DBI::DBI_CLASS = 'MySubDBI';
+my $dbh = Scope::Container::DBI->connect("dbi:SQLite::memory:","","");
+ok($dbh);
+is ref($dbh), 'MySubDBI::db';
+
+done_testing;

xt/01_podspell.t

@@ -12,4 +12,5 @@ Scope::Container::DBI
 DBI's
 InactiveDestroy
 dsn
+RootClass