[Bast-commits] r5479 - in DBIx-Class/0.08/trunk: lib/DBIx/Class/Storage/DBI/Oracle t

[caelum at dev.catalyst.perl.org]

Author: caelum
Date: 2009-02-15 23:30:00 +0000 (Sun, 15 Feb 2009)
New Revision: 5479

Modified:
   DBIx-Class/0.08/trunk/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm
   DBIx-Class/0.08/trunk/t/73oracle.t
Log:
Improved connected logic for Oracle storage


Modified: DBIx-Class/0.08/trunk/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm
===================================================================
--- DBIx-Class/0.08/trunk/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm	2009-02-15 16:23:46 UTC (rev 5478)
+++ DBIx-Class/0.08/trunk/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm	2009-02-15 23:30:00 UTC (rev 5479)
@@ -25,6 +25,7 @@
 =cut
 
 use Carp::Clan qw/^DBIx::Class/;
+use Scalar::Util ();
 
 use base qw/DBIx::Class::Storage::DBI::MultiDistinctEmulation/;
 
@@ -84,6 +85,30 @@
   return $id;
 }
 
+sub connected {
+  my $self = shift;
+  
+  if ($self->SUPER::connected(@_)) {
+      my $dbh = $self->_dbh;
+  
+      my $ping_sth = $dbh->prepare_cached("select 1 from dual");
+  
+      local $dbh->{RaiseError} = 1;
+      eval {
+          $ping_sth->execute;
+          $ping_sth->finish;
+      };
+  
+      if ($@) {
+          return 0;
+      } else {
+          return 1;
+      }
+  }
+  
+  return 0;
+}
+
 sub _dbh_execute {
   my $self = shift;
   my ($dbh, $op, $extra_bind, $ident, $bind_attributes, @args) = @_;
@@ -103,17 +128,8 @@
       }
     };
     $exception = $@;
-    if ($exception =~ /ORA-(?:00028|01012)/) {
-# ORA-00028: your session has been killed
-# ORA-01012: not logged on
-      $self->disconnect;
-
-      $self->throw_exception($exception) if $self->{_in_dbh_do};
-      $self->throw_exception($exception) if $self->transaction_depth;
-
-      $self->ensure_connected;
-    } elsif ($exception =~ /ORA-01003/) { # invalid cursor
-# ORA-01003: no statement parsed (someone renamed a column or something,
+    if ($exception =~ /ORA-01003/) {
+# ORA-01003: no statement parsed (someone changed the table somehow,
 # invalidating your cursor.)
       my ($sql, $bind) = $self->_prep_for_execute($op, $extra_bind, $ident, \@args);
       delete $dbh->{CachedKids}{$sql};

Modified: DBIx-Class/0.08/trunk/t/73oracle.t
===================================================================
--- DBIx-Class/0.08/trunk/t/73oracle.t	2009-02-15 16:23:46 UTC (rev 5478)
+++ DBIx-Class/0.08/trunk/t/73oracle.t	2009-02-15 23:30:00 UTC (rev 5479)
@@ -149,7 +149,7 @@
 
 # clean up our mess
 END {
-    if($dbh) {
+    if($dbh = $schema->storage->dbh) {
         $dbh->do("DROP SEQUENCE artist_seq");
         $dbh->do("DROP SEQUENCE pkid1_seq");
         $dbh->do("DROP SEQUENCE pkid2_seq");