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

add tests for table creation

Browse Source
This commit is contained in:
Matthew Bucci
2023-11-26 01:34:10 -08:00
parent a389cc7adc
commit 91f1612d10
2 changed files with 62 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pg4wp/rewriters/CreateTableSQLRewriter.php
+21 -4
View File
@@ -6,6 +6,7 @@ class CreateTableSQLRewriter extends AbstractSQLRewriter
'bigint(20)' => 'bigint',
'bigint(10)' => 'int',
'int(11)' => 'int',
'int(1)' => 'smallint',
'tinytext' => 'text',
'mediumtext' => 'text',
'longtext' => 'text',
@@ -45,22 +46,22 @@ class CreateTableSQLRewriter extends AbstractSQLRewriter
$sql = trim($sql) . ';';
// Translate types and some other replacements
$sql = str_replace(
$sql = str_ireplace(
array_keys($this->stringReplacements),
array_values($this->stringReplacements),
$sql
);
// Fix auto_increment by adding a sequence
$pattern = '/int[ ]+NOT NULL auto_increment/';
$pattern = '/int[ ]+NOT NULL auto_increment/i';
preg_match($pattern, $sql, $matches);
if($matches) {
$seq = $table . '_seq';
$sql = str_replace('NOT NULL auto_increment', "NOT NULL DEFAULT nextval('$seq'::text)", $sql);
$sql = str_ireplace('NOT NULL auto_increment', "NOT NULL DEFAULT nextval('$seq'::text)", $sql);
$sql .= "\nCREATE SEQUENCE $seq;";
}
// Support for INDEX creation
// Support for UNIQUE INDEX creation
$pattern = '/,\s+(UNIQUE |)KEY\s+([^\s]+)\s+\(((?:[\w]+(?:\([\d]+\))?[,]?)*)\)/';
if(preg_match_all($pattern, $sql, $matches, PREG_SET_ORDER)) {
foreach($matches as $match) {
@@ -75,6 +76,22 @@ class CreateTableSQLRewriter extends AbstractSQLRewriter
}
// Now remove handled indexes
$sql = preg_replace($pattern, '', $sql);
// Support for PRIMARY INDEX creation
$pattern = '/,\s+(PRIMARY |)KEY\s+\(((?:[\w]+(?:\([\d]+\))?[,]?)*)\)/';
if(preg_match_all($pattern, $sql, $matches, PREG_SET_ORDER)) {
foreach($matches as $match) {
$primary = $match[1];
$columns = $match[2];
$columns = preg_replace('/\(\d+\)/', '', $columns);
$index = $columns;
// Workaround for index name duplicate
$index = $table . '_' . $index;
$sql .= "\nCREATE {$primary}INDEX $index ON $table ($columns);";
}
}
// Now remove handled indexes
$sql = preg_replace($pattern, '', $sql);
return $sql;
}