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

make sure insertOrReplace() gets valid input, throw an error if it doesn't

Browse Source
This commit is contained in:
Sergey Tsalkov
2020-01-27 08:45:36 -08:00
parent 027681b350
commit 18d9ab2337
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
db.class.php
View File

@@ -335,10 +335,12 @@ class MeekroDB {
$keys = array_keys($datas); $keys = array_keys($datas);
$values = array_values($datas); $values = array_values($datas);
} }
if ($which != 'INSERT' && $which != 'INSERT IGNORE' && $which != 'REPLACE') {
return $this->nonSQLError('insertOrReplace() must be called with one of: INSERT, INSERT IGNORE, REPLACE');
}
if (isset($options['ignore']) && $options['ignore']) $which = 'INSERT IGNORE'; if (isset($options['update']) && is_array($options['update']) && $options['update'] && $which == 'INSERT') {
if (isset($options['update']) && is_array($options['update']) && $options['update'] && strtoupper($which) == 'INSERT') {
if (array_values($options['update']) !== $options['update']) { if (array_values($options['update']) !== $options['update']) {
return $this->query( return $this->query(
str_replace('%', $this->param_char, "INSERT INTO %b %lb VALUES $var ON DUPLICATE KEY UPDATE %hc"), str_replace('%', $this->param_char, "INSERT INTO %b %lb VALUES $var ON DUPLICATE KEY UPDATE %hc"),
@@ -360,7 +362,7 @@ class MeekroDB {
} }
public function insert($table, $data) { return $this->insertOrReplace('INSERT', $table, $data); } public function insert($table, $data) { return $this->insertOrReplace('INSERT', $table, $data); }
public function insertIgnore($table, $data) { return $this->insertOrReplace('INSERT', $table, $data, array('ignore' => true)); } public function insertIgnore($table, $data) { return $this->insertOrReplace('INSERT IGNORE', $table, $data); }
public function replace($table, $data) { return $this->insertOrReplace('REPLACE', $table, $data); } public function replace($table, $data) { return $this->insertOrReplace('REPLACE', $table, $data); }
public function insertUpdate() { public function insertUpdate() {