X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FSchema%2FTable.pm;h=2e9f2531633cd459eb06c7e81d9568f099eed847;hb=f9c5e794de9710a4aa68716d49da47316834d8c8;hp=047699288ffa34e6461a14f2523ba4a7b0939cfe;hpb=69c7a62f38fa4519a8cdb14bed4af37f2c6206ff;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Schema/Table.pm b/lib/SQL/Translator/Schema/Table.pm index 0476992..2e9f253 100644 --- a/lib/SQL/Translator/Schema/Table.pm +++ b/lib/SQL/Translator/Schema/Table.pm @@ -1,7 +1,7 @@ package SQL::Translator::Schema::Table; # ---------------------------------------------------------------------- -# $Id: Table.pm,v 1.15 2003-08-29 05:38:56 allenday Exp $ +# $Id: Table.pm,v 1.21 2003-09-25 17:28:37 allenday Exp $ # ---------------------------------------------------------------------- # Copyright (C) 2003 Ken Y. Clark # @@ -46,11 +46,12 @@ use SQL::Translator::Schema::Constants; use SQL::Translator::Schema::Constraint; use SQL::Translator::Schema::Field; use SQL::Translator::Schema::Index; +use Data::Dumper; use base 'Class::Base'; use vars qw( $VERSION $FIELD_ORDER ); -$VERSION = sprintf "%d.%02d", q$Revision: 1.15 $ =~ /(\d+)\.(\d+)/; +$VERSION = sprintf "%d.%02d", q$Revision: 1.21 $ =~ /(\d+)\.(\d+)/; # ---------------------------------------------------------------------- sub init { @@ -88,10 +89,10 @@ sub add_constraint { Add a constraint to the table. Returns the newly created C object. - my $c1 = $table->add_constraint( - name => 'pk', - type => PRIMARY_KEY, - fields => [ 'foo_id' ], + my $c1 = $table->add_constraint( + name => 'pk', + type => PRIMARY_KEY, + fields => [ 'foo_id' ], ); my $c2 = SQL::Translator::Schema::Constraint->new( name => 'uniq' ); @@ -171,7 +172,7 @@ sub add_index { Add an index to the table. Returns the newly created C object. - my $i1 = $table->add_index( + my $i1 = $table->add_index( name => 'name', fields => [ 'name' ], type => 'normal', @@ -213,17 +214,17 @@ C object. The "name" parameter is required. If you try to create a field with the same name as an existing field, you will get an error and the field will not be created. - my $f1 = $table->add_field( + my $f1 = $table->add_field( name => 'foo_id', data_type => 'integer', size => 11, ); - my $f2 = SQL::Translator::Schema::Field->new( + my $f2 = SQL::Translator::Schema::Field->new( name => 'name', table => $table, ); - $f2 = $table->add_field( $field2 ) or die $table->error; + $f2 = $table->add_field( $field2 ) or die $table->error; =cut @@ -246,7 +247,7 @@ existing field, you will get an error and the field will not be created. my $field_name = $field->name or return $self->error('No name'); if ( exists $self->{'fields'}{ $field_name } ) { - return $self->error(qq[Can\'t create field: "$field_name" exists]); + return $self->error(qq[Can't create field: "$field_name" exists]); } else { $self->{'fields'}{ $field_name } = $field; @@ -418,22 +419,62 @@ Determine whether the view is valid or not. return 1; } -sub is_data { - my $self = shift; - return $self->{'is_data'} if defined $self->{'is_data'}; +# ---------------------------------------------------------------------- +sub is_trivial_link { + +=pod + +=head2 is_data - $self->{'is_data'} = 0; +=cut - foreach my $field ($self->get_fields){ - if(!$field->is_primary_key or !$field->is_foreign_key){ - $self->{'is_data'} = 1; - return $self->{'is_data'} + my $self = shift; + return 0 if $self->is_data; + return $self->{'is_trivial_link'} if defined $self->{'is_trivial_link'}; + + $self->{'is_trivial_link'} = 1; + + my %fk = (); + + foreach my $field ( $self->get_fields ) { + next unless $field->is_foreign_key; + $fk{$field->foreign_key_reference->reference_table}++; } - } - return $self->{'is_data'}; + foreach my $referenced (keys %fk){ + if($fk{$referenced} > 1){ + $self->{'is_trivial_link'} = 0; + last; + } + + return $self->{'is_trivial_link'}; + +} + +sub is_data { + +=pod + +=head2 is_data + +=cut + + my $self = shift; + return $self->{'is_data'} if defined $self->{'is_data'}; + + $self->{'is_data'} = 0; + + foreach my $field ( $self->get_fields ) { + if ( !$field->is_primary_key and !$field->is_foreign_key ) { + $self->{'is_data'} = 1; + return $self->{'is_data'}; + } + } + + return $self->{'is_data'}; } +# ---------------------------------------------------------------------- sub can_link { =pod @@ -446,42 +487,83 @@ Determine whether the table can link two arg tables via many-to-many. =cut - my($self,$table1,$table2) = @_; + my ( $self, $table1, $table2 ) = @_; - #get tables in abc order - ($table1,$table2) = sort {$a->name cmp $b->name} ($table1,$table2); + return $self->{'can_link'}{ $table1->name }{ $table2->name } + if defined $self->{'can_link'}{ $table1->name }{ $table2->name }; - return $self->{'can_link'}{$table1->name}{$table2->name} if defined $self->{'can_link'}{$table1->name}{$table2->name}; + if ( $self->is_data == 1 ) { + $self->{'can_link'}{ $table1->name }{ $table2->name } = [0]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = [0]; + return $self->{'can_link'}{ $table1->name }{ $table2->name }; + } - if($self->is_data == 1){ - $self->{'can_link'}{$table1->name}{$table2->name} = 0; - return $self->{'can_link'}{$table1->name}{$table2->name}; - } + my %fk = (); - my %fk = (); + foreach my $field ( $self->get_fields ) { + if ( $field->is_foreign_key ) { + push @{ $fk{ $field->foreign_key_reference->reference_table } }, + $field->foreign_key_reference; + } + } - foreach my $field ($self->get_fields){ - #if the table has non-key fields, it can't be a link - if(!$field->is_primary_key or !$field->is_foreign_key){ - $self->{'can_link'}{$table1->name}{$table2->name} = 0; - return $self->{'can_link'}{$table1->name}{$table2->name}; + if ( !defined( $fk{ $table1->name } ) or !defined( $fk{ $table2->name } ) ) + { + $self->{'can_link'}{ $table1->name }{ $table2->name } = [0]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = [0]; + return $self->{'can_link'}{ $table1->name }{ $table2->name }; + } - #otherwise, count up how many fields refer to each FK table.field - } elsif($field->is_foreign_key){ - $fk{$field->foreign_key_reference->reference_table->name}++; - } - } - - if($fk{ $table1->name } == 1 - and - $fk{ $table2->name } == 1 - ){ - $self->{'can_link'}{$table1->name}{$table2->name} = 1; - } else { - $self->{'can_link'}{$table1->name}{$table2->name} = 0; - } - - return $self->{'can_link'}{$table1->name}{$table2->name}; + # trivial traversal, only one way to link the two tables + if ( scalar( @{ $fk{ $table1->name } } == 1 ) + and scalar( @{ $fk{ $table2->name } } == 1 ) ) + { + $self->{'can_link'}{ $table1->name }{ $table2->name } = + [ 'one2one', $fk{ $table1->name }, $fk{ $table2->name } ]; + $self->{'can_link'}{ $table1->name }{ $table2->name } = + [ 'one2one', $fk{ $table2->name }, $fk{ $table1->name } ]; + + # non-trivial traversal. one way to link table2, + # many ways to link table1 + } + elsif ( scalar( @{ $fk{ $table1->name } } > 1 ) + and scalar( @{ $fk{ $table2->name } } == 1 ) ) + { + $self->{'can_link'}{ $table1->name }{ $table2->name } = + [ 'many2one', $fk{ $table1->name }, $fk{ $table2->name } ]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = + [ 'one2many', $fk{ $table2->name }, $fk{ $table1->name } ]; + + # non-trivial traversal. one way to link table1, + # many ways to link table2 + } + elsif ( scalar( @{ $fk{ $table1->name } } == 1 ) + and scalar( @{ $fk{ $table2->name } } > 1 ) ) + { + $self->{'can_link'}{ $table1->name }{ $table2->name } = + [ 'one2many', $fk{ $table1->name }, $fk{ $table2->name } ]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = + [ 'many2one', $fk{ $table2->name }, $fk{ $table1->name } ]; + + # non-trivial traversal. many ways to link table1 and table2 + } + elsif ( scalar( @{ $fk{ $table1->name } } > 1 ) + and scalar( @{ $fk{ $table2->name } } > 1 ) ) + { + $self->{'can_link'}{ $table1->name }{ $table2->name } = + [ 'many2many', $fk{ $table1->name }, $fk{ $table2->name } ]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = + [ 'many2many', $fk{ $table2->name }, $fk{ $table1->name } ]; + + # one of the tables didn't export a key + # to this table, no linking possible + } + else { + $self->{'can_link'}{ $table1->name }{ $table2->name } = [0]; + $self->{'can_link'}{ $table2->name }{ $table1->name } = [0]; + } + + return $self->{'can_link'}{ $table1->name }{ $table2->name }; } # ---------------------------------------------------------------------- @@ -491,7 +573,7 @@ sub name { =head2 name -Get or set the table\'s name. +Get or set the table's name. If provided an argument, checks the schema object for a table of that name and disallows the change if one exists. @@ -504,7 +586,7 @@ that name and disallows the change if one exists. if ( my $arg = shift ) { if ( my $schema = $self->schema ) { - return $self->error( qq[Can\'t use table name "$arg": table exists] ) + return $self->error( qq[Can't use table name "$arg": table exists] ) if $schema->get_table( $arg ); } $self->{'name'} = $arg; @@ -520,7 +602,7 @@ sub schema { =head2 schema -Get or set the table\'s schema object. +Get or set the table's schema object. my $schema = $table->schema; @@ -541,9 +623,9 @@ sub primary_key { =pod -=head2 options +=head2 primary_key -Gets or sets the table\'s primary key(s). Takes one or more field +Gets or sets the table's primary key(s). Takes one or more field names (as a string, list or array[ref]) as an argument. If the field names are present, it will create a new PK if none exists, or it will add to the fields of an existing PK (and will unique the field names). @@ -609,7 +691,7 @@ sub options { =head2 options -Get or set the table\'s options (e.g., table types for MySQL). Returns +Get or set the table's options (e.g., table types for MySQL). Returns an array or array reference. my @options = $table->options; @@ -636,7 +718,7 @@ sub order { =head2 order -Get or set the table\'s order. +Get or set the table's order. my $order = $table->order(3); @@ -666,8 +748,9 @@ sub DESTROY { =pod -=head1 AUTHOR +=head1 AUTHORS -Ken Y. Clark Ekclark@cpan.orgE +Ken Y. Clark Ekclark@cpan.orgE, +Allen Day Eallenday@ucla.eduE. =cut