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