fixed for multiple primary key

tsucchi · Jan 18, 2017 · a0b8b5d · a0b8b5d
1 parent a18e066
commit a0b8b5d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/Otogiri/Plugin/TableInfo/Pg.pm b/lib/Otogiri/Plugin/TableInfo/Pg.pm
@@ -117,10 +117,12 @@ sub _is_sequence_column {
 
 sub _build_pk_defs {
     my ($self, $table) = @_;
-    my $result = "";
-    for my $column ( $table->primary_key() ) {
-        $result .= "ALTER TABLE ONLY " . $table->name . "\n";
-        $result .= "    ADD CONSTRAINT " . $column->{PG_COLUMN} . " PRIMARY KEY (" . $column->name . ");\n";
+    my @pks = $table->primary_key();
+    my $result = '';
+    if ( @pks ) {
+        my @pk_names = map { $_->name } @pks;
+        $result = "ALTER TABLE ONLY " . $table->name . "\n";
+        $result .= "    ADD CONSTRAINT " . $pks[0]->{PG_COLUMN} . " PRIMARY KEY (" . join(', ', @pk_names) . ");\n";
     }
     return $result;
 }