Added checks for existing columns

felixwillmann · Jul 15, 2015 · 89eafeb · 89eafeb
1 parent 67ff180
commit 89eafeb
Showing 1 changed file with 32 additions and 23 deletions.
diff --git a/src/CompositeNumberRangeBehavior.php b/src/CompositeNumberRangeBehavior.php
@@ -73,29 +73,38 @@ public function modifyTable()
             }
         }
 
-        $foreignIdColumn = $table->addColumn(
-            array(
-                'name' => $foreignIdColumnName,
-                'type' => 'integer',
-                'required' => true
-            )
-        );
-
-        $compositeKeyForeignKeyName = $tableName . '_FK_' . $foreignIdColumnName;
-        $foreignKey = new ForeignKey($compositeKeyForeignKeyName);
-        $foreignKey->addReference($foreignIdColumnName, 'id');
-        $foreignKey->setForeignTableCommonName($foreignTableName);
-        $foreignKey->setOnUpdate(ForeignKey::CASCADE);
-        $foreignKey->setOnDelete(ForeignKey::CASCADE);
-        $table->addForeignKey($foreignKey);
-
-        $compositeKeyColumn = $table->addColumn(
-            array(
-                'name' => $compositeKeyColumnName,
-                'type' => 'integer',
-                'required' => false,
-            )
-        );
+        if ($table->hasColumn($foreignIdColumnName)) {
+            $foreignIdColumn = $table->getColumn($foreignIdColumnName);
+        } else {
+            $foreignIdColumn = $table->addColumn(
+                array(
+                    'name' => $foreignIdColumnName,
+                    'type' => 'integer',
+                    'required' => true
+                )
+            );
+
+            $compositeKeyForeignKeyName = $tableName . '_FK_' . $foreignIdColumnName;
+            $foreignKey = new ForeignKey($compositeKeyForeignKeyName);
+            $foreignKey->addReference($foreignIdColumnName, 'id');
+            $foreignKey->setForeignTableCommonName($foreignTableName);
+            $foreignKey->setOnUpdate(ForeignKey::CASCADE);
+            $foreignKey->setOnDelete(ForeignKey::CASCADE);
+            $table->addForeignKey($foreignKey);
+
+        }
+
+        if ($table->hasColumn($compositeKeyColumnName)) {
+            $compositeKeyColumn = $table->getColumn($compositeKeyColumnName);
+        } else {
+            $compositeKeyColumn = $table->addColumn(
+                array(
+                    'name' => $compositeKeyColumnName,
+                    'type' => 'integer',
+                    'required' => false,
+                )
+            );
+        }
 
         $index = new Unique($tableName . '_UQ_' . $foreignIdColumnName . '_' . $compositeKeyColumnName);
         $index->addColumn($foreignIdColumn);