Structure::getPrimaryKeySequence(): fixed retrieving sequence name if table has composite PK

Author: Unlink

src/Database/Structure.php

@@ -74,13 +74,15 @@ public function getPrimaryKeySequence($table)
 		}
 
 		$primary = $this->getPrimaryKey($table);
-		if (!$primary || is_array($primary)) {
+		if (!$primary) {
 			return NULL;
 		}
 
-		foreach ($this->structure['columns'][$table] as $columnMeta) {
-			if ($columnMeta['name'] === $primary) {
-				return isset($columnMeta['vendor']['sequence']) ? $columnMeta['vendor']['sequence'] : NULL;
+		foreach ((array) $primary as $key) {
+			foreach ($this->structure['columns'][$table] as $columnMeta) {
+				if ($columnMeta['name'] === $key) {
+					return isset($columnMeta['vendor']['sequence']) ? $columnMeta['vendor']['sequence'] : NULL;
+				}
 			}
 		}
 

tests/Database/Table/bugs/Selection.insert().phpt

@@ -38,8 +38,8 @@ test(function () use ($context) {
 //Insert into table with composite primary key and one of them is auto_increment
 test(function () use ($context, $driverName) {
 
-	//Sqlite doesn't allow this type of table and sqlsrv's driver don't implement reflection, i don't know now why this not working in pgsql
-	if ($driverName == 'mysql') {
+	//Sqlite doesn't allow this type of table and sqlsrv's driver don't implement reflection
+	if ($driverName == 'mysql' || $driverName == 'pgsql') {
 		$inserted = $context->table('composite_pk_ai')->insert([
 			'id2' => 10,
 			'name' => 'Michal'