feedc0de/postgresql-for-wordpress
1
0
Fork 0
forked from PostgreSQL-For-Wordpress/postgresql-for-wordpress
Code Pull Requests Activity

improve primary key detection to use first column found if no primary key found

Browse Source
This commit is contained in:
Matthew Bucci
2024-02-26 23:44:17 -08:00
parent b92d8dde92
commit 2d5f178426
1 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
pg4wp/driver_pgsql.php
View File

@ -469,18 +469,30 @@ function wpsqli_rollback(&$connection, $flags = 0, $name = null)
function get_primary_key_for_table(&$connection, $table)
{
$query = <<<SQL
SELECT a.attname
FROM pg_index i
JOIN pg_attribute a ON a.attrelid = i.indrelid
AND a.attnum = ANY(i.indkey)
SELECT a.attname, i.indisprimary
FROM pg_index i
JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
WHERE i.indrelid = '$table'::regclass
AND i.indisprimary
SQL;
$result = pg_query($connection, $query);
if (!$result) {
return null;
}
$row = pg_fetch_row($result);
return $row[0];
$firstRow = null;
while ($row = pg_fetch_row($result)) {
if ($firstRow === null) {
$firstRow = $row; // Save the first row in case no match is found
}
if ($row[1] == true) {
return $row[0]; // Return the first row where $row[1] == true
}
}
// If no row where $row[1] == true was found, return the first row encountered
return $firstRow ? $firstRow[0] : null;
}
/**