SetVariable converted to QueryDatabase

This commit is contained in:
Arthur Ice 2014-07-03 18:21:21 -07:00 committed by Arthur Ice
parent 74d65f6fda
commit 6d6746f5a4

View File

@ -986,9 +986,7 @@ bool Database::GetVariable(const char* varname, char* varvalue, uint16 varvalue_
} }
bool Database::SetVariable(const char* varname_in, const char* varvalue_in) { bool Database::SetVariable(const char* varname_in, const char* varvalue_in) {
char errbuf[MYSQL_ERRMSG_SIZE]; char *query = nullptr;
char *query = 0;
uint32 affected_rows = 0;
char *varname,*varvalue; char *varname,*varvalue;
@ -997,32 +995,37 @@ bool Database::SetVariable(const char* varname_in, const char* varvalue_in) {
DoEscapeString(varname, varname_in, strlen(varname_in)); DoEscapeString(varname, varname_in, strlen(varname_in));
DoEscapeString(varvalue, varvalue_in, strlen(varvalue_in)); DoEscapeString(varvalue, varvalue_in, strlen(varvalue_in));
if (RunQuery(query, MakeAnyLenString(&query, "Update variables set value='%s' WHERE varname like '%s'", varvalue, varname), errbuf, 0, &affected_rows)) { auto results = QueryDatabase(query, MakeAnyLenString(&query, "Update variables set value='%s' WHERE varname like '%s'", varvalue, varname));
if (!results.Success())
{
std::cerr << "Error in SetVariable query '" << query << "' " << results.ErrorMessage() << std::endl;
safe_delete_array(query); safe_delete_array(query);
if (affected_rows == 1) { free(varname);
LoadVariables(); // refresh cache free(varvalue);
free(varname); return false;
free(varvalue);
return true;
}
else {
if (RunQuery(query, MakeAnyLenString(&query, "Insert Into variables (varname, value) values ('%s', '%s')", varname, varvalue), errbuf, 0, &affected_rows)) {
safe_delete_array(query);
if (affected_rows == 1) {
LoadVariables(); // refresh cache
free(varname);
free(varvalue);
return true;
}
}
}
} }
else {
std::cerr << "Error in SetVariable query '" << query << "' " << errbuf << std::endl; safe_delete_array(query);
safe_delete_array(query);
if (results.RowsAffected() == 1)
{
LoadVariables(); // refresh cache
free(varname);
free(varvalue);
return true;
} }
results = QueryDatabase(query, MakeAnyLenString(&query, "Insert Into variables (varname, value) values ('%s', '%s')", varname, varvalue));
safe_delete_array(query);
free(varname); free(varname);
free(varvalue); free(varvalue);
if (results.RowsAffected() == 1) {
LoadVariables(); // refresh cache
return true;
}
return false; return false;
} }