Converted LoadVariables, and LoadVariables_result to QueryDatabase

This commit is contained in:
Arthur Ice 2014-07-03 17:42:35 -07:00 committed by Arthur Ice
parent 87cb659dc2
commit 74d65f6fda
2 changed files with 64 additions and 61 deletions

View File

@ -843,7 +843,7 @@ void Database::GetAccountName(uint32 accountid, char* name, uint32* oLSAccountID
if (!results.Success()) if (!results.Success())
{ {
std::cerr << "Error in GetAccountName query '" << query << "' " << errbuf << std::endl; std::cerr << "Error in GetAccountName query '" << query << "' " << results.ErrorMessage() << std::endl;
safe_delete_array(query); safe_delete_array(query);
return; return;
} }
@ -881,82 +881,85 @@ void Database::GetCharName(uint32 char_id, char* name) {
} }
bool Database::LoadVariables() { bool Database::LoadVariables() {
char errbuf[MYSQL_ERRMSG_SIZE]; char *query = nullptr;
char *query = 0;
MYSQL_RES *result;
if (RunQuery(query, LoadVariables_MQ(&query), errbuf, &result)) { auto results = QueryDatabase(query, LoadVariables_MQ(&query));
if (!results.Success())
{
std::cerr << "Error in LoadVariables query '" << query << "' " << results.ErrorMessage() << std::endl;
safe_delete_array(query); safe_delete_array(query);
bool ret = LoadVariables_result(result); return false;
mysql_free_result(result);
return ret;
} }
else {
std::cerr << "Error in LoadVariables query '" << query << "' " << errbuf << std::endl; safe_delete_array(query);
safe_delete_array(query); return LoadVariables_result(results);
}
return false;
} }
uint32 Database::LoadVariables_MQ(char** query) { uint32 Database::LoadVariables_MQ(char** query)
// the read of this single variable should be atomic... this was causing strange problems {
// LockMutex lock(&Mvarcache);
return MakeAnyLenString(query, "SELECT varname, value, unix_timestamp() FROM variables where unix_timestamp(ts) >= %d", varcache_lastupdate); return MakeAnyLenString(query, "SELECT varname, value, unix_timestamp() FROM variables where unix_timestamp(ts) >= %d", varcache_lastupdate);
} }
bool Database::LoadVariables_result(MYSQL_RES* result) { // always returns true? not sure about this.
uint32 i; bool Database::LoadVariables_result(MySQLRequestResult results) {
MYSQL_ROW row; uint32 i = 0;
LockMutex lock(&Mvarcache); LockMutex lock(&Mvarcache);
if (mysql_num_rows(result) > 0) {
if (!varcache_array) { if (results.RowCount() == 0)
varcache_max = mysql_num_rows(result); return true;
varcache_array = new VarCache_Struct*[varcache_max];
for (i=0; i<varcache_max; i++) if (!varcache_array) {
varcache_array[i] = 0; varcache_max = results.RowCount();
} varcache_array = new VarCache_Struct*[varcache_max];
else { for (i=0; i<varcache_max; i++)
uint32 tmpnewmax = varcache_max + mysql_num_rows(result); varcache_array[i] = 0;
VarCache_Struct** tmp = new VarCache_Struct*[tmpnewmax]; }
for (i=0; i<tmpnewmax; i++) else {
tmp[i] = 0; uint32 tmpnewmax = varcache_max + results.RowCount();
for (i=0; i<varcache_max; i++) VarCache_Struct** tmp = new VarCache_Struct*[tmpnewmax];
tmp[i] = varcache_array[i]; for (i=0; i<tmpnewmax; i++)
VarCache_Struct** tmpdel = varcache_array; tmp[i] = 0;
varcache_array = tmp; for (i=0; i<varcache_max; i++)
varcache_max = tmpnewmax; tmp[i] = varcache_array[i];
delete [] tmpdel; VarCache_Struct** tmpdel = varcache_array;
} varcache_array = tmp;
while ((row = mysql_fetch_row(result))) { varcache_max = tmpnewmax;
varcache_lastupdate = atoi(row[2]); delete [] tmpdel;
for (i=0; i<varcache_max; i++) { }
if (varcache_array[i]) {
if (strcasecmp(varcache_array[i]->varname, row[0]) == 0) { for (auto row = results.begin(); row != results.end(); ++row)
delete varcache_array[i]; {
varcache_array[i] = (VarCache_Struct*) new uint8[sizeof(VarCache_Struct) + strlen(row[1]) + 1]; varcache_lastupdate = atoi(row[2]);
strn0cpy(varcache_array[i]->varname, row[0], sizeof(varcache_array[i]->varname)); for (i=0; i<varcache_max; i++) {
strcpy(varcache_array[i]->value, row[1]); if (varcache_array[i]) {
break; if (strcasecmp(varcache_array[i]->varname, row[0]) == 0) {
} delete varcache_array[i];
}
else {
varcache_array[i] = (VarCache_Struct*) new uint8[sizeof(VarCache_Struct) + strlen(row[1]) + 1]; varcache_array[i] = (VarCache_Struct*) new uint8[sizeof(VarCache_Struct) + strlen(row[1]) + 1];
strcpy(varcache_array[i]->varname, row[0]); strn0cpy(varcache_array[i]->varname, row[0], sizeof(varcache_array[i]->varname));
strcpy(varcache_array[i]->value, row[1]); strcpy(varcache_array[i]->value, row[1]);
break; break;
} }
} }
} else {
uint32 max_used = 0; varcache_array[i] = (VarCache_Struct*) new uint8[sizeof(VarCache_Struct) + strlen(row[1]) + 1];
for (i=0; i<varcache_max; i++) { strcpy(varcache_array[i]->varname, row[0]);
if (varcache_array[i]) { strcpy(varcache_array[i]->value, row[1]);
if (i > max_used) break;
max_used = i;
} }
} }
max_used++;
varcache_max = max_used;
} }
uint32 max_used = 0;
for (i=0; i<varcache_max; i++) {
if (varcache_array[i]) {
if (i > max_used)
max_used = i;
}
}
varcache_max = max_used + 1;
return true; return true;
} }

View File

@ -223,7 +223,7 @@ public:
bool SetVariable(const char* varname, const char* varvalue); bool SetVariable(const char* varname, const char* varvalue);
bool LoadVariables(); bool LoadVariables();
uint32 LoadVariables_MQ(char** query); uint32 LoadVariables_MQ(char** query);
bool LoadVariables_result(MYSQL_RES* result); bool LoadVariables_result(MySQLRequestResult results);
/* /*
* General Queries * General Queries