Fixed user-defined sequence assignment
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / PK / Auto / Pg.pm
1 package DBIx::Class::PK::Auto::Pg;
2
3 use strict;
4 use warnings;
5
6 use base qw/DBIx::Class/;
7
8 __PACKAGE__->load_components(qw/PK::Auto/);
9
10 sub last_insert_id {
11   my $self=shift;
12   $self->get_autoinc_seq unless $self->{_autoinc_seq};
13   $self->storage->dbh->last_insert_id(undef,undef,undef,undef,
14     {sequence=>$self->{_autoinc_seq}});
15 }
16
17 sub get_autoinc_seq {
18   my $self=shift;
19   
20   # return the user-defined sequence if known
21   if ($self->sequence) {
22     return $self->{_autoinc_seq} = $self->sequence;
23   }
24   
25   my $dbh= $self->storage->dbh;
26     my $sth     = $dbh->column_info( undef, undef, $self->_table_name, '%');
27     while (my $foo = $sth->fetchrow_arrayref){
28       if(defined $foo->[12] && $foo->[12] =~ /^nextval/) {
29         ($self->{_autoinc_seq}) = $foo->[12] =~ 
30           m!^nextval\('"?([^"']+)"?'::text\)!;
31       }
32     }
33 }
34
35 1;
36
37 =head1 NAME 
38
39 DBIx::Class::PK::Auto::Pg- Automatic Primary Key class for Postgresql
40
41 =head1 SYNOPSIS
42
43 =head1 DESCRIPTION
44
45 This class implements autoincrements for Postgresql.
46
47 =head1 AUTHORS
48
49 Marcus Ramberg <m.ramberg@cpan.org>
50
51 =head1 LICENSE
52
53 You may distribute this code under the same terms as Perl itself.
54
55 =cut
56