X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FSchema%2FLoader%2FDBI.pm;h=33a274ea4f1d517a3acf63062dcc035342df4a12;hb=eb040f78817c0cac743893bd9a252340972464f5;hp=e896ab1e3beb2e8e81da600c2c438dc5436c28eb;hpb=1bcb47d3c20055e9c5d9fe522b3ddd1a7f1bbec9;p=dbsrgits%2FDBIx-Class-Schema-Loader.git diff --git a/lib/DBIx/Class/Schema/Loader/DBI.pm b/lib/DBIx/Class/Schema/Loader/DBI.pm index e896ab1..33a274e 100644 --- a/lib/DBIx/Class/Schema/Loader/DBI.pm +++ b/lib/DBIx/Class/Schema/Loader/DBI.pm @@ -6,7 +6,7 @@ use base qw/DBIx::Class::Schema::Loader::Base/; use Class::C3; use Carp::Clan qw/^DBIx::Class/; -our $VERSION = '0.04999_08'; +our $VERSION = '0.07000'; =head1 NAME @@ -35,19 +35,23 @@ things. sub new { my $self = shift->next::method(@_); - # rebless to vendor-specific class if it exists and loads - my $dbh = $self->schema->storage->dbh; - my $driver = $dbh->{Driver}->{Name}; + # rebless to vendor-specific class if it exists and loads and we're not in a + # custom class. + if (not $self->loader_class) { + my $dbh = $self->schema->storage->dbh; + my $driver = $dbh->{Driver}->{Name}; - my $subclass = 'DBIx::Class::Schema::Loader::DBI::' . $driver; - if ($self->load_optional_class($subclass)) { - bless $self, $subclass unless $self->isa($subclass); - $self->_rebless; + my $subclass = 'DBIx::Class::Schema::Loader::DBI::' . $driver; + if ($self->load_optional_class($subclass)) { + bless $self, $subclass unless $self->isa($subclass); + $self->_rebless; + } } # Set up the default quoting character and name seperators - $self->{_quoter} = $self->_build_quoter; + $self->{_quoter} = $self->_build_quoter; $self->{_namesep} = $self->_build_namesep; + # For our usage as regex matches, concatenating multiple quoter # values works fine (e.g. s/\Q<>\E// if quoter was [ '<', '>' ]) if( ref $self->{_quoter} eq 'ARRAY') { @@ -83,23 +87,63 @@ sub _rebless { } # Returns an array of table names sub _tables_list { - my $self = shift; + my ($self, $opts) = (shift, shift); my ($table, $type) = @_ ? @_ : ('%', '%'); my $dbh = $self->schema->storage->dbh; my @tables = $dbh->tables(undef, $self->db_schema, $table, $type); - my $qt = qr/\Q$self->{_quoter}\E/; + my $qt = qr/[\Q$self->{_quoter}\E"'`\[\]]/; + + my $all_tables_quoted = (grep /$qt/, @tables) == @tables; - if ($self->{_quoter} && $tables[0] =~ /$qt/) { + if ($self->{_quoter} && $all_tables_quoted) { s/.* $qt (?= .* $qt)//xg for @tables; } else { s/^.*\Q$self->{_namesep}\E// for @tables; } s/$qt//g for @tables; - return @tables; + return $self->_filter_tables(\@tables, $opts); +} + +# apply constraint/exclude and ignore bad tables and views +sub _filter_tables { + my ($self, $tables, $opts) = @_; + + my @tables = @$tables; + my @filtered_tables; + + $opts ||= {}; + my $constraint = $opts->{constraint}; + my $exclude = $opts->{exclude}; + + @tables = grep { /$constraint/ } @$tables if defined $constraint; + @tables = grep { ! /$exclude/ } @$tables if defined $exclude; + + for my $table (@tables) { + eval { + my $sth = $self->_sth_for($table, undef, \'1 = 0'); + $sth->execute; + }; + if (not $@) { + push @filtered_tables, $table; + } + else { + warn "Bad table or view '$table', ignoring: $@\n"; + local $@; + eval { + my $schema = $self->schema; + # in older DBIC it's a private method + my $unregister = $schema->can('unregister_source') + || $schema->can('_unregister_source'); + $schema->$unregister($self->_table2moniker($table)); + }; + } + } + + return @filtered_tables; } =head2 load @@ -116,12 +160,9 @@ sub load { $self->next::method(@_); } -# Returns an arrayref of column names -sub _table_columns { +sub _table_as_sql { my ($self, $table) = @_; - my $dbh = $self->schema->storage->dbh; - if($self->{db_schema}) { $table = $self->{db_schema} . $self->{_namesep} . $self->_quote_table_name($table); @@ -129,9 +170,27 @@ sub _table_columns { $table = $self->_quote_table_name($table); } - my $sth = $dbh->prepare($self->schema->storage->sql_maker->select(\$table, undef, \'1 = 0')); + return $table; +} + +sub _sth_for { + my ($self, $table, $fields, $where) = @_; + + my $dbh = $self->schema->storage->dbh; + + my $sth = $dbh->prepare($self->schema->storage->sql_maker + ->select(\$self->_table_as_sql($table), $fields, $where)); + + return $sth; +} + +# Returns an arrayref of column names +sub _table_columns { + my ($self, $table) = @_; + + my $sth = $self->_sth_for($table, undef, \'1 = 0'); $sth->execute; - my $retval = \@{$sth->{NAME_lc}}; + my $retval = $self->preserve_case ? \@{$sth->{NAME}} : \@{$sth->{NAME_lc}}; $sth->finish; $retval; @@ -143,7 +202,7 @@ sub _table_pk_info { my $dbh = $self->schema->storage->dbh; - my @primary = map { lc } $dbh->primary_key('', $self->db_schema, $table); + my @primary = map { $self->_lc($_) } $dbh->primary_key('', $self->db_schema, $table); s/\Q$self->{_quoter}\E//g for @primary; return \@primary; @@ -233,46 +292,65 @@ sub _columns_info_for { if ($dbh->can('column_info')) { my %result; eval { - my $sth = $dbh->column_info( undef, $self->db_schema, $table, '%' ); + my $sth = eval { local $SIG{__WARN__} = sub {}; $dbh->column_info( undef, $self->db_schema, $table, '%' ); }; while ( my $info = $sth->fetchrow_hashref() ){ - my %column_info; - $column_info{data_type} = $info->{TYPE_NAME}; - $column_info{size} = $info->{COLUMN_SIZE}; - $column_info{is_nullable} = $info->{NULLABLE} ? 1 : 0; - $column_info{default_value} = $info->{COLUMN_DEF}; + my $column_info = {}; + $column_info->{data_type} = lc $info->{TYPE_NAME}; + + my $size = $info->{COLUMN_SIZE}; + + if (defined $size && defined $info->{DECIMAL_DIGITS}) { + $column_info->{size} = [$size, $info->{DECIMAL_DIGITS}]; + } + elsif (defined $size) { + $column_info->{size} = $size; + } + + $column_info->{is_nullable} = $info->{NULLABLE} ? 1 : 0; + $column_info->{default_value} = $info->{COLUMN_DEF} if defined $info->{COLUMN_DEF}; my $col_name = $info->{COLUMN_NAME}; $col_name =~ s/^\"(.*)\"$/$1/; - my $extra_info = $self->_extra_column_info($info) || {}; + my $extra_info = $self->_extra_column_info( + $table, $col_name, $column_info, $info + ) || {}; + $column_info = { %$column_info, %$extra_info }; - $result{$col_name} = { %column_info, %$extra_info }; + $result{$col_name} = $column_info; } $sth->finish; }; return \%result if !$@ && scalar keys %result; } - if($self->db_schema) { - $table = $self->db_schema . $self->{_namesep} . $table; - } my %result; - my $sth = $dbh->prepare($self->schema->storage->sql_maker->select($table, undef, \'1 = 0')); + my $sth = $self->_sth_for($table, undef, \'1 = 0'); $sth->execute; - my @columns = @{$sth->{NAME_lc}}; + my @columns = @{ $self->preserve_case ? $sth->{NAME} : $sth->{NAME_lc} }; for my $i ( 0 .. $#columns ){ - my %column_info; - $column_info{data_type} = $sth->{TYPE}->[$i]; - $column_info{size} = $sth->{PRECISION}->[$i]; - $column_info{is_nullable} = $sth->{NULLABLE}->[$i] ? 1 : 0; - - if ($column_info{data_type} =~ m/^(.*?)\((.*?)\)$/) { - $column_info{data_type} = $1; - $column_info{size} = $2; + my $column_info = {}; + $column_info->{data_type} = lc $sth->{TYPE}->[$i]; + + my $size = $sth->{PRECISION}[$i]; + + if (defined $size && defined $sth->{SCALE}[$i]) { + $column_info->{size} = [$size, $sth->{SCALE}[$i]]; } + elsif (defined $size) { + $column_info->{size} = $size; + } + + $column_info->{is_nullable} = $sth->{NULLABLE}->[$i] ? 1 : 0; - my $extra_info = $self->_extra_column_info($table, $columns[$i], $sth, $i) || {}; + if ($column_info->{data_type} =~ m/^(.*?)\((.*?)\)$/) { + $column_info->{data_type} = $1; + $column_info->{size} = $2; + } - $result{$columns[$i]} = { %column_info, %$extra_info }; + my $extra_info = $self->_extra_column_info($table, $columns[$i], $column_info) || {}; + $column_info = { %$column_info, %$extra_info }; + + $result{$columns[$i]} = $column_info; } $sth->finish; @@ -280,10 +358,10 @@ sub _columns_info_for { my $colinfo = $result{$col}; my $type_num = $colinfo->{data_type}; my $type_name; - if(defined $type_num && $dbh->can('type_info')) { + if(defined $type_num && $type_num =~ /^\d+\z/ && $dbh->can('type_info')) { my $type_info = $dbh->type_info($type_num); $type_name = $type_info->{TYPE_NAME} if $type_info; - $colinfo->{data_type} = $type_name if $type_name; + $colinfo->{data_type} = lc $type_name if $type_name; } } @@ -298,6 +376,16 @@ sub _extra_column_info {} L +=head1 AUTHOR + +See L and L. + +=head1 LICENSE + +This library is free software; you can redistribute it and/or modify it under +the same terms as Perl itself. + =cut 1; +# vim:et sts=4 sw=4 tw=0: