X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FParser%2FPostgreSQL.pm;h=6fcd1746457800e78705efb75730d15e5624be8a;hb=734dfc91a0ea94092e5aef80874c46ab1e8d718c;hp=f488b29f643de46ec2cec72b8fa5237b7409c927;hpb=3022f45b20e1eda2d73a4815aa88c779cd53c602;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Parser/PostgreSQL.pm b/lib/SQL/Translator/Parser/PostgreSQL.pm index f488b29..6fcd174 100644 --- a/lib/SQL/Translator/Parser/PostgreSQL.pm +++ b/lib/SQL/Translator/Parser/PostgreSQL.pm @@ -1,7 +1,7 @@ package SQL::Translator::Parser::PostgreSQL; # ------------------------------------------------------------------- -# $Id: PostgreSQL.pm,v 1.13 2003-05-03 04:09:50 kycl4rk Exp $ +# $Id: PostgreSQL.pm,v 1.27 2003-08-17 00:46:23 rossta Exp $ # ------------------------------------------------------------------- # Copyright (C) 2003 Ken Y. Clark , # Allen Day , @@ -111,7 +111,7 @@ View table: use strict; use vars qw[ $DEBUG $VERSION $GRAMMAR @EXPORT_OK ]; -$VERSION = sprintf "%d.%02d", q$Revision: 1.13 $ =~ /(\d+)\.(\d+)/; +$VERSION = sprintf "%d.%02d", q$Revision: 1.27 $ =~ /(\d+)\.(\d+)/; $DEBUG = 0 unless defined $DEBUG; use Data::Dumper; @@ -149,36 +149,42 @@ statement : create | grant | revoke | drop + | insert | connect + | update | set | connect : /^\s*\\\connect.*\n/ -set : /SET/ /[^;]*/ ';' +set : /set/i /[^;]*/ ';' -revoke : /revoke/i WORD(s /,/) /on/i table_name /from/i name_with_opt_quotes(s /,/) ';' +revoke : /revoke/i WORD(s /,/) /on/i TABLE(?) table_name /from/i name_with_opt_quotes(s /,/) ';' { my $table_name = $item{'table_name'}; push @{ $tables{ $table_name }{'permissions'} }, { type => 'revoke', actions => $item[2], - users => $item[6], + users => $item[7], } } -grant : /grant/i WORD(s /,/) /on/i table_name /to/i name_with_opt_quotes(s /,/) ';' +grant : /grant/i WORD(s /,/) /on/i TABLE(?) table_name /to/i name_with_opt_quotes(s /,/) ';' { my $table_name = $item{'table_name'}; push @{ $tables{ $table_name }{'permissions'} }, { type => 'grant', actions => $item[2], - users => $item[6], + users => $item[7], } } drop : /drop/i /[^;]*/ ';' +insert : /insert/i /[^;]*/ ';' + +update : /update/i /[^;]*/ ';' + # # Create table. # @@ -206,17 +212,17 @@ create : create_table table_name '(' create_definition(s /,/) ')' table_option(s elsif ( $definition->{'type'} eq 'constraint' ) { $definition->{'type'} = $definition->{'constraint_type'}; # group FKs at the field level - if ( $definition->{'type'} eq 'foreign_key' ) { - for my $fld ( @{ $definition->{'fields'} || [] } ) { - push @{ - $tables{$table_name}{'fields'}{$fld}{'constraints'} - }, $definition; - } - } - else { +# if ( $definition->{'type'} eq 'foreign_key' ) { +# for my $fld ( @{ $definition->{'fields'} || [] } ) { +# push @{ +# $tables{$table_name}{'fields'}{$fld}{'constraints'} +# }, $definition; +# } +# } +# else { push @{ $tables{ $table_name }{'constraints'} }, $definition; - } +# } } else { push @{ $tables{ $table_name }{'indices'} }, $definition; @@ -264,13 +270,22 @@ comment : /^\s*(?:#|-{2}).*\n/ field : comment(s?) field_name data_type field_meta(s?) comment(s?) { my ( $default, @constraints, $is_pk ); + my $null = 1; for my $meta ( @{ $item[4] } ) { - $default = $meta if $meta->{'meta_type'} eq 'default'; - push @constraints, $meta if $meta->{'meta_type'} eq 'constraint'; - $is_pk = $meta->{'type'} eq 'primary_key'; - } + if ( $meta->{'type'} eq 'default' ) { + $default = $meta; + next; + } + elsif ( $meta->{'type'} eq 'not_null' ) { + $null = 0; +# next; + } + elsif ( $meta->{'type'} eq 'primary_key' ) { + $is_pk = 1; + } - my $null = ( grep { $_->{'type'} eq 'not_null' } @constraints ) ? 0 : 1; + push @constraints, $meta if $meta->{'supertype'} eq 'constraint'; + } my @comments = ( @{ $item[1] }, @{ $item[5] } ); @@ -279,7 +294,6 @@ field : comment(s?) field_name data_type field_meta(s?) comment(s?) name => $item{'field_name'}, data_type => $item{'data_type'}{'type'}, size => $item{'data_type'}{'size'}, - list => $item{'data_type'}{'list'}, null => $null, default => $default->{'value'}, constraints => [ @constraints ], @@ -290,8 +304,7 @@ field : comment(s?) field_name data_type field_meta(s?) comment(s?) | field_meta : default_val - | - column_constraint + | column_constraint column_constraint : constraint_name(?) column_constraint_type deferrable(?) deferred(?) { @@ -301,11 +314,11 @@ column_constraint : constraint_name(?) column_constraint_type deferrable(?) defe my $expression = $desc->{'expression'} || ''; $return = { - meta_type => 'constraint', + supertype => 'constraint', name => $item{'constraint_name'}[0] || '', type => $type, expression => $type eq 'check' ? $expression : '', - deferreable => $item{'deferrable'}, + deferrable => $item{'deferrable'}, deferred => $item{'deferred'}, reference_table => $desc->{'reference_table'}, reference_fields => $desc->{'reference_fields'}, @@ -319,10 +332,10 @@ constraint_name : /constraint/i name_with_opt_quotes { $item[2] } column_constraint_type : /not null/i { $return = { type => 'not_null' } } | - /null/ + /null/i { $return = { type => 'null' } } | - /unique/ + /unique/i { $return = { type => 'unique' } } | /primary key/i @@ -372,96 +385,95 @@ data_type : pg_data_type parens_value_list(?) } pg_data_type : - /(bigint|int8|bigserial|serial8)/ + /(bigint|int8)/i { $return = { - type => 'integer', - size => 8, - auto_increment => 1, + type => 'integer', + size => [8], }; } | - /(smallint|int2)/ + /(smallint|int2)/i { $return = { type => 'integer', - size => 2, + size => [2], }; } | - /int(eger)?|int4/ + /(integer|int4?)/i # interval must come before this { $return = { type => 'integer', - size => 4, + size => [4], }; } - | - /(double precision|float8?)/ + | + /(real|float4)/i { $return = { - type => 'float', - size => 8, - }; + type => 'real', + size => [4], + }; } | - /(real|float4)/ + /(double precision|float8?)/i { $return = { - type => 'real', - size => 4, - }; + type => 'float', + size => [8], + }; } | - /serial4?/ + /(bigserial|serial8)/i { $return = { - type => 'integer', - size => 4, + type => 'integer', + size => [8], auto_increment => 1, }; } | - /bigserial/ + /serial4?/i { $return = { - type => 'integer', - size => 8, + type => 'integer', + size => [4], auto_increment => 1, }; } | - /(bit varying|varbit)/ + /(bit varying|varbit)/i { $return = { type => 'varbit' }; } | - /character varying/ + /character varying/i { $return = { type => 'varchar' }; } | - /char(acter)?/ + /char(acter)?/i { $return = { type => 'char' }; } | - /bool(ean)?/ + /bool(ean)?/i { $return = { type => 'boolean' }; } | - /(bytea|binary data)/ + /bytea/i { - $return = { type => 'binary' }; + $return = { type => 'bytea' }; } | - /timestampz?/ + /(timestamptz|timestamp)/i { $return = { type => 'timestamp' }; } | - /(bit|box|cidr|circle|date|inet|interval|line|lseg|macaddr|money|numeric|decimal|path|point|polygon|text|time|varchar)/ + /(bit|box|cidr|circle|date|inet|interval|line|lseg|macaddr|money|numeric|decimal|path|point|polygon|text|timetz|time|varchar)/i { $return = { type => $item[1] }; } @@ -493,7 +505,7 @@ table_constraint : comment(s?) constraint_name(?) table_constraint_type deferrab constraint_type => $type, fields => $type ne 'check' ? $fields : [], expression => $type eq 'check' ? $expression : '', - deferreable => $item{'deferrable'}, + deferrable => $item{'deferrable'}, deferred => $item{'deferred'}, reference_table => $desc->{'reference_table'}, reference_fields => $desc->{'reference_fields'}, @@ -520,7 +532,7 @@ table_constraint_type : /primary key/i '(' name_with_opt_quotes(s /,/) ')' } } | - /check/ '(' /(.+)/ ')' + /check/i '(' /(.+)/ ')' { $return = { type => 'check', @@ -600,19 +612,28 @@ alter_table : /alter/i /table/i only(?) only : /only/i -create_table : /create/i /table/i +create_table : /create/i TABLE create_index : /create/i /index/i -default_val : /default/i /(?:')?[\w\d.-]*(?:')?/ +default_val : /default/i /(\d+|'[^']*'|\w+\(.*?\))|\w+/ { - my $val = $item[2] || ''; - $val =~ s/'//g; + my $val = defined $item[2] ? $item[2] : ''; + $val =~ s/^'|'$//g; $return = { - meta_type => 'default', + supertype => 'constraint', + type => 'default', value => $val, } } + | /null/i + { + $return = { + supertype => 'constraint', + type => 'default', + value => 'NULL', + } + } name_with_opt_paren : NAME parens_value_list(s?) { $item[2][0] ? "$item[1]($item[2][0][0])" : $item[1] } @@ -631,6 +652,8 @@ table_option : /inherits/i '(' name_with_opt_quotes(s /,/) ')' $return = { type => $item[1] =~ /out/i ? 'without_oids' : 'with_oids' } } +TABLE : /table/i + SEMICOLON : /\s*;\n?/ WORD : /\w+/ @@ -650,7 +673,7 @@ VALUE : /[-+]?\.?\d+(?:[eE]\d+)?/ { $item[1] } | /'.*?'/ # XXX doesn't handle embedded quotes { $item[1] } - | /NULL/ + | /null/i { 'NULL' } !; @@ -671,15 +694,75 @@ sub parse { my $result = $parser->startrule($data); die "Parse failed.\n" unless defined $result; warn Dumper($result) if $DEBUG; - return $result; + + my $schema = $translator->schema; + 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 "Couldn't create table '$table_name': " . $schema->error; + + my @fields = sort { + $tdata->{'fields'}->{$a}->{'order'} + <=> + $tdata->{'fields'}->{$b}->{'order'} + } keys %{ $tdata->{'fields'} }; + + for my $fname ( @fields ) { + my $fdata = $tdata->{'fields'}{ $fname }; + my $field = $table->add_field( + name => $fdata->{'name'}, + data_type => $fdata->{'data_type'}, + size => $fdata->{'size'}, + default_value => $fdata->{'default'}, + is_auto_increment => $fdata->{'is_auto_inc'}, + is_nullable => $fdata->{'null'}, + ) or die $table->error; + + $table->primary_key( $field->name ) if $fdata->{'is_primary_key'}; + + 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 1; } 1; -#----------------------------------------------------- -# Where man is not nature is barren. -# William Blake -#----------------------------------------------------- +# ------------------------------------------------------------------- +# Rescue the drowning and tie your shoestrings. +# Henry David Thoreau +# ------------------------------------------------------------------- =pod