add support to schema modification with multi-column primary key

This commit is contained in:
fabrixxm 2015-12-08 10:35:08 +01:00
parent 95508bf4bb
commit f97e20acd3

View file

@ -142,26 +142,37 @@ function update_structure($verbose, $action, $tables=null, $definition=null) {
// Get the definition // Get the definition
if (is_null($definition)) if (is_null($definition))
$definition = db_definition(); $definition = db_definition();
// Compare it // Compare it
foreach ($definition AS $name => $structure) { foreach ($definition AS $name => $structure) {
$is_new_table = False;
$sql3=""; $sql3="";
if (!isset($database[$name])) { if (!isset($database[$name])) {
$r = db_create_table($name, $structure["fields"], $verbose, $action); $r = db_create_table($name, $structure["fields"], $verbose, $action, $structure['indexes']);
if(false === $r) { if(false === $r) {
$errors .= t('Errors encountered creating database tables.').$name.EOL; $errors .= t('Errors encountered creating database tables.').$name.EOL;
} }
$is_new_table = True;
} else { } else {
// Drop the index if it isn't present in the definition and index name doesn't start with "local_" // Drop the index if it isn't present in the definition
foreach ($database[$name]["indexes"] AS $indexname => $fieldnames) // or the definition differ from current status
if (!isset($structure["indexes"][$indexname]) && substr($indexname, 0, 6) != 'local_') { // and index name doesn't start with "local_"
foreach ($database[$name]["indexes"] AS $indexname => $fieldnames) {
$current_index_definition = implode(",",$fieldnames);
if (isset($structure["indexes"][$indexname])) {
$new_index_definition = implode(",",$structure["indexes"][$indexname]);
} else {
$new_index_definition = "__NOT_SET__";
}
if ($current_index_definition != $new_index_definition && substr($indexname, 0, 6) != 'local_') {
$sql2=db_drop_index($indexname); $sql2=db_drop_index($indexname);
if ($sql3 == "") if ($sql3 == "")
$sql3 = "ALTER TABLE `".$name."` ".$sql2; $sql3 = "ALTER TABLE `".$name."` ".$sql2;
else else
$sql3 .= ", ".$sql2; $sql3 .= ", ".$sql2;
} }
}
// Compare the field structure field by field // Compare the field structure field by field
foreach ($structure["fields"] AS $fieldname => $parameters) { foreach ($structure["fields"] AS $fieldname => $parameters) {
if (!isset($database[$name]["fields"][$fieldname])) { if (!isset($database[$name]["fields"][$fieldname])) {
@ -186,19 +197,28 @@ function update_structure($verbose, $action, $tables=null, $definition=null) {
} }
} }
// Create the index // Create the index if the index don't exists in database
foreach ($structure["indexes"] AS $indexname => $fieldnames) { // or the definition differ from the current status.
if (!isset($database[$name]["indexes"][$indexname])) { // Don't create keys if table is new
$sql2=db_create_index($indexname, $fieldnames); if (!$is_new_table) {
if ($sql2 != "") { foreach ($structure["indexes"] AS $indexname => $fieldnames) {
if ($sql3 == "") if (isset($database[$name]["indexes"][$indexname])) {
$sql3 = "ALTER TABLE `".$name."` ".$sql2; $current_index_definition = implode(",",$database[$name]["indexes"][$indexname]);
else } else {
$sql3 .= ", ".$sql2; $current_index_definition = "__NOT_SET__";
}
$new_index_definition = implode(",",$fieldnames);
if ($current_index_definition != $new_index_definition) {
$sql2=db_create_index($indexname, $fieldnames);
if ($sql2 != "") {
if ($sql3 == "")
$sql3 = "ALTER TABLE `".$name."` ".$sql2;
else
$sql3 .= ", ".$sql2;
}
} }
} }
} }
if ($sql3 != "") { if ($sql3 != "") {
$sql3 .= ";"; $sql3 .= ";";
@ -244,7 +264,7 @@ function db_create_table($name, $fields, $verbose, $action, $indexes=null) {
$r = true; $r = true;
$sql = ""; $sql = "";
$sql_rows = array(); $sql_rows = array();
$primary_keys = array(); $primary_keys = array();
foreach($fields AS $fieldname => $field) { foreach($fields AS $fieldname => $field) {
@ -255,17 +275,12 @@ function db_create_table($name, $fields, $verbose, $action, $indexes=null) {
} }
if (!is_null($indexes)) { if (!is_null($indexes)) {
foreach ($indexes AS $indexname => $fieldnames) { foreach ($indexes AS $indexname => $fieldnames) {
$sql_index = db_create_index($indexname, $fieldnames, ""); $sql_index = db_create_index($indexname, $fieldnames, "");
if (!is_null($sql_index)) $sql_rows[] = $sql_index; if (!is_null($sql_index)) $sql_rows[] = $sql_index;
} }
} }
if (count($primary_keys)>0) {
$sql_rows[] = sprintf("PRIMARY KEY(`%s`)", implode("`,`", $primary_keys));
}
$sql = implode(",\n\t", $sql_rows); $sql = implode(",\n\t", $sql_rows);
$sql = sprintf("CREATE TABLE IF NOT EXISTS `%s` (\n\t", dbesc($name)).$sql."\n) DEFAULT CHARSET=utf8"; $sql = sprintf("CREATE TABLE IF NOT EXISTS `%s` (\n\t", dbesc($name)).$sql."\n) DEFAULT CHARSET=utf8";
@ -295,8 +310,16 @@ function db_drop_index($indexname) {
function db_create_index($indexname, $fieldnames, $method="ADD") { function db_create_index($indexname, $fieldnames, $method="ADD") {
if ($indexname == "PRIMARY") $method = strtoupper(trim($method));
return; if ($method!="" && $method!="ADD") {
throw new Exception("Invalid parameter 'method' in db_create_index(): '$method'");
killme();
}
if ($indexname == "PRIMARY") {
return sprintf("%s PRIMARY KEY(`%s`)", $method, implode("`,`", $fieldnames));
}
$names = ""; $names = "";
foreach ($fieldnames AS $fieldname) { foreach ($fieldnames AS $fieldname) {
@ -309,11 +332,6 @@ function db_create_index($indexname, $fieldnames, $method="ADD") {
$names .= "`".dbesc($fieldname)."`"; $names .= "`".dbesc($fieldname)."`";
} }
$method = strtoupper(trim($method));
if ($method!="" && $method!="ADD") {
throw new Exception("Invalid parameter 'method' in db_create_index(): '$method'");
killme();
}
$sql = sprintf("%s INDEX `%s` (%s)", $method, dbesc($indexname), $names); $sql = sprintf("%s INDEX `%s` (%s)", $method, dbesc($indexname), $names);
return($sql); return($sql);