1 package SQL::Translator::Parser::DBI;
3 use MooseX::Types::Moose qw(Maybe Str);
4 use DBI::Const::GetInfoType;
5 use DBI::Const::GetInfo::ANSI;
6 use DBI::Const::GetInfoReturn;
7 use SQL::Translator::Object::Column;
8 use SQL::Translator::Object::Index;
9 use SQL::Translator::Object::Table;
10 use SQL::Translator::Object::View;
17 default => sub { shift->dbh->get_info(29) || q{"} }
25 default => sub { shift->dbh->get_info(41) || '.' }
28 has 'schema_name' => (
36 has 'catalog_name' => (
49 my $dbtype = $self->dbh->get_info($GetInfoType{SQL_DBMS_NAME}) || $self->dbh->{Driver}{Name};
51 my $class = __PACKAGE__ . '::'. $dbtype;
52 Class::MOP::load_class($class);
53 $class->meta->apply($self);
60 my $sth = $self->dbh->table_info($self->catalog_name, $self->schema_name, '%', 'TABLE,VIEW');
61 while (my $table_info = $sth->fetchrow_hashref) {
62 if ($table_info->{TABLE_TYPE} eq 'TABLE') {
63 my $table = SQL::Translator::Object::Table->new({ name => $table_info->{TABLE_NAME} });
64 $schema->add_table($table);
65 $self->_add_columns($table);
66 $self->_add_primary_key($table);
68 elsif ($table_info->{TABLE_TYPE} eq 'VIEW') {
69 my $sql = $self->_get_view_sql($table_info->{TABLE_NAME});
70 $schema->add_view(SQL::Translator::Object::View->new({ name => $table_info->{TABLE_NAME}, sql => $sql }));
79 my $sth = $self->dbh->column_info($self->catalog_name, $self->schema_name, $table->name, '%');
80 while (my $col_info = $sth->fetchrow_hashref) {
81 my $column = SQL::Translator::Object::Column->new({ name => $col_info->{COLUMN_NAME},
82 data_type => $col_info->{TYPE_NAME},
83 size => $col_info->{COLUMN_SIZE},
84 default_value => $col_info->{COLUMN_DEF},
85 is_nullable => $col_info->{NULLABLE}, });
86 $table->add_column($column);
90 sub _add_primary_key {
94 my $pk_info = $self->dbh->primary_key_info($self->catalog_name, $self->schema_name, $table->name);
95 my ($pk_name, @pk_cols);
96 while (my $pk_col = $pk_info->fetchrow_hashref) {
97 $pk_name = $pk_col->{PK_NAME};
98 push @pk_cols, $pk_col->{COLUMN_NAME};
100 my $index = SQL::Translator::Object::Index->new({ name => $pk_name, type => 'PRIMARY_KEY' });
101 $index->add_column($table->get_column($_)) for @pk_cols;
102 $table->add_index($index);