X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FParser%2FMySQL.pm;h=01c95da1999c823821bd5db9015e10b603b57dcf;hb=1f58ba76e745ac78067d841f68e6deac803f84b9;hp=3ebb7c8ed5c1a1399726e4dec93d6f80da6a44ae;hpb=c736c39c54019a4fbd6e787e848847fb102bb2e3;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Parser/MySQL.pm b/lib/SQL/Translator/Parser/MySQL.pm index 3ebb7c8..01c95da 100644 --- a/lib/SQL/Translator/Parser/MySQL.pm +++ b/lib/SQL/Translator/Parser/MySQL.pm @@ -1,7 +1,7 @@ package SQL::Translator::Parser::MySQL; # ------------------------------------------------------------------- -# $Id: MySQL.pm,v 1.21 2003-06-03 22:42:10 kycl4rk Exp $ +# $Id: MySQL.pm,v 1.32 2003-08-17 07:44:06 rossta Exp $ # ------------------------------------------------------------------- # Copyright (C) 2003 Ken Y. Clark , # darren chamberlain , @@ -123,7 +123,7 @@ Here's the word from the MySQL site use strict; use vars qw[ $DEBUG $VERSION $GRAMMAR @EXPORT_OK ]; -$VERSION = sprintf "%d.%02d", q$Revision: 1.21 $ =~ /(\d+)\.(\d+)/; +$VERSION = sprintf "%d.%02d", q$Revision: 1.32 $ =~ /(\d+)\.(\d+)/; $DEBUG = 0 unless defined $DEBUG; use Data::Dumper; @@ -155,12 +155,20 @@ startrule : statement(s) eofile { \%tables } eofile : /^\Z/ statement : comment + | use + | set | drop | create | +use : /use/i WORD ';' + +set : /set/i /[^;]+/ ';' + drop : /drop/i WORD(s) ';' +create : CREATE /database/i WORD ';' + create : CREATE TEMPORARY(?) TABLE opt_if_not_exists(?) table_name '(' create_definition(s /,/) ')' table_option(s?) ';' { my $table_name = $item{'table_name'}; @@ -169,14 +177,15 @@ create : CREATE TEMPORARY(?) TABLE opt_if_not_exists(?) table_name '(' create_de my $i = 1; for my $definition ( @{ $item[7] } ) { - if ( $definition->{'type'} eq 'field' ) { + if ( $definition->{'supertype'} eq 'field' ) { + my $field_name = $definition->{'name'}; $tables{ $table_name }{'fields'}{ $field_name } = { %$definition, order => $i }; $i++; if ( $definition->{'is_primary_key'} ) { - push @{ $tables{ $table_name }{'indices'} }, + push @{ $tables{ $table_name }{'constraints'} }, { type => 'primary_key', fields => [ $field_name ], @@ -184,15 +193,19 @@ create : CREATE TEMPORARY(?) TABLE opt_if_not_exists(?) table_name '(' create_de ; } } - elsif ( $definition->{'type'} eq 'foreign_key' ) { - for my $field ( @{ $definition->{'fields'} } ) { - push @{ - $tables{$table_name}{'fields'}{$field}{'constraints'} - }, - $definition; - } + elsif ( $definition->{'supertype'} eq 'constraint' ) { + # prob get rid of this? +# for my $field ( @{ $definition->{'fields'} } ) { +# push @{ +# $tables{$table_name}{'fields'}{$field}{'constraints'} +# }, +# $definition; +# } + + # this should be the only one needed + push @{ $tables{ $table_name }{'constraints'} }, $definition; } - else { + elsif ( $definition->{'supertype'} eq 'index' ) { push @{ $tables{ $table_name }{'indices'} }, $definition; } @@ -209,7 +222,7 @@ create : CREATE TEMPORARY(?) TABLE opt_if_not_exists(?) table_name '(' create_de opt_if_not_exists : /if not exists/i -create : /CREATE/i unique(?) /(INDEX|KEY)/i index_name /on/i table_name '(' field_name(s /,/) ')' ';' +create : CREATE UNIQUE(?) /(index|key)/i index_name /on/i table_name '(' field_name(s /,/) ')' ';' { push @{ $tables{ $item{'table_name'} }{'indices'} }, { @@ -220,16 +233,21 @@ create : /CREATE/i unique(?) /(INDEX|KEY)/i index_name /on/i table_name '(' fiel ; } -create_definition : index - | foreign_key +create_definition : constraint + | index | field | -comment : /^\s*(?:#|-{2}).*\n/ +comment : /^\s*(?:#|-{2}).*\n/ { + my $comment = $item[1]; + $comment =~ s/^\s*(#|-{2})\s*//; + $comment =~ s/\s*$//; + $return = $comment; +} blank : /\s*/ -field : field_name data_type field_qualifier(s?) reference_definition(?) +field : comment(s?) field_name data_type field_qualifier(s?) reference_definition(?) comment(s?) { my %qualifiers = map { %$_ } @{ $item{'field_qualifier(s?)'} || [] }; my $null = defined $item{'not_null'} ? $item{'not_null'} : 1; @@ -238,14 +256,17 @@ field : field_name data_type field_qualifier(s?) reference_definition(?) $qualifiers{ $_ } = 1 for @type_quals; } + my @comments = ( @{ $item[1] }, @{ $item[6] } ); + $return = { - type => 'field', - name => $item{'field_name'}, - data_type => $item{'data_type'}{'type'}, - size => $item{'data_type'}{'size'}, - list => $item{'data_type'}{'list'}, - null => $null, - constraints => $item{'reference_definition(?)'}, + supertype => 'field', + name => $item{'field_name'}, + data_type => $item{'data_type'}{'type'}, + size => $item{'data_type'}{'size'}, + list => $item{'data_type'}{'list'}, + null => $null, + constraints => $item{'reference_definition(?)'}, + comments => [ @comments ], %qualifiers, } } @@ -286,6 +307,13 @@ field_qualifier : unsigned } } +field_qualifier : /character set/i WORD + { + $return = { + character_set => $item[2], + } + } + reference_definition : /references/i table_name parens_field_list(?) match_type(?) on_delete_do(?) on_update_do(?) { $return = { @@ -298,7 +326,6 @@ reference_definition : /references/i table_name parens_field_list(?) match_type( } } - match_type : /match full/i { 'match_full' } | /match partial/i { 'match_partial' } @@ -316,15 +343,13 @@ reference_option: /restrict/i | /set default/i { $item[1] } -index : primary_key_index - | unique_index +index : normal_index | fulltext_index - | normal_index | -table_name : WORD +table_name : NAME -field_name : WORD +field_name : NAME index_name : WORD @@ -354,12 +379,13 @@ data_type : WORD parens_value_list(s?) type_qualifier(s?) $size = [9]; } elsif ( $type =~ /^int(eger)?$/ ) { + $type = 'int'; $size = [11]; } elsif ( lc $type eq 'bigint' ) { $size = [20]; } - elsif ( lc $type =~ /(float|double|decimal|numeric|real)/ ) { + elsif ( lc $type =~ /(float|double|decimal|numeric|real|fixed|dec)/ ) { $size = [8,2]; } } @@ -399,9 +425,15 @@ auto_inc : /auto_increment/i { 1 } primary_key : /primary/i /key/i { 1 } -foreign_key : opt_constraint(?) /foreign key/i WORD(?) parens_field_list reference_definition +constraint : primary_key_def + | unique_key_def + | foreign_key_def + | + +foreign_key_def : opt_constraint(?) /foreign key/i WORD(?) parens_field_list reference_definition { $return = { + supertype => 'constraint', type => 'foreign_key', name => $item[3][0], fields => $item[4], @@ -411,50 +443,52 @@ foreign_key : opt_constraint(?) /foreign key/i WORD(?) parens_field_list referen opt_constraint : /constraint/i WORD -primary_key_index : primary_key index_name(?) '(' field_name(s /,/) ')' +primary_key_def : primary_key index_name(?) '(' name_with_opt_paren(s /,/) ')' { - $return = { - name => $item{'index_name(?)'}[0], - type => 'primary_key', - fields => $item[4], + $return = { + supertype => 'constraint', + name => $item{'index_name(?)'}[0], + type => 'primary_key', + fields => $item[4], }; } -normal_index : key index_name(?) '(' name_with_opt_paren(s /,/) ')' +unique_key_def : UNIQUE KEY(?) index_name(?) '(' name_with_opt_paren(s /,/) ')' { - $return = { - name => $item{'index_name(?)'}[0], - type => 'normal', - fields => $item[4], + $return = { + supertype => 'constraint', + name => $item{'index_name(?)'}[0], + type => 'unique', + fields => $item[5], } } -unique_index : unique key(?) index_name(?) '(' name_with_opt_paren(s /,/) ')' +normal_index : KEY index_name(?) '(' name_with_opt_paren(s /,/) ')' { - $return = { - name => $item{'index_name(?)'}[0], - type => 'unique', - fields => $item[5], + $return = { + supertype => 'index', + type => 'normal', + name => $item{'index_name(?)'}[0], + fields => $item[4], } } -fulltext_index : fulltext key(?) index_name(?) '(' name_with_opt_paren(s /,/) ')' +fulltext_index : /fulltext/i KEY(?) index_name(?) '(' name_with_opt_paren(s /,/) ')' { - $return = { - name => $item{'index_name(?)'}[0], - type => 'fulltext', - fields => $item[5], + $return = { + supertype => 'index', + type => 'fulltext', + name => $item{'index_name(?)'}[0], + fields => $item[5], } } name_with_opt_paren : NAME parens_value_list(s?) { $item[2][0] ? "$item[1]($item[2][0][0])" : $item[1] } -fulltext : /fulltext/i { 1 } +UNIQUE : /unique/i { 1 } -unique : /unique/i { 1 } - -key : /key/i | /index/i +KEY : /key/i | /index/i table_option : /[^\s;]*/ { @@ -480,8 +514,13 @@ NAME : "`" /\w+/ "`" VALUE : /[-+]?\.?\d+(?:[eE]\d+)?/ { $item[1] } - | /'.*?'/ # XXX doesn't handle embedded quotes - { $item[1] } + | /'.*?'/ + { + # remove leading/trailing quotes + my $val = $item[1]; + $val =~ s/^['"]|['"]$//g; + $return = $val; + } | /NULL/ { 'NULL' } @@ -505,12 +544,22 @@ sub parse { warn Dumper( $result ) if $DEBUG; my $schema = $translator->schema; - for my $table_name ( keys %{ $result } ) { + my @tables = sort { + $result->{ $a }->{'order'} <=> $result->{ $b }->{'order'} + } keys %{ $result }; + + for my $table_name ( @tables ) { my $tdata = $result->{ $table_name }; my $table = $schema->add_table( name => $tdata->{'table_name'}, ) or die $schema->error; +# for my $opt ( @{ $tdata->{'table_options'} } ) { +# if ( my ( $key, $val ) = each %$opt ) { +# $tables->options( +# } +# } + my @fields = sort { $tdata->{'fields'}->{$a}->{'order'} <=> @@ -526,19 +575,65 @@ sub parse { default_value => $fdata->{'default'}, is_auto_increment => $fdata->{'is_auto_inc'}, is_nullable => $fdata->{'null'}, + comments => $fdata->{'comments'}, + ) or die $table->error; + + $table->primary_key( $field->name ) if $fdata->{'is_primary_key'}; + + for my $qual ( qw[ binary unsigned zerofill list ] ) { + if ( my $val = $fdata->{ $qual } || $fdata->{ uc $qual } ) { + next if ref $val eq 'ARRAY' && !@$val; + $field->extra( $qual, $val ); + } + } + + if ( $field->data_type =~ /(set|enum)/i && !$field->size ) { + my %extra = $field->extra; + my $longest = 0; + for my $len ( map { length } @{ $extra{'list'} || [] } ) { + $longest = $len if $len > $longest; + } + $field->size( $longest ) if $longest; + } + + for my $cdata ( @{ $fdata->{'constraints'} } ) { + next unless $cdata->{'type'} eq 'foreign_key'; + $cdata->{'fields'} ||= [ $field->name ]; + push @{ $tdata->{'constraints'} }, $cdata; + } + } + + for my $idata ( @{ $tdata->{'indices'} || [] } ) { + my $index = $table->add_index( + name => $idata->{'name'}, + type => uc $idata->{'type'}, + fields => $idata->{'fields'}, + ) or die $table->error; + } + + for my $cdata ( @{ $tdata->{'constraints'} || [] } ) { + my $constraint = $table->add_constraint( + name => $cdata->{'name'}, + type => $cdata->{'type'}, + fields => $cdata->{'fields'}, + reference_table => $cdata->{'reference_table'}, + reference_fields => $cdata->{'reference_fields'}, + match_type => $cdata->{'match_type'} || '', + on_delete => $cdata->{'on_delete_do'}, + on_update => $cdata->{'on_update_do'}, ) or die $table->error; } } - return $result; + return 1; } 1; -# ---------------------------------------------------- +# ------------------------------------------------------------------- # Where man is not nature is barren. # William Blake -# ---------------------------------------------------- +# ------------------------------------------------------------------- =pod