lib/SQL/Translator/Parser/DBI.pm

   1 package SQL::Translator::Parser::DBI;
   2 use namespace::autoclean;
   3 use Moose::Role;
   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';
  14
  15 has 'quoter' => (
  16     is => 'rw',
  17     isa => Str,
  18     lazy => 1,
  19     default => sub { shift->dbh->get_info(29) || q{"} }
  20 );
  21
  22 has 'namesep' => (
  23     is => 'rw',
  24     isa => Str,
  25     lazy => 1,
  26     default => sub { shift->dbh->get_info(41) || '.' }
  27 );
  28
  29 has 'schema_name' => (
  30     is => 'rw',
  31     isa => Maybe[Str],
  32     lazy => 1,
  33     default => undef
  34 );
  35
  36 has 'catalog_name' => (
  37     is => 'rw',
  38     isa => Maybe[Str],
  39     lazy => 1,
  40     default => undef
  41 );
  42
  43 sub _subclass {
  44     my $self = shift;
  45
  46     my $dbtype = $self->dbh->get_info($GetInfoType{SQL_DBMS_NAME}) || $self->dbh->{Driver}{Name};
  47
  48     my $class = __PACKAGE__ . '::'. $dbtype;
  49     Class::MOP::load_class($class);
  50     $class->meta->apply($self);
  51 }
  52
  53 sub _is_auto_increment { 0 }
  54
  55 sub _column_default_value {
  56     my $self = shift;
  57     my $column_info = shift;
  58
  59     return $column_info->{COLUMN_DEF};
  60 }
  61
  62 sub _add_tables {
  63     my $self = shift;
  64     my $schema = shift;
  65
  66     my $sth = $self->dbh->table_info($self->catalog_name, $self->schema_name, '%', "TABLE,VIEW,'LOCAL TEMPORARY','GLOBAL TEMPORARY'");
  67     while (my $table_info = $sth->fetchrow_hashref) {
  68         if ($table_info->{TABLE_TYPE} =~ /^(TABLE|LOCAL TEMPORARY|GLOBAL TEMPORARY)$/) {
  69             my $temp = $table_info->{TABLE_TYPE} =~ /TEMPORARY$/ ? 1 : 0;
  70             my $table = Table->new({ name => $table_info->{TABLE_NAME}, temporary => $temp });
  71             $schema->add_table($table);
  72             $self->_add_columns($table);
  73             $self->_add_primary_key($table);
  74             $self->_add_indexes($table);
  75         }
  76         elsif ($table_info->{TABLE_TYPE} eq 'VIEW') {
  77             my $sql = $self->_get_view_sql($table_info->{TABLE_NAME});
  78             my $view = View->new({ name => $table_info->{TABLE_NAME}, sql => $sql });
  79             $schema->add_view($view);
  80             $self->_add_columns($view);
  81         }
  82     }
  83     $self->_add_foreign_keys($schema->get_table($_), $schema) for $schema->table_ids;
  84 }
  85
  86 sub _add_columns {
  87     my $self  = shift;
  88     my $table = shift;
  89
  90     my $sth = $self->dbh->column_info($self->catalog_name, $self->schema_name, $table->name, '%');
  91     while (my $column_info = $sth->fetchrow_hashref) {
  92         my $column = Column->new({ name => $column_info->{COLUMN_NAME},
  93                                    data_type => $column_info->{DATA_TYPE},
  94                                    size => $column_info->{COLUMN_SIZE},
  95                                    default_value => $self->_column_default_value($column_info),
  96                                    is_auto_increment => $self->_is_auto_increment($column_info),
  97                                    is_nullable => $column_info->{NULLABLE},
  98                                  });
  99         $table->add_column($column);
 100     }
 101 }
 102
 103 sub _add_primary_key {
 104     my $self = shift;
 105     my $table = shift;
 106
 107     my $pk_info = $self->dbh->primary_key_info($self->catalog_name, $self->schema_name, $table->name);
 108     my ($pk_name, @pk_cols);
 109     while (my $pk_col = $pk_info->fetchrow_hashref) {
 110         $pk_name = $pk_col->{PK_NAME};
 111         push @pk_cols, $pk_col->{COLUMN_NAME};
 112     }
 113     my $pk = PrimaryKey->new({ name => $pk_name });
 114     $pk->add_column($table->get_column($_)) for @pk_cols;
 115     $table->add_index($pk);
 116 }
 117
 118 sub _add_foreign_keys {
 119     my $self = shift;
 120     my $table = shift;
 121     my $schema = shift;
 122
 123     my $fk_info = $self->dbh->foreign_key_info($self->catalog_name, $self->schema_name, $table->name, $self->catalog_name, $self->schema_name, undef);
 124     return unless $fk_info;
 125
 126     my $fk_data;
 127     while (my $fk_col = $fk_info->fetchrow_hashref) {
 128         my $fk_name = $fk_col->{FK_NAME};
 129
 130         push @{$fk_data->{$fk_name}{columns}}, $fk_col->{FK_COLUMN_NAME};
 131         $fk_data->{$fk_name}{table} = $fk_col->{FK_TABLE_NAME};
 132         $fk_data->{$fk_name}{uk} = $schema->get_table($fk_col->{UK_TABLE_NAME})->get_index($fk_col->{UK_NAME});
 133     }
 134
 135     foreach my $fk_name (keys %$fk_data) {
 136         my $fk = ForeignKey->new({ name => $fk_name, references => $fk_data->{$fk_name}{uk} });
 137         $fk->add_column($schema->get_table($fk_data->{$fk_name}{table})->get_column($_)) for @{$fk_data->{$fk_name}{columns}};
 138         $table->add_constraint($fk);
 139     }
 140 }
 141
 142 sub _add_indexes {
 143     my $self = shift;
 144     my $table = shift;
 145
 146     my $index_info = $self->dbh->statistics_info($self->catalog_name, $self->schema_name, $table->name, 1, 0);
 147
 148     my ($index_name, $index_type, @index_cols);
 149     while (my $index_col = $index_info->fetchrow_hashref) {
 150         $index_name = $index_col->{INDEX_NAME};
 151         $index_type = $index_col->{NON_UNIQUE} ? 'NORMAL' : 'UNIQUE';
 152         push @index_cols, $index_col->{COLUMN_NAME};
 153     }
 154     return if $table->exists_index($index_name);
 155     my $index = Index->new({ name => $index_name, type => $index_type });
 156     $index->add_column($table->get_column($_)) for @index_cols;
 157     $table->add_index($index);
 158 }
 159
 160 1;