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