Small change to comment.

[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Schema / Table.pm

diff --git a/lib/SQL/Translator/Schema/Table.pm b/lib/SQL/Translator/Schema/Table.pm
index 196b159..ae7f34e 100644 (file)
--- 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.6 2003-06-06 00:10:32 kycl4rk Exp $
+# $Id: Table.pm,v 1.8 2003-06-09 02:10:10 kycl4rk Exp $
 # ----------------------------------------------------------------------
 # Copyright (C) 2003 Ken Y. Clark <kclark@cpan.org>
 #
@@ -70,9 +70,9 @@ Object constructor.
 
     my ( $self, $config ) = @_;
     
-    for my $arg ( qw[ schema name ] ) {
+    for my $arg ( qw[ schema name comments ] ) {
         next unless defined $config->{ $arg };
-        $self->$arg( $config->{ $arg } ) or return;
+        defined $self->$arg( $config->{ $arg } ) or return;
     }
 
     return $self;
@@ -246,6 +246,36 @@ existing field, you will get an error and the field will not be created.
 }
 
 # ----------------------------------------------------------------------
+sub comments {
+
+=pod
+
+=head2 comments
+
+Get or set the comments on a table.  May be called several times to 
+set and it will accumulate the comments.  Called in an array context,
+returns each comment individually; called in a scalar context, returns
+all the comments joined on newlines.
+
+  $table->comments('foo');
+  $table->comments('bar');
+  print join( ', ', $table->comments ); # prints "foo, bar"
+
+=cut
+
+    my $self = shift;
+
+    for my $arg ( @_ ) {
+        $arg = $arg->[0] if ref $arg;
+        push @{ $self->{'comments'} }, $arg;
+    }
+
+    return wantarray 
+        ? @{ $self->{'comments'} || [] }
+        : join( "\n", @{ $self->{'comments'} || [] } );
+}
+
+# ----------------------------------------------------------------------
 sub get_constraints {
 
 =pod
@@ -473,7 +503,7 @@ These are eqivalent:
             $constraint = $self->add_constraint(
                 type   => PRIMARY_KEY,
                 fields => $fields,
-            );
+            ) or return;
         }
     }