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::ForeignKey';
10 use aliased 'SQL::Translator::Object::Index';
11 use aliased 'SQL::Translator::Object::PrimaryKey';
12 use aliased 'SQL::Translator::Object::Table';
13 use aliased 'SQL::Translator::Object::View';
19 default => sub { shift->dbh->get_info(29) || q{"} }
26 default => sub { shift->dbh->get_info(41) || '.' }
29 has 'schema_name' => (
36 has 'catalog_name' => (
46 my $dbtype = $self->dbh->get_info($GetInfoType{SQL_DBMS_NAME}) || $self->dbh->{Driver}{Name};
48 my $class = __PACKAGE__ . '::'. $dbtype;
49 Class::MOP::load_class($class);
50 $class->meta->apply($self);
53 sub _is_auto_increment { 0 }
55 sub _column_default_value {
57 my $column_info = shift;
59 return $column_info->{COLUMN_DEF};
66 my $sth = $self->dbh->table_info($self->catalog_name, $self->schema_name, '%', 'TABLE,VIEW');
67 while (my $table_info = $sth->fetchrow_hashref) {
68 if ($table_info->{TABLE_TYPE} eq 'TABLE') {
69 my $table = Table->new({ name => $table_info->{TABLE_NAME} });
70 $schema->add_table($table);
71 $self->_add_columns($table);
72 $self->_add_primary_key($table);
73 $self->_add_indexes($table);
75 elsif ($table_info->{TABLE_TYPE} eq 'VIEW') {
76 my $sql = $self->_get_view_sql($table_info->{TABLE_NAME});
77 my $view = View->new({ name => $table_info->{TABLE_NAME}, sql => $sql });
78 $schema->add_view($view);
79 $self->_add_columns($view);
82 $self->_add_foreign_keys($schema->get_table($_), $schema) for $schema->table_ids;
89 my $sth = $self->dbh->column_info($self->catalog_name, $self->schema_name, $table->name, '%');
90 while (my $column_info = $sth->fetchrow_hashref) {
91 my $column = Column->new({ name => $column_info->{COLUMN_NAME},
92 data_type => $column_info->{DATA_TYPE},
93 size => $column_info->{COLUMN_SIZE},
94 default_value => $self->_column_default_value($column_info),
95 is_auto_increment => $self->_is_auto_increment($column_info),
96 is_nullable => $column_info->{NULLABLE},
98 $table->add_column($column);
102 sub _add_primary_key {
106 my $pk_info = $self->dbh->primary_key_info($self->catalog_name, $self->schema_name, $table->name);
107 my ($pk_name, @pk_cols);
108 while (my $pk_col = $pk_info->fetchrow_hashref) {
109 $pk_name = $pk_col->{PK_NAME};
110 push @pk_cols, $pk_col->{COLUMN_NAME};
112 my $pk = PrimaryKey->new({ name => $pk_name });
113 $pk->add_column($table->get_column($_)) for @pk_cols;
114 $table->add_index($pk);
117 sub _add_foreign_keys {
122 my $fk_info = $self->dbh->foreign_key_info($self->catalog_name, $self->schema_name, $table->name, $self->catalog_name, $self->schema_name, undef);
123 return unless $fk_info;
126 while (my $fk_col = $fk_info->fetchrow_hashref) {
127 my $fk_name = $fk_col->{FK_NAME};
129 push @{$fk_data->{$fk_name}{columns}}, $fk_col->{FK_COLUMN_NAME};
130 $fk_data->{$fk_name}{table} = $fk_col->{FK_TABLE_NAME};
131 $fk_data->{$fk_name}{uk} = $schema->get_table($fk_col->{UK_TABLE_NAME})->get_index($fk_col->{UK_NAME});
134 foreach my $fk_name (keys %$fk_data) {
135 my $fk = ForeignKey->new({ name => $fk_name, references => $fk_data->{$fk_name}{uk} });
136 $fk->add_column($schema->get_table($fk_data->{$fk_name}{table})->get_column($_)) for @{$fk_data->{$fk_name}{columns}};
137 $table->add_constraint($fk);
145 my $index_info = $self->dbh->statistics_info($self->catalog_name, $self->schema_name, $table->name, 1, 0);
147 my ($index_name, $index_type, @index_cols);
148 while (my $index_col = $index_info->fetchrow_hashref) {
149 $index_name = $index_col->{INDEX_NAME};
150 $index_type = $index_col->{NON_UNIQUE} ? 'NON_UNIQUE' : 'UNIQUE';
151 push @index_cols, $index_col->{COLUMN_NAME};
153 my $index = Index->new({ name => $index_name, type => $index_type });
154 $index->add_column($table->get_column($_)) for @index_cols;
155 $table->add_index($index);