remove all columns before adding new ones via ->fields or ->columns

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Object / Compat.pm

diff --git a/lib/SQL/Translator/Object/Compat.pm b/lib/SQL/Translator/Object/Compat.pm
index e85b9b8..8f2389f 100644 (file)
--- a/lib/SQL/Translator/Object/Compat.pm
+++ b/lib/SQL/Translator/Object/Compat.pm
@@ -6,6 +6,7 @@ role SQL::Translator::Object::Compat {
     use MooseX::MultiMethods;
 
     multi method fields(Str $columns) {
+        $self->clear_columns;
         my @columns = split /\s*,\s*/, $columns;
         for my $column (@columns) {
             die "Column '$column' does not exist!" unless $self->table->exists_column($column);
@@ -15,6 +16,7 @@ role SQL::Translator::Object::Compat {
     }
 
     multi method fields(ArrayRef $columns) {
+        $self->clear_columns;
         for my $column (@$columns) {
             die "Column '$column' does not exist!" unless $self->table->exists_column($column);
             $self->add_column($self->table->get_column($column));