Commit | Line | Data |
5d567302 |
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 { |
b6b65a3e |
11 | my $self = shift; |
5d567302 |
12 | $self->get_autoinc_seq unless $self->{_autoinc_seq}; |
66d9ef6b |
13 | $self->result_source->storage->dbh->last_insert_id(undef,undef,undef,undef, |
5d567302 |
14 | {sequence=>$self->{_autoinc_seq}}); |
15 | } |
16 | |
17 | sub get_autoinc_seq { |
b6b65a3e |
18 | my $self = shift; |
97cc0025 |
19 | |
20 | # return the user-defined sequence if known |
5b34b2f9 |
21 | if ($self->sequence) { |
22 | return $self->{_autoinc_seq} = $self->sequence; |
23 | } |
97cc0025 |
24 | |
47d2f099 |
25 | my @pri = $self->primary_columns; |
fe2b6875 |
26 | my $dbh = $self->result_source->storage->dbh; |
c5b7d799 |
27 | my ($schema,$table) = $self->table =~ /^(.+)\.(.+)$/ ? ($1,$2) : (undef,$self->table); |
b6b65a3e |
28 | while (my $col = shift @pri) { |
c5b7d799 |
29 | my $info = $dbh->column_info(undef,$schema,$table,$col)->fetchrow_arrayref; |
b6b65a3e |
30 | if (defined $info->[12] and $info->[12] =~ |
31 | /^nextval\('"?([^"']+)"?'::(?:text|regclass)\)/) |
32 | { |
33 | $self->{_autoinc_seq} = $1; |
34 | last; |
35 | } |
36 | } |
5d567302 |
37 | } |
38 | |
39 | 1; |
40 | |
41 | =head1 NAME |
42 | |
eb49d4e3 |
43 | DBIx::Class::PK::Auto::Pg - Automatic primary key class for PostgreSQL |
5d567302 |
44 | |
45 | =head1 SYNOPSIS |
46 | |
eb49d4e3 |
47 | # In your table classes |
48 | __PACKAGE__->load_components(qw/PK::Auto::Pg Core/); |
49 | __PACKAGE__->set_primary_key('id'); |
6718c5f0 |
50 | |
5d567302 |
51 | =head1 DESCRIPTION |
52 | |
eb49d4e3 |
53 | This class implements autoincrements for PostgreSQL. |
5d567302 |
54 | |
55 | =head1 AUTHORS |
56 | |
d00123a3 |
57 | Marcus Ramberg <m.ramberg@cpan.org> |
5d567302 |
58 | |
59 | =head1 LICENSE |
60 | |
61 | You may distribute this code under the same terms as Perl itself. |
62 | |
63 | =cut |