X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FProducer%2FMySQL.pm;h=86b32bf2179cddf8e0af2f5c3ad6e9e860480c8b;hb=9a96648f006e57816543756a548303d55397010e;hp=39d58f648df2ff9cb43f87a6e7d7d25bcdd0628e;hpb=bfb5a56826726f1c0da047c5bd4864facfc00964;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Producer/MySQL.pm b/lib/SQL/Translator/Producer/MySQL.pm index 39d58f6..86b32bf 100644 --- a/lib/SQL/Translator/Producer/MySQL.pm +++ b/lib/SQL/Translator/Producer/MySQL.pm @@ -1,7 +1,7 @@ package SQL::Translator::Producer::MySQL; # ------------------------------------------------------------------- -# $Id: MySQL.pm,v 1.49 2006-07-23 14:03:52 schiffbruechige Exp $ +# $Id: MySQL.pm,v 1.54 2007-11-10 03:36:43 mwz444 Exp $ # ------------------------------------------------------------------- # Copyright (C) 2002-4 SQLFairy Authors # @@ -40,6 +40,15 @@ There are still some issues to be worked out with syntax differences between MySQL versions 3 and 4 ("SET foreign_key_checks," character sets for fields, etc.). +=head1 ARGUMENTS + +This producer takes a single optional producer_arg C, which +provides the desired version for the target database. By default MySQL v3 is +assumed, and statements pertaining to any features introduced in later versions +(e.g. CREATE VIEW) are not produced. + +Valid version specifiers for C are listed L + =head2 Table Types Normally the tables will be created without any explicit table type given and @@ -55,34 +64,34 @@ The producer recognises the following extra attributes on the Schema objects. =over 4 -=item field.list +=item B Set the list of allowed values for Enum fields. -=item field.binary field.unsigned field.zerofill +=item B, B, B Set the MySQL field options of the same name. -=item table.mysql_table_type +=item B, B + +Use when producing diffs to indicate that the current table/field has been +renamed from the old name as given in the attribute value. + +=item B Set the type of the table e.g. 'InnoDB', 'MyISAM'. This will be automatically set for tables involved in foreign key constraints if it is not already set explicitly. See L<"Table Types">. -=item mysql_character_set +Please note that the C option is the prefered method of specifying +the MySQL storage engine to use, but this method still works for backwards +compatability. -MySql-4.1+. Set the tables character set. -Run SHOW CHARACTER SET to see list. - -=item mysql_collate - -MySql-4.1+. Set the tables colation order. - -=item table.mysql_charset table.mysql_collate +=item B, B Set the tables default charater set and collation order. -=item field.mysql_charset field.mysql_collate +=item B, B Set the fields charater set and collation order. @@ -92,13 +101,18 @@ Set the fields charater set and collation order. use strict; use warnings; -use vars qw[ $VERSION $DEBUG ]; -$VERSION = sprintf "%d.%02d", q$Revision: 1.49 $ =~ /(\d+)\.(\d+)/; +use vars qw[ $VERSION $DEBUG %used_names ]; +$VERSION = sprintf "%d.%02d", q$Revision: 1.54 $ =~ /(\d+)\.(\d+)/; $DEBUG = 0 unless defined $DEBUG; +# Maximum length for most identifiers is 64, according to: +# http://dev.mysql.com/doc/refman/4.1/en/identifiers.html +# http://dev.mysql.com/doc/refman/5.0/en/identifiers.html +my $DEFAULT_MAX_ID_LENGTH = 64; + use Data::Dumper; use SQL::Translator::Schema::Constants; -use SQL::Translator::Utils qw(debug header_comment); +use SQL::Translator::Utils qw(debug header_comment truncate_id_uniquely parse_mysql_version); # # Use only lowercase for the keys (e.g. "long" and not "LONG") @@ -128,56 +142,230 @@ my %translate = ( 'datetime' => 'datetime', ); + +sub preprocess_schema { + my ($schema) = @_; + + # extra->{mysql_table_type} used to be the type. It belongs in options, so + # move it if we find it. Return Engine type if found in extra or options + # Similarly for mysql_charset and mysql_collate + my $extra_to_options = sub { + my ($table, $extra_name, $opt_name) = @_; + + my $extra = $table->extra; + + my $extra_type = delete $extra->{$extra_name}; + + # Now just to find if there is already an Engine or Type option... + # and lets normalize it to ENGINE since: + # + # The ENGINE table option specifies the storage engine for the table. + # TYPE is a synonym, but ENGINE is the preferred option name. + # + + # We have to use the hash directly here since otherwise there is no way + # to remove options. + my $options = ( $table->{options} ||= []); + + # If multiple option names, normalize to the first one + if (ref $opt_name) { + OPT_NAME: for ( @$opt_name[1..$#$opt_name] ) { + for my $idx ( 0..$#{$options} ) { + my ($key, $value) = %{ $options->[$idx] }; + + if (uc $key eq $_) { + $options->[$idx] = { $opt_name->[0] => $value }; + last OPT_NAME; + } + } + } + $opt_name = $opt_name->[0]; + + } + + + # This assumes that there isn't both a Type and an Engine option. + OPTION: + for my $idx ( 0..$#{$options} ) { + my ($key, $value) = %{ $options->[$idx] }; + + next unless uc $key eq $opt_name; + + # make sure case is right on option name + delete $options->[$idx]{$key}; + return $options->[$idx]{$opt_name} = $value || $extra_type; + + } + + if ($extra_type) { + push @$options, { $opt_name => $extra_type }; + return $extra_type; + } + + }; + + # Names are only specific to a given schema + local %used_names = (); + + # + # Work out which tables need to be InnoDB to support foreign key + # constraints. We do this first as we need InnoDB at both ends. + # + foreach my $table ( $schema->get_tables ) { + + $extra_to_options->($table, 'mysql_table_type', ['ENGINE', 'TYPE'] ); + $extra_to_options->($table, 'mysql_charset', 'CHARACTER SET' ); + $extra_to_options->($table, 'mysql_collate', 'COLLATE' ); + + foreach my $c ( $table->get_constraints ) { + next unless $c->type eq FOREIGN_KEY; + + # Normalize constraint names here. + my $c_name = $c->name; + # Give the constraint a name if it doesn't have one, so it doens't feel + # left out + $c_name = $table->name . '_fk' unless length $c_name; + + $c->name( next_unused_name($c_name) ); + + for my $meth (qw/table reference_table/) { + my $table = $schema->get_table($c->$meth) || next; + # This normalizes the types to ENGINE and returns the value if its there + next if $extra_to_options->($table, 'mysql_table_type', ['ENGINE', 'TYPE']); + $table->options( { 'ENGINE' => 'InnoDB' } ); + } + } # foreach constraints + + my %map = ( mysql_collate => 'collate', mysql_charset => 'character set'); + foreach my $f ( $table->get_fields ) { + my $extra = $f->extra; + for (keys %map) { + $extra->{$map{$_}} = delete $extra->{$_} if exists $extra->{$_}; + } + + my @size = $f->size; + if ( !$size[0] && $f->data_type =~ /char$/ ) { + $f->size( (255) ); + } + } + + } +} + sub produce { my $translator = shift; local $DEBUG = $translator->debug; + local %used_names; my $no_comments = $translator->no_comments; my $add_drop_table = $translator->add_drop_table; my $schema = $translator->schema; my $show_warnings = $translator->show_warnings || 0; + my $producer_args = $translator->producer_args; + my $mysql_version = parse_mysql_version ($producer_args->{mysql_version}, 'perl') || 0; + my $max_id_length = $producer_args->{mysql_max_id_length} || $DEFAULT_MAX_ID_LENGTH; - debug("PKG: Beginning production\n"); + my ($qt, $qf, $qc) = ('','', ''); + $qt = '`' if $translator->quote_table_names; + $qf = '`' if $translator->quote_field_names; - my $create; + debug("PKG: Beginning production\n"); + %used_names = (); + my $create = ''; $create .= header_comment unless ($no_comments); # \todo Don't set if MySQL 3.x is set on command line - $create .= "SET foreign_key_checks=0;\n\n"; + my @create = "SET foreign_key_checks=0"; - # - # Work out which tables need to be InnoDB to support foreign key - # constraints. We do this first as we need InnoDB at both ends. - # - foreach ( map { $_->get_constraints } $schema->get_tables ) { - next unless $_->type eq FOREIGN_KEY; - foreach my $meth (qw/table reference_table/) { - my $table = $schema->get_table($_->$meth) || next; - next if $table->extra('mysql_table_type'); - $table->extra( 'mysql_table_type' => 'InnoDB'); - } - } + preprocess_schema($schema); # # Generate sql # my @table_defs =(); + for my $table ( $schema->get_tables ) { # print $table->name, "\n"; push @table_defs, create_table($table, - { add_drop_table => $add_drop_table, - show_warnings => $show_warnings, - no_comments => $no_comments + { add_drop_table => $add_drop_table, + show_warnings => $show_warnings, + no_comments => $no_comments, + quote_table_names => $qt, + quote_field_names => $qf, + max_id_length => $max_id_length, + mysql_version => $mysql_version + }); + } + + if ($mysql_version > 5.0) { + for my $view ( $schema->get_views ) { + push @table_defs, create_view($view, + { add_replace_view => $add_drop_table, + show_warnings => $show_warnings, + no_comments => $no_comments, + quote_table_names => $qt, + quote_field_names => $qf, + max_id_length => $max_id_length, + mysql_version => $mysql_version }); + } } + # print "@table_defs\n"; + push @table_defs, "SET foreign_key_checks=1"; - return wantarray ? ($create, @table_defs, 'SET foreign_key_checks=1;') : $create . join ('', @table_defs, "SET foreign_key_checks=1;\n\n"); + return wantarray ? ($create ? $create : (), @create, @table_defs) : ($create . join('', map { $_ ? "$_;\n\n" : () } (@create, @table_defs))); +} + +sub create_view { + my ($view, $options) = @_; + my $qt = $options->{quote_table_names} || ''; + my $qf = $options->{quote_field_names} || ''; + + my $view_name = $view->name; + debug("PKG: Looking at view '${view_name}'\n"); + + # Header. Should this look like what mysqldump produces? + my $create = ''; + $create .= "--\n-- View: ${qt}${view_name}${qt}\n--\n" unless $options->{no_comments}; + $create .= 'CREATE'; + $create .= ' OR REPLACE' if $options->{add_replace_view}; + $create .= "\n"; + + my $extra = $view->extra; + # ALGORITHM + if( exists($extra->{mysql_algorithm}) && defined(my $algorithm = $extra->{mysql_algorithm}) ){ + $create .= " ALGORITHM = ${algorithm}\n" if $algorithm =~ /(?:UNDEFINED|MERGE|TEMPTABLE)/i; + } + # DEFINER + if( exists($extra->{mysql_definer}) && defined(my $user = $extra->{mysql_definer}) ){ + $create .= " DEFINER = ${user}\n"; + } + # SECURITY + if( exists($extra->{mysql_security}) && defined(my $security = $extra->{mysql_security}) ){ + $create .= " SQL SECURITY ${security}\n" if $security =~ /(?:DEFINER|INVOKER)/i; + } + + #Header, cont. + $create .= " VIEW ${qt}${view_name}${qt}"; + + if( my @fields = $view->fields ){ + my $list = join ', ', map { "${qf}${_}${qf}"} @fields; + $create .= " ( ${list} )"; + } + if( my $sql = $view->sql ){ + $create .= " AS (\n ${sql}\n )"; + } +# $create .= ""; + return $create; } sub create_table { my ($table, $options) = @_; + my $qt = $options->{quote_table_names} || ''; + my $qf = $options->{quote_field_names} || ''; + my $table_name = $table->name; debug("PKG: Looking at table '$table_name'\n"); @@ -185,16 +373,17 @@ sub create_table # Header. Should this look like what mysqldump produces? # my $create = ''; - $create .= "--\n-- Table: $table_name\n--\n" unless $options->{no_comments}; - $create .= qq[DROP TABLE IF EXISTS $table_name;\n] if $options->{add_drop_table}; - $create .= "CREATE TABLE $table_name (\n"; + my $drop; + $create .= "--\n-- Table: $qt$table_name$qt\n--\n" unless $options->{no_comments}; + $drop = qq[DROP TABLE IF EXISTS $qt$table_name$qt] if $options->{add_drop_table}; + $create .= "CREATE TABLE $qt$table_name$qt (\n"; # # Fields # my @field_defs; for my $field ( $table->get_fields ) { - push @field_defs, create_field($field); + push @field_defs, create_field($field, $options); } # @@ -203,7 +392,7 @@ sub create_table my @index_defs; my %indexed_fields; for my $index ( $table->get_indices ) { - push @index_defs, create_index($index); + push @index_defs, create_index($index, $options); $indexed_fields{ $_ } = 1 for $index->fields; } @@ -216,10 +405,10 @@ sub create_table my $constr = create_constraint($c, $options); push @constraint_defs, $constr if($constr); - unless ( $indexed_fields{ ($c->fields())[0] } ) { - push @index_defs, "INDEX (" . ($c->fields())[0] . ")"; - $indexed_fields{ ($c->fields())[0] } = 1; - } + unless ( $indexed_fields{ ($c->fields())[0] } || $c->type ne FOREIGN_KEY ) { + push @index_defs, "INDEX ($qf" . ($c->fields())[0] . "$qf)"; + $indexed_fields{ ($c->fields())[0] } = 1; + } } $create .= join(",\n", map { " $_" } @@ -230,40 +419,60 @@ sub create_table # Footer # $create .= "\n)"; - my $table_type_defined = 0; - for my $t1_option_ref ( $table->options ) { - my($key, $value) = %{$t1_option_ref}; - $table_type_defined = 1 - if uc $key eq 'ENGINE' or uc $key eq 'TYPE'; - $create .= " $key=$value"; - } - my $mysql_table_type = $table->extra('mysql_table_type'); - #my $charset = $table->extra('mysql_character_set'); - #my $collate = $table->extra('mysql_collate'); - #$create .= " Type=$mysql_table_type" if $mysql_table_type; - #$create .= " DEFAULT CHARACTER SET $charset" if $charset; - #$create .= " COLLATE $collate" if $collate; - $create .= " Type=$mysql_table_type" - if $mysql_table_type && !$table_type_defined; - my $charset = $table->extra('mysql_charset'); - my $collate = $table->extra('mysql_collate'); - my $comments = $table->comments; - - $create .= " DEFAULT CHARACTER SET $charset" if $charset; - $create .= " COLLATE $collate" if $collate; - $create .= qq[ comment='$comments'] if $comments; - $create .= ";\n\n"; + $create .= generate_table_options($table, $options) || ''; +# $create .= ";\n\n"; - return $create; + return $drop ? ($drop,$create) : $create; +} + +sub generate_table_options +{ + my ($table, $options) = @_; + my $create; + + my $table_type_defined = 0; + my $qf = $options->{quote_field_names} ||= ''; + my $charset = $table->extra('mysql_charset'); + my $collate = $table->extra('mysql_collate'); + my $union = undef; + for my $t1_option_ref ( $table->options ) { + my($key, $value) = %{$t1_option_ref}; + $table_type_defined = 1 + if uc $key eq 'ENGINE' or uc $key eq 'TYPE'; + if (uc $key eq 'CHARACTER SET') { + $charset = $value; + next; + } elsif (uc $key eq 'COLLATE') { + $collate = $value; + next; + } elsif (uc $key eq 'UNION') { + $union = "($qf". join("$qf, $qf", @$value) ."$qf)"; + next; + } + $create .= " $key=$value"; + } + + my $mysql_table_type = $table->extra('mysql_table_type'); + $create .= " ENGINE=$mysql_table_type" + if $mysql_table_type && !$table_type_defined; + my $comments = $table->comments; + + $create .= " DEFAULT CHARACTER SET $charset" if $charset; + $create .= " COLLATE $collate" if $collate; + $create .= " UNION=$union" if $union; + $create .= qq[ comment='$comments'] if $comments; + return $create; } sub create_field { - my ($field) = @_; + my ($field, $options) = @_; + + my $qf = $options->{quote_field_names} ||= ''; my $field_name = $field->name; debug("PKG: Looking at field '$field_name'\n"); - my $field_def = $field_name; + my $field_def = "$qf$field_name$qf"; # data type and size my $data_type = $field->data_type; @@ -275,6 +484,7 @@ sub create_field my $charset = $extra{'mysql_charset'}; my $collate = $extra{'mysql_collate'}; + my $mysql_version = $options->{mysql_version} || 0; # # Oracle "number" type -- figure best MySQL type # @@ -295,17 +505,21 @@ sub create_field } # # Convert a large Oracle varchar to "text" + # (not necessary as of 5.0.3 http://dev.mysql.com/doc/refman/5.0/en/char.html) # elsif ( $data_type =~ /char/i && $size[0] > 255 ) { - $data_type = 'text'; - @size = (); - } - elsif ( $data_type =~ /char/i && ! $size[0] ) { - @size = (255); + unless ($size[0] <= 65535 && $mysql_version >= 5.000003 ) { + $data_type = 'text'; + @size = (); + } } elsif ( $data_type =~ /boolean/i ) { - $data_type = 'enum'; - $commalist = "'0','1'"; + if ($mysql_version >= 4) { + $data_type = 'boolean'; + } else { + $data_type = 'enum'; + $commalist = "'0','1'"; + } } elsif ( exists $translate{ lc $data_type } ) { $data_type = $translate{ lc $data_type }; @@ -319,9 +533,9 @@ sub create_field $field_def .= " $data_type"; - if ( lc $data_type eq 'enum' ) { + if ( lc($data_type) eq 'enum' || lc($data_type) eq 'set') { $field_def .= '(' . $commalist . ')'; - } + } elsif ( defined $size[0] && $size[0] > 0 ) { $field_def .= '(' . join( ', ', @size ) . ')'; } @@ -346,11 +560,13 @@ sub create_field # Default? XXX Need better quoting! my $default = $field->default_value; if ( defined $default ) { - if ( uc $default eq 'NULL') { - $field_def .= ' DEFAULT NULL'; - } else { - $field_def .= " DEFAULT '$default'"; - } + SQL::Translator::Producer->_apply_default_value( + \$field_def, + $default, + [ + 'NULL' => \'NULL', + ], + ); } if ( my $comments = $field->comments ) { @@ -363,50 +579,122 @@ sub create_field return $field_def; } +sub alter_create_index +{ + my ($index, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + my $qf = $options->{quote_field_names} || ''; + + return join( ' ', + 'ALTER TABLE', + $qt.$index->table->name.$qt, + 'ADD', + create_index(@_) + ); +} + sub create_index { - my ($index) = @_; + my ($index, $options) = @_; + + my $qf = $options->{quote_field_names} || ''; + + return join( ' ', + lc $index->type eq 'normal' ? 'INDEX' : $index->type . ' INDEX', + truncate_id_uniquely( $index->name, $options->{max_id_length} || $DEFAULT_MAX_ID_LENGTH ), + '(' . $qf . join( "$qf, $qf", $index->fields ) . $qf . ')' + ); + +} + +sub alter_drop_index +{ + my ($index, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + my $qf = $options->{quote_field_names} || ''; return join( ' ', - lc $index->type eq 'normal' ? 'INDEX' : $index->type, - $index->name, - '(' . join( ', ', $index->fields ) . ')' + 'ALTER TABLE', + $qt.$index->table->name.$qt, + 'DROP', + 'INDEX', + $index->name || $index->fields ); } +sub alter_drop_constraint +{ + my ($c, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + my $qc = $options->{quote_field_names} || ''; + + my $out = sprintf('ALTER TABLE %s DROP %s %s', + $qt . $c->table->name . $qt, + $c->type eq FOREIGN_KEY ? $c->type : "INDEX", + $qc . $c->name . $qc ); + + return $out; +} + +sub alter_create_constraint +{ + my ($index, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + return join( ' ', + 'ALTER TABLE', + $qt.$index->table->name.$qt, + 'ADD', + create_constraint(@_) ); +} + sub create_constraint { my ($c, $options) = @_; + my $qf = $options->{quote_field_names} || ''; + my $qt = $options->{quote_table_names} || ''; + my $leave_name = $options->{leave_name} || undef; + my @fields = $c->fields or next; if ( $c->type eq PRIMARY_KEY ) { - return 'PRIMARY KEY (' . join(', ', @fields). ')'; + return 'PRIMARY KEY (' . $qf . join("$qf, $qf", @fields). $qf . ')'; } elsif ( $c->type eq UNIQUE ) { return - 'UNIQUE '. - (defined $c->name ? $c->name.' ' : ''). - '(' . join(', ', @fields). ')'; + 'UNIQUE '. + (defined $c->name ? $qf.truncate_id_uniquely( $c->name, $options->{max_id_length} || $DEFAULT_MAX_ID_LENGTH ).$qf.' ' : ''). + '(' . $qf . join("$qf, $qf", @fields). $qf . ')'; } elsif ( $c->type eq FOREIGN_KEY ) { # # Make sure FK field is indexed or MySQL complains. # + my $table = $c->table; + my $c_name = truncate_id_uniquely( $c->name, $options->{max_id_length} || $DEFAULT_MAX_ID_LENGTH ); + my $def = join(' ', - map { $_ || () } 'CONSTRAINT', $c->table . '_' . $c->name, 'FOREIGN KEY' - ); + map { $_ || () } + 'CONSTRAINT', + $qf . $c_name . $qf, + 'FOREIGN KEY' + ); + - $def .= ' (' . join( ', ', @fields ) . ')'; + $def .= ' ('.$qf . join( "$qf, $qf", @fields ) . $qf . ')'; - $def .= ' REFERENCES ' . $c->reference_table; + $def .= ' REFERENCES ' . $qt . $c->reference_table . $qt; my @rfields = map { $_ || () } $c->reference_fields; unless ( @rfields ) { my $rtable_name = $c->reference_table; - if ( my $ref_table = $c->table->schema->get_table( $rtable_name ) ) { + if ( my $ref_table = $table->schema->get_table( $rtable_name ) ) { push @rfields, $ref_table->primary_key; } else { @@ -416,10 +704,10 @@ sub create_constraint } if ( @rfields ) { - $def .= ' (' . join( ', ', @rfields ) . ')'; + $def .= ' (' . $qf . join( "$qf, $qf", @rfields ) . $qf . ')'; } else { - warn "FK constraint on " . $c->table->name . '.' . + warn "FK constraint on " . $table->name . '.' . join('', @fields) . " has no reference fields\n" if $options->{show_warnings}; } @@ -442,25 +730,45 @@ sub create_constraint return undef; } +sub alter_table +{ + my ($to_table, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + + my $table_options = generate_table_options($to_table, $options) || ''; + my $out = sprintf('ALTER TABLE %s%s', + $qt . $to_table->name . $qt, + $table_options); + + return $out; +} + +sub rename_field { alter_field(@_) } sub alter_field { - my ($from_field, $to_field) = @_; + my ($from_field, $to_field, $options) = @_; + + my $qf = $options->{quote_field_names} || ''; + my $qt = $options->{quote_table_names} || ''; my $out = sprintf('ALTER TABLE %s CHANGE COLUMN %s %s', - $to_field->table->name, - $to_field->name, - create_field($to_field)); + $qt . $to_field->table->name . $qt, + $qf . $from_field->name . $qf, + create_field($to_field, $options)); return $out; } sub add_field { - my ($new_field) = @_; + my ($new_field, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; my $out = sprintf('ALTER TABLE %s ADD COLUMN %s', - $new_field->table->name, - create_field($new_field)); + $qt . $new_field->table->name . $qt, + create_field($new_field, $options)); return $out; @@ -468,16 +776,132 @@ sub add_field sub drop_field { - my ($old_field) = @_; + my ($old_field, $options) = @_; + my $qf = $options->{quote_field_names} || ''; + my $qt = $options->{quote_table_names} || ''; + my $out = sprintf('ALTER TABLE %s DROP COLUMN %s', - $old_field->table->name, - $old_field->name); + $qt . $old_field->table->name . $qt, + $qf . $old_field->name . $qf); return $out; } +sub batch_alter_table { + my ($table, $diff_hash, $options) = @_; + + # InnoDB has an issue with dropping and re-adding a FK constraint under the + # name in a single alter statment, see: http://bugs.mysql.com/bug.php?id=13741 + # + # We have to work round this. + + my %fks_to_alter; + my %fks_to_drop = map { + $_->type eq FOREIGN_KEY + ? ( $_->name => $_ ) + : ( ) + } @{$diff_hash->{alter_drop_constraint} }; + + my %fks_to_create = map { + if ( $_->type eq FOREIGN_KEY) { + $fks_to_alter{$_->name} = $fks_to_drop{$_->name} if $fks_to_drop{$_->name}; + ( $_->name => $_ ); + } else { ( ) } + } @{$diff_hash->{alter_create_constraint} }; + + my @drop_stmt; + if (scalar keys %fks_to_alter) { + $diff_hash->{alter_drop_constraint} = [ + grep { !$fks_to_alter{$_->name} } @{ $diff_hash->{alter_drop_constraint} } + ]; + + @drop_stmt = batch_alter_table($table, { alter_drop_constraint => [ values %fks_to_alter ] }, $options); + + } + + my @stmts = map { + if (@{ $diff_hash->{$_} || [] }) { + my $meth = __PACKAGE__->can($_) or die __PACKAGE__ . " cant $_"; + map { $meth->( (ref $_ eq 'ARRAY' ? @$_ : $_), $options ) } @{ $diff_hash->{$_} } + } else { () } + } qw/rename_table + alter_drop_constraint + alter_drop_index + drop_field + add_field + alter_field + rename_field + alter_create_index + alter_create_constraint + alter_table/; + + # rename_table makes things a bit more complex + my $renamed_from = ""; + $renamed_from = $diff_hash->{rename_table}[0][0]->name + if $diff_hash->{rename_table} && @{$diff_hash->{rename_table}}; + + return unless @stmts; + # Just zero or one stmts. return now + return (@drop_stmt,@stmts) unless @stmts > 1; + + # Now strip off the 'ALTER TABLE xyz' of all but the first one + + my $qt = $options->{quote_table_names} || ''; + my $table_name = $qt . $table->name . $qt; + + + my $re = $renamed_from + ? qr/^ALTER TABLE (?:\Q$table_name\E|\Q$qt$renamed_from$qt\E) / + : qr/^ALTER TABLE \Q$table_name\E /; + + my $first = shift @stmts; + my ($alter_table) = $first =~ /($re)/; + + my $padd = " " x length($alter_table); + + return @drop_stmt, join( ",\n", $first, map { s/$re//; $padd . $_ } @stmts); + +} + +sub drop_table { + my ($table, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + + # Drop (foreign key) constraints so table drops cleanly + my @sql = batch_alter_table($table, { alter_drop_constraint => [ grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints ] }, $options); + + return (@sql, "DROP TABLE $qt$table$qt"); +# return join("\n", @sql, "DROP TABLE $qt$table$qt"); + +} + +sub rename_table { + my ($old_table, $new_table, $options) = @_; + + my $qt = $options->{quote_table_names} || ''; + + return "ALTER TABLE $qt$old_table$qt RENAME TO $qt$new_table$qt"; +} + +sub next_unused_name { + my $name = shift || ''; + if ( !defined($used_names{$name}) ) { + $used_names{$name} = $name; + return $name; + } + + my $i = 1; + while ( defined($used_names{$name . '_' . $i}) ) { + ++$i; + } + $name .= '_' . $i; + $used_names{$name} = $name; + return $name; +} + 1; # -------------------------------------------------------------------