Commit | Line | Data |
18fca96a |
1 | package DBIx::Class::Schema::Loader::Pg; |
a78e3fed |
2 | |
3 | use strict; |
18fca96a |
4 | use base 'DBIx::Class::Schema::Loader::Generic'; |
a78e3fed |
5 | use Carp; |
6 | |
a78e3fed |
7 | =head1 NAME |
8 | |
18fca96a |
9 | DBIx::Class::Schema::Loader::Pg - DBIx::Class::Schema::Loader Postgres Implementation. |
a78e3fed |
10 | |
11 | =head1 SYNOPSIS |
12 | |
18fca96a |
13 | use DBIx::Class::Schema::Loader; |
a78e3fed |
14 | |
18fca96a |
15 | # $loader is a DBIx::Class::Schema::Loader::Pg |
16 | my $loader = DBIx::Class::Schema::Loader->new( |
a78e3fed |
17 | dsn => "dbi:Pg:dbname=dbname", |
18 | user => "postgres", |
19 | password => "", |
a78e3fed |
20 | ); |
a78e3fed |
21 | |
22 | =head1 DESCRIPTION |
23 | |
18fca96a |
24 | See L<DBIx::Class::Schema::Loader>. |
a78e3fed |
25 | |
26 | =cut |
27 | |
28 | sub _db_classes { |
29 | return qw/DBIx::Class::PK::Auto::Pg/; |
30 | } |
31 | |
32 | sub _tables { |
a4a19f3c |
33 | my $class = shift; |
34 | my $dbh = $class->storage->dbh; |
a78e3fed |
35 | |
36 | # This is split out to avoid version parsing errors... |
37 | my $is_dbd_pg_gte_131 = ( $DBD::Pg::VERSION >= 1.31 ); |
38 | my @tables = $is_dbd_pg_gte_131 ? |
a4a19f3c |
39 | $dbh->tables( undef, $class->loader_data->{_db_schema}, "", "table", { noprefix => 1, pg_noprefix => 1 } ) |
a78e3fed |
40 | : $dbh->tables; |
41 | |
a78e3fed |
42 | s/"//g for @tables; |
43 | return @tables; |
44 | } |
45 | |
46 | sub _table_info { |
a4a19f3c |
47 | my ( $class, $table ) = @_; |
48 | my $dbh = $class->storage->dbh; |
a78e3fed |
49 | |
a4a19f3c |
50 | my $sth = $dbh->column_info(undef, $class->loader_data->{_db_schema}, $table, undef); |
a78e3fed |
51 | my @cols = map { $_->[3] } @{ $sth->fetchall_arrayref }; |
52 | s/"//g for @cols; |
53 | |
a4a19f3c |
54 | my @primary = $dbh->primary_key(undef, $class->loader_data->{_db_schema}, $table); |
a78e3fed |
55 | |
56 | s/"//g for @primary; |
57 | |
58 | return ( \@cols, \@primary ); |
59 | } |
60 | |
61 | =head1 SEE ALSO |
62 | |
18fca96a |
63 | L<DBIx::Class::Schema::Loader> |
a78e3fed |
64 | |
65 | =cut |
66 | |
67 | 1; |