1 package SQL::Translator::Generator::DDL::SQLServer;
3 # AUTHOR: Arthur Axel fREW Schmidt
4 # Copyright: Same as Perl 5
7 use SQL::Translator::Generator::Utils;
8 use SQL::Translator::Schema::Constants;
10 with 'SQL::Translator::Generator::Role::Quote';
11 with 'SQL::Translator::Generator::Role::DDL';
13 sub quote_chars { [qw([ ])] }
16 sub _build_numeric_types {
22 sub _build_unquoted_defaults {
35 sub _build_sizeless_types {
37 qw( tinyint smallint int integer bigint text bit image datetime ) }
41 my ($self, $field) = @_;
43 return join ' ', $self->field_name($field), ($self->field_type($field)||die 'type is required'),
44 $self->field_autoinc($field),
45 $self->field_nullable($field),
46 $self->field_default($field),
49 sub field_autoinc { ( $_[1]->is_auto_increment ? 'IDENTITY' : () ) }
51 sub primary_key_constraint {
53 $_[0]->quote($_[1]->name || $_[1]->table->name . '_pk') .
55 join( ', ', map $_[0]->quote($_), $_[1]->fields ) .
61 $_[0]->quote($_[1]->name || $_[1]->table->name . '_idx') .
62 ' ON ' . $_[0]->quote($_[1]->table->name) .
63 ' (' . join( ', ', map $_[0]->quote($_), $_[1]->fields ) . ');'
66 sub unique_constraint_single {
67 my ($self, $constraint) = @_;
70 $self->unique_constraint_name($constraint) .
71 ' UNIQUE (' . join( ', ', map $self->quote($_), $constraint->fields ) . ')'
74 sub unique_constraint_name {
75 my ($self, $constraint) = @_;
76 $self->quote($constraint->name || $constraint->table->name . '_uc' )
79 sub unique_constraint_multiple {
80 my ($self, $constraint) = @_;
82 'CREATE UNIQUE NONCLUSTERED INDEX ' .
83 $self->unique_constraint_name($constraint) .
84 ' ON ' . $self->quote($constraint->table->name) . ' (' .
85 join( ', ', $constraint->fields ) . ')' .
86 ' WHERE ' . join( ' AND ',
87 map $self->quote($_->name) . ' IS NOT NULL',
88 grep { $_->is_nullable } $constraint->fields ) . ';'
91 sub foreign_key_constraint {
92 my ($self, $constraint) = @_;
94 my $on_delete = uc ($constraint->on_delete || '');
95 my $on_update = uc ($constraint->on_update || '');
97 # The default implicit constraint action in MSSQL is RESTRICT
98 # but you can not specify it explicitly. Go figure :)
99 for (map uc $_ || '', $on_delete, $on_update) {
100 undef $_ if $_ eq 'RESTRICT'
103 'ALTER TABLE ' . $self->quote($constraint->table->name) .
105 $self->quote($constraint->name || $constraint->table->name . '_fk') .
107 ' (' . join( ', ', map $self->quote($_), $constraint->fields ) . ') REFERENCES '.
108 $self->quote($constraint->reference_table) .
109 ' (' . join( ', ', map $self->quote($_), $constraint->reference_fields ) . ')'
111 $on_delete && $on_delete ne "NO ACTION"
112 ? ' ON DELETE ' . $on_delete
115 $on_update && $on_update ne "NO ACTION"
116 ? ' ON UPDATE ' . $on_update
121 sub enum_constraint_name {
122 my ($self, $field_name) = @_;
123 $self->quote($field_name . '_chk' )
126 sub enum_constraint {
127 my ( $self, $field_name, $vals ) = @_;
130 'CONSTRAINT ' . $self->enum_constraint_name($field_name) .
131 ' CHECK (' . $self->quote($field_name) .
132 ' IN (' . join( ',', map qq('$_'), @$vals ) . '))'
137 my ($self, $table) = @_;
139 (map $self->enum_constraint($_->name, { $_->extra }->{list} || []),
140 grep { 'enum' eq lc $_->data_type } $table->get_fields),
142 (map $self->primary_key_constraint($_),
143 grep { $_->type eq PRIMARY_KEY } $table->get_constraints),
145 (map $self->unique_constraint_single($_),
147 $_->type eq UNIQUE &&
148 !grep { $_->is_nullable } $_->fields
149 } $table->get_constraints),
153 my ($self, $table) = @_;
154 join ( "\n", $self->table_comments($table), '' ) .
156 'CREATE TABLE ' . $self->quote($table->name) . " (\n".
159 $self->fields($table),
160 $self->constraints($table),
163 $self->unique_constraints_multiple($table),
164 $self->indices($table),
168 sub unique_constraints_multiple {
169 my ($self, $table) = @_;
170 (map $self->unique_constraint_multiple($_),
172 $_->type eq UNIQUE &&
173 grep { $_->is_nullable } $_->fields
174 } $table->get_constraints)
178 my ($self, $table) = @_;
179 my $name = $table->name;
180 my $q_name = $self->quote($name);
181 "IF EXISTS (SELECT name FROM sysobjects WHERE name = '$name' AND type = 'U')" .
182 " DROP TABLE $q_name;"
185 sub remove_table_constraints {
186 my ($self, $table) = @_;
187 my $name = $table->name;
188 my $q_name = $self->quote($name);
189 "IF EXISTS (SELECT name FROM sysobjects WHERE name = '$name' AND type = 'U')" .
190 " ALTER TABLE $q_name NOCHECK CONSTRAINT all;"
194 my ($self, $schema) = shift;
196 if ($self->add_drop_table) {
197 my @tables = sort { $b->order <=> $a->order } $schema->get_tables;
199 ( $self->add_comments ? (
201 '-- Turn off constraints',
205 (map $self->remove_table_constraints($_), @tables),
206 ( $self->add_comments ? (
212 (map $self->drop_table($_), @tables),
218 sub foreign_key_constraints {
219 my ($self, $schema) = @_;
220 ( map $self->foreign_key_constraint($_),
221 grep { $_->type eq FOREIGN_KEY }
222 map $_->get_constraints,
223 $schema->get_tables )
227 my ($self, $schema) = @_;
229 $self->header_comments .
230 $self->drop_tables($schema) .
231 join("\n\n", map $self->table($_), grep { $_->name } $schema->get_tables) .
232 "\n" . join "\n", $self->foreign_key_constraints($schema)