Commit | Line | Data |
4f4fd192 |
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', |
651a0290 |
15 | SQL_FLOAT() => 'numeric', |
4f4fd192 |
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 | } |
287d4603 |
36 | } |
4f4fd192 |
37 | |
38 | print $create_table . ";\n"; |
39 | return (@create, $create_table, @index_defs, @constraint_defs); |
287d4603 |
40 | } |
4f4fd192 |
41 | |
42 | method _create_column(Column $column) { |
43 | my $size = $column->size; |
44 | my $default_value = $column->default_value; |
a584e206 |
45 | $default_value =~ s/^CURRENT_TIMESTAMP/now()/i if $default_value; |
4f4fd192 |
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; |
287d4603 |
57 | } |
4f4fd192 |
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; |
287d4603 |
72 | } |
287d4603 |
73 | } |