Merge pull request #1641 from JRDuncan/master

Allow creation of MSSQL identity columns with user specified seed and increment

Author: garas

src/Phinx/Db/Adapter/SqlServerAdapter.php

@@ -1186,7 +1186,9 @@ protected function getColumnSqlDefinition(Column $column, $create = true)
         }
 
         if ($column->isIdentity()) {
-            $buffer[] = 'IDENTITY(1, 1)';
+            $seed = $column->getSeed() ?: 1;
+            $increment = $column->getIncrement() ?: 1;
+            $buffer[] = sprintf('IDENTITY(%d,%d)', $seed, $increment);
         }
 
         return implode(' ', $buffer);

src/Phinx/Db/Table/Column.php

@@ -66,6 +66,16 @@ class Column
      */
     protected $identity = false;
 
+    /**
+     * @var integer
+     */
+    protected $seed;
+
+    /**
+     * @var integer
+     */
+    protected $increment;
+
     /**
      * @var integer
      */
@@ -349,6 +359,52 @@ public function getPrecision()
         return $this->limit;
     }
 
+    /**
+     * Gets the column identity seed.
+     *
+     * @return int
+     */
+    public function getSeed()
+    {
+        return $this->seed;
+    }
+
+    /**
+     * Gets the column identity increment.
+     *
+     * @return int
+     */
+    public function getIncrement()
+    {
+        return $this->increment;
+    }
+
+    /**
+     * Sets the column identity seed.
+     *
+     * @param int $seed Number seed
+     * @return \Phinx\Db\Table\Column
+     */
+    public function setSeed($seed)
+    {
+        $this->seed = $seed;
+
+        return $this;
+    }
+
+    /**
+     * Sets the column identity increment.
+     *
+     * @param int $increment Number increment
+     * @return \Phinx\Db\Table\Column
+     */
+    public function setIncrement($increment)
+    {
+        $this->increment = $increment;
+
+        return $this;
+    }
+
     /**
      * Sets the number scale for decimal or float column.
      *
@@ -627,6 +683,8 @@ protected function getValidOptions()
             'values',
             'collation',
             'encoding',
+            'seed',
+            'increment',
         ];
     }
 

tests/Phinx/Db/Adapter/SqlServerAdapterTest.php

@@ -144,7 +144,22 @@ public function testCreateTableCustomIdColumn()
         $this->assertTrue($this->adapter->hasColumn('ntable', 'email'));
         $this->assertFalse($this->adapter->hasColumn('ntable', 'address'));
     }
+    public function testCreateTableIdentityColumn()
+    {
+        $table = new \Phinx\Db\Table('ntable', ['id' => false, 'primary_key' => 'id'], $this->adapter);
+        $table->addColumn('id', 'integer', ['identity' => true, 'seed' => 1, 'increment' => 10 ])
+              ->save();
+        $this->assertTrue($this->adapter->hasTable('ntable'));
+        $this->assertTrue($this->adapter->hasColumn('ntable', 'id'));
 
+        $rows = $this->adapter->fetchAll("SELECT CAST(seed_value AS INT) seed_value, CAST(increment_value AS INT) increment_value
+FROM sys.columns c JOIN sys.tables t ON c.object_id=t.object_id
+JOIN sys.identity_columns ic ON c.object_id=ic.object_id AND c.column_id=ic.column_id
+WHERE t.name='ntable'");
+        $identity = $rows[0];
+        $this->assertEquals($identity['seed_value'], '1');
+        $this->assertEquals($identity['increment_value'], '10');
+    }
     public function testCreateTableWithNoPrimaryKey()
     {
         $options = [