fix some issues with multi-db_schema support
[dbsrgits/DBIx-Class-Schema-Loader.git] / lib / DBIx / Class / Schema / Loader / DBI / Oracle.pm
index b2e1f5a..4f65da7 100644 (file)
@@ -6,10 +6,9 @@ use base qw/
     DBIx::Class::Schema::Loader::DBI::Component::QuotedDefault
     DBIx::Class::Schema::Loader::DBI
 /;
-use Carp::Clan qw/^DBIx::Class/;
-use Class::C3;
+use mro 'c3';
 
-our $VERSION = '0.07000';
+our $VERSION = '0.07010';
 
 =head1 NAME
 
@@ -18,7 +17,7 @@ Oracle Implementation.
 
 =head1 DESCRIPTION
 
-See L<DBIx::Class::Schema::Loader::Base>.
+See L<DBIx::Class::Schema::Loader> and L<DBIx::Class::Schema::Loader::Base>.
 
 =cut
 
@@ -27,84 +26,87 @@ sub _setup {
 
     $self->next::method(@_);
 
-    my $dbh = $self->schema->storage->dbh;
+    my ($current_schema) = $self->dbh->selectrow_array('SELECT USER FROM DUAL');
 
-    my ($current_schema) = $dbh->selectrow_array('SELECT USER FROM DUAL', {});
+    $self->db_schema([ $current_schema ]) unless $self->db_schema;
 
-    $self->{db_schema} ||= $current_schema;
-
-    if (lc($self->db_schema) ne lc($current_schema)) {
-        $dbh->do('ALTER SESSION SET current_schema=' . $self->db_schema);
+    if (@{ $self->db_schema } == 1 && $self->db_schema->[0] ne '%'
+        && lc($self->db_schema->[0]) ne lc($current_schema)) {
+        $self->dbh->do('ALTER SESSION SET current_schema=' . $self->db_schema->[0]);
     }
 
     if (not defined $self->preserve_case) {
         $self->preserve_case(0);
     }
     elsif ($self->preserve_case) {
-        $self->schema->storage->quote_char('"');
-        $self->schema->storage->name_sep('.');
+        $self->schema->storage->sql_maker->quote_char('"');
+        $self->schema->storage->sql_maker->name_sep('.');
     }
 }
 
-sub _table_as_sql {
-    my ($self, $table) = @_;
+sub _build_name_sep { '.' }
+
+sub _system_schemas {
+    my $self = shift;
 
-    return $self->_quote_table_name($table);
+    # From http://www.adp-gmbh.ch/ora/misc/known_schemas.html
+
+    return ($self->next::method(@_), qw/ANONYMOUS APEX_PUBLIC_USER APEX_030200 APPQOSSYS CTXSYS DBSNMP DIP DMSYS EXFSYS LBACSYS MDDATA MDSYS MGMT_VIEW OLAPSYS ORACLE_OCM ORDDATA ORDPLUGINS ORDSYS OUTLN SI_INFORMTN_SCHEMA SPATIAL_CSW_ADMIN_USR SPATIAL_WFS_ADMIN_USR SYS SYSMAN SYSTEM TRACESRV MTSSYS OASPUBLIC OWBSYS OWBSYS_AUDIT WEBSYS WK_PROXY WKSYS WK_TEST WMSYS XDB OSE$HTTP$ADMIN AURORA$JIS$UTILITY$ AURORA$ORB$UNAUTHENTICATED/, qr/^FLOWS_\d\d\d\d\d\d\z/);
 }
 
-sub _tables_list { 
-    my ($self, $opts) = @_;
+sub _system_tables {
+    my $self = shift;
 
-    my $dbh = $self->schema->storage->dbh;
+    return ($self->next::method(@_), 'PLAN_TABLE');
+}
 
-    my @tables;
-    for my $table ( $dbh->tables(undef, $self->db_schema, '%', 'TABLE,VIEW') ) { #catalog, schema, table, type
-        my $quoter = $dbh->get_info(29);
-        $table =~ s/$quoter//g;
+sub _dbh_tables {
+    my ($self, $schema) = @_;
 
-        # remove "user." (schema) prefixes
-        $table =~ s/\w+\.//;
+    return $self->dbh->tables(undef, $schema, '%', 'TABLE,VIEW');
+}
 
-        next if $table eq 'PLAN_TABLE';
-        $table = $self->_lc($table);
-        push @tables, $1
-          if $table =~ /\A(\w+)\z/;
-    }
+sub _filter_tables {
+    my $self = shift;
 
-    return $self->_filter_tables(\@tables, $opts);
+    # silence a warning from older DBD::Oracles in tests
+    my $warn_handler = $SIG{__WARN__} || sub { warn @_ };
+    local $SIG{__WARN__} = sub {
+        $warn_handler->(@_)
+        unless $_[0] =~ /^Field \d+ has an Oracle type \(\d+\) which is not explicitly supported/;
+    };
+
+    return $self->next::method(@_);
 }
 
 sub _table_columns {
     my ($self, $table) = @_;
 
-    my $dbh = $self->schema->storage->dbh;
-
-    my $sth = $dbh->column_info(undef, $self->db_schema, $self->_uc($table), '%');
+    my $sth = $self->dbh->column_info(undef, $table->schema, $table, '%');
 
-    return [ map lc($_->{COLUMN_NAME}), @{ $sth->fetchall_arrayref({ COLUMN_NAME => 1 }) || [] } ];
+    return [ map $self->_lc($_->{COLUMN_NAME}), @{ $sth->fetchall_arrayref({ COLUMN_NAME => 1 }) || [] } ];
 }
 
 sub _table_uniq_info {
     my ($self, $table) = @_;
 
-    my $dbh = $self->schema->storage->dbh;
+    my $sth = $self->dbh->prepare_cached(<<'EOF', {}, 1);
+SELECT ac.constraint_name, acc.column_name
+FROM all_constraints ac, all_cons_columns acc
+WHERE acc.table_name=? AND acc.owner = ?
+    AND ac.table_name = acc.table_name AND ac.owner = acc.owner
+    AND acc.constraint_name = ac.constraint_name
+    AND ac.constraint_type='U'
+ORDER BY acc.position
+EOF
 
-    my $sth = $dbh->prepare_cached(
-        q{
-            SELECT constraint_name, acc.column_name
-            FROM all_constraints JOIN all_cons_columns acc USING (constraint_name)
-            WHERE acc.table_name=? and acc.owner = ? AND constraint_type='U'
-            ORDER BY acc.position
-        },
-        {}, 1);
+    $sth->execute($table->name, $table->schema);
 
-    $sth->execute($self->_uc($table),$self->{db_schema} );
     my %constr_names;
+
     while(my $constr = $sth->fetchrow_arrayref) {
-        my $constr_name = $constr->[0];
+        my $constr_name = $self->_lc($constr->[0]);
         my $constr_col  = $self->_lc($constr->[1]);
-        $constr_name =~ s/\Q$self->{_quoter}\E//;
-        $constr_col  =~ s/\Q$self->{_quoter}\E//;
         push @{$constr_names{$constr_name}}, $constr_col;
     }
     
@@ -112,44 +114,73 @@ sub _table_uniq_info {
     return \@uniqs;
 }
 
-sub _table_pk_info {
-    my ($self, $table) = (shift, shift);
+sub _table_comment {
+    my $self = shift;
+    my ($table) = @_;
+
+    my $table_comment = $self->next::method(@_);
 
-    return $self->next::method($self->_uc($table), @_);
+    return $table_comment if $table_comment;
+
+    ($table_comment) = $self->dbh->selectrow_array(<<'EOF', {}, $table->schema, $table->name);
+SELECT comments FROM all_tab_comments
+WHERE owner = ? 
+  AND table_name = ?
+  AND (table_type = 'TABLE' OR table_type = 'VIEW')
+EOF
+
+    return $table_comment
 }
 
-sub _table_fk_info {
-    my ($self, $table) = (shift, shift);
+sub _column_comment {
+    my $self = shift;
+    my ($table, $column_number, $column_name) = @_;
 
-    my $rels = $self->next::method($self->_uc($table), @_);
+    my $column_comment = $self->next::method(@_);
 
-    foreach my $rel (@$rels) {
-        $rel->{remote_table} = $self->_lc($rel->{remote_table});
-    }
+    return $column_comment if $column_comment;
 
-    return $rels;
+    ($column_comment) = $self->dbh->selectrow_array(<<'EOF', {}, $table->schema, $table->name, $self->_uc($column_name));
+SELECT comments FROM all_col_comments
+WHERE owner = ? 
+  AND table_name = ?
+  AND column_name = ?
+EOF
+
+    return $column_comment
 }
 
 sub _columns_info_for {
-    my ($self, $table) = (shift, shift);
+    my $self = shift;
+    my ($table) = @_;
 
-    my $result = $self->next::method($self->_uc($table), @_);
+    my $result = $self->next::method(@_);
 
-    my $dbh = $self->schema->storage->dbh;
+    local $self->dbh->{LongReadLen} = 100000;
+    local $self->dbh->{LongTruncOk} = 1;
 
-    my $sth = $dbh->prepare_cached(q{
-SELECT atc.column_name
-FROM all_triggers ut
-JOIN all_trigger_cols atc USING (trigger_name)
-WHERE atc.table_name = ?
-AND lower(column_usage) LIKE '%new%' AND lower(column_usage) LIKE '%out%'
+    my $sth = $self->dbh->prepare_cached(<<'EOF', {}, 1);
+SELECT trigger_body
+FROM all_triggers
+WHERE table_name = ? AND table_owner = ?
 AND upper(trigger_type) LIKE '%BEFORE EACH ROW%' AND lower(triggering_event) LIKE '%insert%'
-    }, {}, 1);
+EOF
+
+    $sth->execute($table->name, $table->schema);
 
-    $sth->execute($self->_uc($table));
+    while (my ($trigger_body) = $sth->fetchrow_array) {
+        if (my ($seq_schema, $seq_name) = $trigger_body =~ /(?:\."?(\w+)"?)?"?(\w+)"?\.nextval/i) {
+            if (my ($col_name) = $trigger_body =~ /:new\.(\w+)/i) {
+                $col_name = $self->_lc($col_name);
 
-    while (my ($col_name) = $sth->fetchrow_array) {
-        $result->{$self->_lc($col_name)}{is_auto_increment} = 1;
+                $result->{$col_name}{is_auto_increment} = 1;
+
+                $seq_schema = $self->_lc($seq_schema || $table->schema);
+                $seq_name   = $self->_lc($seq_name);
+
+                $result->{$col_name}{sequence} = ($self->qualify_objects ? ($seq_schema . '.') : '') . $seq_name;
+            }
+        }
     }
 
     while (my ($col, $info) = each %$result) {
@@ -163,9 +194,12 @@ AND upper(trigger_type) LIKE '%BEFORE EACH ROW%' AND lower(triggering_event) LIK
             $info->{size} = $info->{size} / 2;
         }
         elsif (lc($info->{data_type}) eq 'number') {
-            $info->{data_type} = 'numeric';
+            $info->{original}{data_type} = 'number';
+            $info->{data_type}           = 'numeric';
 
             if (eval { $info->{size}[0] == 38 && $info->{size}[1] == 0 }) {
+                $info->{original}{size} = $info->{size};
+
                 $info->{data_type} = 'integer';
                 delete $info->{size};
             }
@@ -228,11 +262,12 @@ AND upper(trigger_type) LIKE '%BEFORE EACH ROW%' AND lower(triggering_event) LIK
             $info->{original}{data_type} = 'binary_double';
         } 
 
-        if (eval { lc(${ $info->{default_value} }) eq 'sysdate' }) {
-            $info->{original}{default_value} = $info->{default_value};
-
+        if ((eval { lc(${ $info->{default_value} }) }||'') eq 'sysdate') {
             my $current_timestamp  = 'current_timestamp';
             $info->{default_value} = \$current_timestamp;
+
+            my $sysdate = 'sysdate';
+            $info->{original}{default_value} = \$sysdate;
         }
     }