1 package SQL::Translator::Generator::DDL::SQLServer;
4 use SQL::Translator::Generator::Utils;
5 use SQL::Translator::Schema::Constants;
7 with 'SQL::Translator::Generator::Role::Quote';
8 with 'SQL::Translator::Generator::Role::DDL';
10 sub quote_chars { [qw([ ])] }
13 sub _build_numeric_types {
19 sub _build_unquoted_defaults {
32 has sizeless_types => (
36 sub _build_sizeless_types {
38 qw( tinyint smallint int integer bigint text bit image datetime ) }
42 my ($self, $field) = @_;
44 return join ' ', $self->field_name($field), ($self->field_type($field)||die 'type is required'),
45 $self->field_autoinc($field),
46 $self->field_nullable($field),
47 $self->field_default($field),
51 my ($self, $field) = @_;
53 ($field->size && !$self->sizeless_types->{$field->data_type}
54 ? '(' . $field->size . ')'
59 sub field_autoinc { ( $_[1]->is_auto_increment ? 'IDENTITY' : () ) }
61 sub primary_key_constraint {
63 $_[0]->quote($_[1]->name || $_[1]->table->name . '_pk') .
65 join( ', ', map $_[0]->quote($_), $_[1]->fields ) .
71 $_[0]->quote($_[1]->name || $_[1]->table->name . '_idx') .
72 ' ON ' . $_[0]->quote($_[1]->table->name) .
73 ' (' . join( ', ', map $_[0]->quote($_), $_[1]->fields ) . ');'
76 sub unique_constraint_single {
77 my ($self, $constraint) = @_;
80 $self->unique_constraint_name($constraint) .
81 ' UNIQUE (' . join( ', ', map $self->quote($_), $constraint->fields ) . ')'
84 sub unique_constraint_name {
85 my ($self, $constraint) = @_;
86 $self->quote($constraint->name || $constraint->table->name . '_uc' )
89 sub unique_constraint_multiple {
90 my ($self, $constraint) = @_;
92 'CREATE UNIQUE NONCLUSTERED INDEX ' .
93 $self->unique_constraint_name($constraint) .
94 ' ON ' . $self->quote($constraint->table->name) . ' (' .
95 join( ', ', $constraint->fields ) . ')' .
96 ' WHERE ' . join( ' AND ',
97 map $self->quote($_->name) . ' IS NOT NULL',
98 grep { $_->is_nullable } $constraint->fields ) . ';'
101 sub foreign_key_constraint {
102 my ($self, $constraint) = @_;
104 my $on_delete = uc ($constraint->on_delete || '');
105 my $on_update = uc ($constraint->on_update || '');
107 # The default implicit constraint action in MSSQL is RESTRICT
108 # but you can not specify it explicitly. Go figure :)
109 for (map uc $_ || '', $on_delete, $on_update) {
110 undef $_ if $_ eq 'RESTRICT'
113 'ALTER TABLE ' . $self->quote($constraint->table->name) .
115 $self->quote($constraint->name || $constraint->table->name . '_fk') .
117 ' (' . join( ', ', map $self->quote($_), $constraint->fields ) . ') REFERENCES '.
118 $self->quote($constraint->reference_table) .
119 ' (' . join( ', ', map $self->quote($_), $constraint->reference_fields ) . ')'
121 $on_delete && $on_delete ne "NO ACTION"
122 ? ' ON DELETE ' . $on_delete
125 $on_update && $on_update ne "NO ACTION"
126 ? ' ON UPDATE ' . $on_update
131 sub enum_constraint_name {
132 my ($self, $field_name) = @_;
133 $self->quote($field_name . '_chk' )
136 sub enum_constraint {
137 my ( $self, $field_name, $vals ) = @_;
140 'CONSTRAINT ' . $self->enum_constraint_name($field_name) .
141 ' CHECK (' . $self->quote($field_name) .
142 ' IN (' . join( ',', map qq('$_'), @$vals ) . '))'
147 my ($self, $table) = @_;
149 (map $self->enum_constraint($_->name, { $_->extra }->{list} || []),
150 grep { 'enum' eq lc $_->data_type } $table->get_fields),
152 (map $self->primary_key_constraint($_),
153 grep { $_->type eq PRIMARY_KEY } $table->get_constraints),
155 (map $self->unique_constraint_single($_),
157 $_->type eq UNIQUE &&
158 !grep { $_->is_nullable } $_->fields
159 } $table->get_constraints),
163 my ($self, $table) = @_;
164 join ( "\n", $self->table_comments($table), '' ) .
166 'CREATE TABLE ' . $self->quote($table->name) . " (\n".
169 $self->fields($table),
170 $self->constraints($table),
173 $self->unique_constraints_multiple($table),
174 $self->indices($table),
178 sub unique_constraints_multiple {
179 my ($self, $table) = @_;
180 (map $self->unique_constraint_multiple($_),
182 $_->type eq UNIQUE &&
183 grep { $_->is_nullable } $_->fields
184 } $table->get_constraints)
188 my ($self, $table) = @_;
189 my $name = $table->name;
190 my $q_name = $self->quote($name);
191 "IF EXISTS (SELECT name FROM sysobjects WHERE name = '$name' AND type = 'U')" .
192 " DROP TABLE $q_name;"
195 sub remove_table_constraints {
196 my ($self, $table) = @_;
197 my $name = $table->name;
198 my $q_name = $self->quote($name);
199 "IF EXISTS (SELECT name FROM sysobjects WHERE name = '$name' AND type = 'U')" .
200 " ALTER TABLE $q_name NOCHECK CONSTRAINT all;"
204 my ($self, $schema) = shift;
206 if ($self->add_drop_table) {
207 my @tables = sort { $b->order <=> $a->order } $schema->get_tables;
209 ( $self->add_comments ? (
211 '-- Turn off constraints',
215 (map $self->remove_table_constraints($_), @tables),
216 ( $self->add_comments ? (
222 (map $self->drop_table($_), @tables),
228 sub foreign_key_constraints {
229 my ($self, $schema) = @_;
230 ( map $self->foreign_key_constraint($_),
231 grep { $_->type eq FOREIGN_KEY }
232 map $_->get_constraints,
233 $schema->get_tables )
237 my ($self, $schema) = @_;
239 $self->header_comments .
240 $self->drop_tables($schema) .
241 join("\n\n", map $self->table($_), grep { $_->name } $schema->get_tables) .
242 "\n" . join "\n", $self->foreign_key_constraints($schema)