X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FParser%2FSQLServer.pm;h=2abd91e9547c1c8165b61edad4e4ee303c9e04d1;hb=821a0fde221f5accf93e3f65efa77b5a6733cb5e;hp=fdeeffeb7b47f3608b31032d88c59e35282383e1;hpb=7cfa0d36eae6642a3f26bd597c340aa5eab7cd44;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Parser/SQLServer.pm b/lib/SQL/Translator/Parser/SQLServer.pm index fdeeffe..2abd91e 100644 --- a/lib/SQL/Translator/Parser/SQLServer.pm +++ b/lib/SQL/Translator/Parser/SQLServer.pm @@ -1,7 +1,7 @@ package SQL::Translator::Parser::SQLServer; # ------------------------------------------------------------------- -# $Id: SQLServer.pm,v 1.3 2005-06-28 23:37:12 duality72 Exp $ +# $Id$ # ------------------------------------------------------------------- # Copyright (C) 2002-4 SQLFairy Authors # @@ -39,7 +39,7 @@ should probably be considered a work in progress. use strict; use vars qw[ $DEBUG $VERSION $GRAMMAR @EXPORT_OK ]; -$VERSION = sprintf "%d.%02d", q$Revision: 1.3 $ =~ /(\d+)\.(\d+)/; +$VERSION = sprintf "%d.%02d", q$Revision$ =~ /(\d+)\.(\d+)/; $DEBUG = 0 unless defined $DEBUG; use Data::Dumper; @@ -56,15 +56,23 @@ $::RD_HINT = 1; $GRAMMAR = q{ { - my ( %tables, @table_comments, $table_order ); + my ( %tables, @table_comments, $table_order, %procedures, $proc_order, %views, $view_order ); } -startrule : statement(s) eofile { \%tables } +startrule : statement(s) eofile + { + return { + tables => \%tables, + procedures => \%procedures, + views => \%views, + } + } eofile : /^\Z/ statement : create_table | create_procedure + | create_view | create_index | create_constraint | comment @@ -129,7 +137,7 @@ comment_middle : m{([^*]+|\*(?!/))*} # # Create table. # -create_table : /create/i /table/i ident '(' create_def(s /,/) ')' lock(?) on_system(?) ';' GO(?) +create_table : /create/i /table/i ident '(' create_def(s /,/) ')' lock(?) on_system(?) END_STATEMENT { my $table_owner = $item[3]{'owner'}; my $table_name = $item[3]{'name'}; @@ -180,14 +188,47 @@ create_index : /create/i index push @{ $tables{ $item[2]{'table'} }{'indices'} }, $item[2]; } -create_procedure : /create/i /procedure/i procedure_body GO +create_procedure : /create/i PROCEDURE WORD not_go GO { @table_comments = (); + my $proc_name = $item[3]; + my $owner = ''; + my $sql = "$item[1] $item[2] $proc_name $item[4]"; + + $procedures{ $proc_name }{'order'} = ++$proc_order; + $procedures{ $proc_name }{'name'} = $proc_name; + $procedures{ $proc_name }{'owner'} = $owner; + $procedures{ $proc_name }{'sql'} = $sql; } -procedure_body : not_go(s) +create_procedure : /create/i PROCEDURE '[' WORD '].' WORD not_go GO + { + @table_comments = (); + my $proc_name = $item[6]; + my $owner = $item[4]; + my $sql = "$item[1] $item[2] [$owner].$proc_name $item[7]"; + + $procedures{ $proc_name }{'order'} = ++$proc_order; + $procedures{ $proc_name }{'name'} = $proc_name; + $procedures{ $proc_name }{'owner'} = $owner; + $procedures{ $proc_name }{'sql'} = $sql; + } -not_go : /((?!go).)*/ +PROCEDURE : /procedure/i + | /function/i + +create_view : /create/i /view/i WORD not_go GO + { + @table_comments = (); + my $view_name = $item[3]; + my $sql = "$item[1] $item[2] $item[3] $item[4]"; + + $views{ $view_name }{'order'} = ++$view_order; + $views{ $view_name }{'name'} = $view_name; + $views{ $view_name }{'sql'} = $sql; + } + +not_go : /((?!\bgo\b).)*/is create_def : constraint | index @@ -268,7 +309,9 @@ nullable : /not/i /null/i | /null/i { $return = 1 } -default_val : /default/i /(?:')?[^']*(?:')?/ +default_val : /default/i /null/i + { $return = 'null' } + | /default/i /'[^']*'/ { $item[2]=~ s/'//g; $return = $item[2] } auto_inc : /identity/i { 1 } @@ -315,9 +358,10 @@ on_delete : /on delete/i reference_option on_update : /on update/i reference_option { $item[2] } -reference_option: /cascade/i | - /no action/i - { $item[1] } +reference_option: /cascade/i + { $item[1] } + | /no action/i + { $item[1] } clustered : /clustered/i { $return = 1 } @@ -352,6 +396,9 @@ ident : QUOTE(?) WORD '.' WORD QUOTE(?) | WORD { $return = { name => $item[1] } } +END_STATEMENT : ';' + | GO + GO : /^go/i NAME : QUOTE(?) /\w+/ QUOTE(?) @@ -386,11 +433,11 @@ sub parse { my $schema = $translator->schema; my @tables = sort { - $result->{ $a }->{'order'} <=> $result->{ $b }->{'order'} - } keys %{ $result }; + $result->{tables}->{ $a }->{'order'} <=> $result->{tables}->{ $b }->{'order'} + } keys %{ $result->{tables} }; for my $table_name ( @tables ) { - my $tdata = $result->{ $table_name }; + my $tdata = $result->{tables}->{ $table_name }; my $table = $schema->add_table( name => $tdata->{'name'} ) or die "Can't create table '$table_name': ", $schema->error; @@ -460,6 +507,27 @@ sub parse { ) or die $table->error; } } + + my @procedures = sort { + $result->{procedures}->{ $a }->{'order'} <=> $result->{procedures}->{ $b }->{'order'} + } keys %{ $result->{procedures} }; + for my $proc_name (@procedures) { + $schema->add_procedure( + name => $proc_name, + owner => $result->{procedures}->{$proc_name}->{owner}, + sql => $result->{procedures}->{$proc_name}->{sql}, + ); + } + + my @views = sort { + $result->{views}->{ $a }->{'order'} <=> $result->{views}->{ $b }->{'order'} + } keys %{ $result->{views} }; + for my $view_name (keys %{ $result->{views} }) { + $schema->add_view( + name => $view_name, + sql => $result->{views}->{$view_name}->{sql}, + ); + } return 1; }