rlkollman/meekrodb
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add insertUpdate() function for "insert ... on duplicate key update" behavior

Browse Source
This commit is contained in:
Sergey Tsalkov
2011-09-23 19:15:41 -07:00
parent 9f3aa571dc
commit 5050205e58
2 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
db.class.php
View File

@@ -215,6 +215,11 @@ class DB
if (isset($options['ignore']) && $options['ignore'] && strtolower($which) == 'insert') { if (isset($options['ignore']) && $options['ignore'] && strtolower($which) == 'insert') {
DB::queryNull("INSERT IGNORE INTO $table ($keys_str) VALUES $values_str"); DB::queryNull("INSERT IGNORE INTO $table ($keys_str) VALUES $values_str");
} else if (isset($options['update']) && $options['update'] && strtolower($which) == 'insert') {
$updatestr = call_user_func_array('DB::parseQueryParams', $options['update']);
DB::queryNull("INSERT INTO $table ($keys_str) VALUES $values_str ON DUPLICATE KEY UPDATE $updatestr");
} else { } else {
DB::queryNull("$which INTO $table ($keys_str) VALUES $values_str"); DB::queryNull("$which INTO $table ($keys_str) VALUES $values_str");
} }
@@ -224,6 +229,13 @@ class DB
public static function insertIgnore($table, $data) { return DB::insertOrReplace('INSERT', $table, $data, array('ignore' => true)); } public static function insertIgnore($table, $data) { return DB::insertOrReplace('INSERT', $table, $data, array('ignore' => true)); }
public static function replace($table, $data) { return DB::insertOrReplace('REPLACE', $table, $data); } public static function replace($table, $data) { return DB::insertOrReplace('REPLACE', $table, $data); }
public static function insertUpdate() {
$args = func_get_args();
$table = array_shift($args);
$data = array_shift($args);
return DB::insertOrReplace('INSERT', $table, $data, array('update' => $args));
}
public static function delete() { public static function delete() {
$args = func_get_args(); $args = func_get_args();
$table = self::formatTableName(array_shift($args)); $table = self::formatTableName(array_shift($args));

19
simpletest/BasicTest.php
View File

@@ -232,6 +232,25 @@ class BasicTest extends SimpleTest {
'height' => 199.194 'height' => 199.194
)); ));
} }
function test_7_insert_update() {
DB::insertUpdate('accounts', array(
'id' => 2, //duplicate primary key
'username' => 'gonesoon',
'password' => 'something',
'age' => 61,
'height' => 199.194
), 'age = age + %i', 1);
$this->assert(DB::affectedRows() === 2); // a quirk of MySQL, even though only 1 row was updated
$result = DB::query("SELECT * FROM accounts WHERE age = %i", 16);
$this->assert(count($result) === 1);
$this->assert($result[0]['height'] === '10.371');
DB::query("UPDATE accounts SET age=age-1 WHERE age=%i", 16);
$this->assert(DB::affectedRows() === 1);
}
} }