lib/SQL/Translator/Producer/SQL/PostgreSQL.pm

   1 use MooseX::Declare;
   2 role SQL::Translator::Producer::SQL::PostgreSQL {
   3     use SQL::Translator::Constants qw(:sqlt_types);
   4     use SQL::Translator::Types qw(Column Index Table);
   5
   6     my %data_type_mapping = (
   7         SQL_LONGVARCHAR() => 'text',
   8         SQL_TIMESTAMP()   => 'timestamp',
   9         SQL_TYPE_TIMESTAMP() => 'timestamp without time zone',
  10         SQL_TYPE_TIMESTAMP_WITH_TIMEZONE() => 'timestamp',
  11         SQL_INTEGER()     => 'integer',
  12         SQL_CHAR()        => 'character',
  13         SQL_VARCHAR()     => 'varchar',
  14         SQL_BIGINT()      => 'bigint',
  15         SQL_FLOAT()       => 'numeric',
  16     );
  17
  18     method _create_table(Table $table) {
  19         my $pg_version = 0;
  20
  21         my $create_table;
  22         my (@create, @column_defs, @index_defs, @constraint_defs);
  23
  24         $create_table .= 'DROP TABLE '   . $table->name . ";\n" if $self->drop_table;
  25         $create_table .= 'CREATE TABLE ' . $table->name . " (\n";
  26
  27         push @column_defs, $self->_create_column($_) for values %{$table->columns};
  28         $create_table .= join(",\n", map { '  ' . $_ } @column_defs ) . "\n)";
  29
  30         foreach my $index (values %{$table->indexes}) {
  31             if ($index->type eq 'NORMAL') {
  32                 push @index_defs, $self->_create_index($index, $table);
  33             } else {
  34                 push @constraint_defs, $self->_create_index($index);
  35             }
  36         }
  37
  38         print $create_table . ";\n";
  39         return (@create, $create_table, @index_defs, @constraint_defs);
  40     }
  41
  42     method _create_column(Column $column) {
  43         my $size = $column->size;
  44         my $default_value = $column->default_value;
  45         $default_value =~ s/^CURRENT_TIMESTAMP/now()/i if $default_value;
  46
  47         my $column_def;
  48         $column_def  = $column->name . ' ';
  49         $column_def .= defined $data_type_mapping{$column->data_type}
  50                        ? $data_type_mapping{$column->data_type}
  51                        : $column->data_type;
  52         $column_def .= '(' . $column->size . ')' if $size;
  53         $column_def .= ' NOT NULL' unless $column->is_nullable;
  54         $column_def .= ' PRIMARY KEY' if $column->is_auto_increment;
  55         $column_def .= ' DEFAULT ' . $default_value if $column->default_value && !$column->is_auto_increment;
  56         $column_def;
  57     }
  58
  59     method _create_index(Index $index, Table $table?) {
  60         my $index_def;
  61         if ($index->type eq 'PRIMARY_KEY') {
  62             $index_def = 'CONSTRAINT ' . $index->name .  ' PRIMARY KEY ' .  '(' . (join ', ', map { $_->name } values %{$index->columns}) . ')';
  63         }
  64         elsif ($index->type eq 'UNIQUE') {
  65             $index_def = 'CONSTRAINT ' . $index->name .  ' UNIQUE ' .  '(' . (join ', ', map { $_->name } values %{$index->columns}) . ')';
  66         }
  67         elsif ($index->type eq 'NORMAL') {
  68             $index_def = 'CREATE INDEX ' . $index->name . ' ON ' . $table->name . '('.  (join ', ', map { $_->name } values %{$index->columns}) . ')';
  69         }
  70
  71         $index_def;
  72     }
  73 }