X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FStorage%2FDBI%2FPg.pm;h=2a7b529d06cbc00f66ced22e1c1ca67830cb7294;hb=d3944540917a76cd054d1bf4ed98f2571c077b2d;hp=bd28e02232cf584a26a8e3c43c4d4db844cc4b51;hpb=d6feb60f4c904b02c0f4367f0aca5d9dc8455f83;p=dbsrgits%2FDBIx-Class-Historic.git diff --git a/lib/DBIx/Class/Storage/DBI/Pg.pm b/lib/DBIx/Class/Storage/DBI/Pg.pm index bd28e02..2a7b529 100644 --- a/lib/DBIx/Class/Storage/DBI/Pg.pm +++ b/lib/DBIx/Class/Storage/DBI/Pg.pm @@ -3,19 +3,20 @@ package DBIx::Class::Storage::DBI::Pg; use strict; use warnings; -use DBD::Pg qw(:pg_types); +use base qw/DBIx::Class::Storage::DBI::MultiColumnIn/; +use mro 'c3'; -use base qw/DBIx::Class::Storage::DBI/; +use DBD::Pg qw(:pg_types); -# __PACKAGE__->load_components(qw/PK::Auto/); +# Ask for a DBD::Pg with array support +warn "DBD::Pg 2.9.2 or greater is strongly recommended\n" + if ($DBD::Pg::VERSION < 2.009002); # pg uses (used?) version::qv() -# Warn about problematic versions of DBD::Pg -warn "DBD::Pg 1.49 is strongly recommended" - if ($DBD::Pg::VERSION < 1.49); +sub with_deferred_fk_checks { + my ($self, $sub) = @_; -sub _dbh_last_insert_id { - my ($self, $dbh, $seq) = @_; - $dbh->last_insert_id(undef, undef, undef, undef, {sequence => $seq}); + $self->_get_dbh->do('SET CONSTRAINTS ALL DEFERRED'); + $sub->(); } sub last_insert_id { @@ -24,30 +25,84 @@ sub last_insert_id { $self->throw_exception("could not fetch primary key for " . $source->name . ", could not " . "get autoinc sequence for $col (check that table and column specifications are correct " . "and in the correct case)") unless defined $seq; - $self->dbh_do('_dbh_last_insert_id', $seq); + + $self->_dbh_last_insert_id ($self->_dbh, $seq); +} + +# there seems to be absolutely no reason to have this as a separate method, +# but leaving intact in case someone is already overriding it +sub _dbh_last_insert_id { + my ($self, $dbh, $seq) = @_; + $dbh->last_insert_id(undef, undef, undef, undef, {sequence => $seq}); +} + + +sub _get_pg_search_path { + my ($self,$dbh) = @_; + # cache the search path as ['schema','schema',...] in the storage + # obj + $self->{_pg_search_path} ||= do { + my @search_path; + my ($sp_string) = $dbh->selectrow_array('SHOW search_path'); + while( $sp_string =~ s/("[^"]+"|[^,]+),?// ) { + unless( defined $1 and length $1 ) { + $self->throw_exception("search path sanity check failed: '$1'") + } + push @search_path, $1; + } + \@search_path + }; } sub _dbh_get_autoinc_seq { my ($self, $dbh, $schema, $table, @pri) = @_; - while (my $col = shift @pri) { - my $info = $dbh->column_info(undef,$schema,$table,$col)->fetchrow_hashref; - if(defined $info->{COLUMN_DEF} and - $info->{COLUMN_DEF} =~ /^nextval\(+'([^']+)'::(?:text|regclass)\)/) { - my $seq = $1; - # may need to strip quotes -- see if this works - return $seq =~ /\./ ? $seq : $info->{TABLE_SCHEM} . "." . $seq; - } + # get the list of postgres schemas to search. if we have a schema + # specified, use that. otherwise, use the search path + my @search_path; + if( defined $schema and length $schema ) { + @search_path = ( $schema ); + } else { + @search_path = @{ $self->_get_pg_search_path($dbh) }; + } + + foreach my $search_schema (@search_path) { + foreach my $col (@pri) { + my $info = $dbh->column_info(undef,$search_schema,$table,$col)->fetchrow_hashref; + if($info) { + # if we get here, we have definitely found the right + # column. + if( defined $info->{COLUMN_DEF} and + $info->{COLUMN_DEF} + =~ /^nextval\(+'([^']+)'::(?:text|regclass)\)/i + ) { + my $seq = $1; + return $seq =~ /\./ ? $seq : $info->{TABLE_SCHEM} . "." . $seq; + } else { + # we have found the column, but cannot figure out + # the nextval seq + return; + } + } + } } return; } sub get_autoinc_seq { my ($self,$source,$col) = @_; - + my @pri = $source->primary_columns; - my ($schema,$table) = $source->name =~ /^(.+)\.(.+)$/ ? ($1,$2) - : (undef,$source->name); + + my $schema; + my $table = $source->name; + + if (ref $table eq 'SCALAR') { + $table = $$table; + } + elsif ($table =~ /^(.+)\.(.+)$/) { + ($schema, $table) = ($1, $2); + } $self->dbh_do('_dbh_get_autoinc_seq', $schema, $table, @pri); } @@ -63,8 +118,9 @@ sub bind_attribute_by_data_type { my $bind_attributes = { bytea => { pg_type => DBD::Pg::PG_BYTEA }, + blob => { pg_type => DBD::Pg::PG_BYTEA }, }; - + if( defined $bind_attributes->{$data_type} ) { return $bind_attributes->{$data_type}; } @@ -75,26 +131,26 @@ sub bind_attribute_by_data_type { sub _sequence_fetch { my ( $self, $type, $seq ) = @_; - my ($id) = $self->dbh->selectrow_array("SELECT nextval('${seq}')"); + my ($id) = $self->_get_dbh->selectrow_array("SELECT nextval('${seq}')"); return $id; } sub _svp_begin { my ($self, $name) = @_; - $self->dbh->pg_savepoint($name); + $self->_get_dbh->pg_savepoint($name); } sub _svp_release { my ($self, $name) = @_; - $self->dbh->pg_release($name); + $self->_get_dbh->pg_release($name); } sub _svp_rollback { my ($self, $name) = @_; - $self->dbh->pg_rollback_to($name); + $self->_get_dbh->pg_rollback_to($name); } 1; @@ -114,9 +170,26 @@ DBIx::Class::Storage::DBI::Pg - Automatic primary key class for PostgreSQL This class implements autoincrements for PostgreSQL. +=head1 POSTGRESQL SCHEMA SUPPORT + +This supports multiple PostgreSQL schemas, with one caveat: for +performance reasons, the schema search path is queried the first time it is +needed and CACHED for subsequent uses. + +For this reason, you should do any necessary manipulation of the +PostgreSQL search path BEFORE instantiating your schema object, or as +part of the on_connect_do option to connect(), for example: + + my $schema = My::Schema->connect + ( $dsn,$user,$pass, + { on_connect_do => + [ 'SET search_path TO myschema, foo, public' ], + }, + ); + =head1 AUTHORS -Marcus Ramberg +See L =head1 LICENSE