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=5940de2f470a6fabba1695062c51faaa485685fa;hpb=45fa82889e637f9382eb2890b44fc58ef122de25;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 5940de2..2a7b529 100644 --- a/lib/DBIx/Class/Storage/DBI/Pg.pm +++ b/lib/DBIx/Class/Storage/DBI/Pg.pm @@ -3,32 +3,108 @@ package DBIx::Class::Storage::DBI::Pg; use strict; use warnings; -use base qw/DBIx::Class::Storage::DBI/; +use base qw/DBIx::Class::Storage::DBI::MultiColumnIn/; +use mro 'c3'; -# __PACKAGE__->load_components(qw/PK::Auto/); +use DBD::Pg qw(:pg_types); + +# 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() + +sub with_deferred_fk_checks { + my ($self, $sub) = @_; + + $self->_get_dbh->do('SET CONSTRAINTS ALL DEFERRED'); + $sub->(); +} sub last_insert_id { my ($self,$source,$col) = @_; my $seq = ($source->column_info($col)->{sequence} ||= $self->get_autoinc_seq($source,$col)); - $self->_dbh->last_insert_id(undef,undef,undef,undef, {sequence => $seq}); + $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_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) = @_; + + # 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 $dbh = $self->_dbh; - my ($schema,$table) = $source->name =~ /^(.+)\.(.+)$/ ? ($1,$2) - : (undef,$source->name); - 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; - return $seq =~ /\./ ? $seq : $info->{TABLE_SCHEM} . "." . $seq; # may need to strip quotes -- see if this works - } + + 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); } sub sqlt_type { @@ -37,6 +113,46 @@ sub sqlt_type { sub datetime_parser_type { return "DateTime::Format::Pg"; } +sub bind_attribute_by_data_type { + my ($self,$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}; + } + else { + return; + } +} + +sub _sequence_fetch { + my ( $self, $type, $seq ) = @_; + my ($id) = $self->_get_dbh->selectrow_array("SELECT nextval('${seq}')"); + return $id; +} + +sub _svp_begin { + my ($self, $name) = @_; + + $self->_get_dbh->pg_savepoint($name); +} + +sub _svp_release { + my ($self, $name) = @_; + + $self->_get_dbh->pg_release($name); +} + +sub _svp_rollback { + my ($self, $name) = @_; + + $self->_get_dbh->pg_rollback_to($name); +} + 1; =head1 NAME @@ -54,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