gesche
/
tibia-client
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Some minor clean ups

This commit is contained in:
BeniS 2013-03-05 17:31:19 +13:00
parent f4263384bc
commit 15ee7962f1
4 changed files with 49 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/framework/sql/database.h
View File

@ -262,7 +262,8 @@ class DBInsert
DatabasePtr m_db;
uint32 m_rows;
std::string m_query, m_buf;
std::string m_query;
std::string m_buf;
};
class DBTransaction

7
src/framework/sql/declarations.h
View File

@ -28,13 +28,14 @@
class Database;
class DBResult;
typedef stdext::shared_object_ptr<Database> DatabasePtr;
typedef stdext::shared_object_ptr<DBResult> DBResultPtr;
// MySQL
class DatabaseMySQL;
class MySQLResult;
typedef stdext::shared_object_ptr<Database> DatabasePtr;
typedef stdext::shared_object_ptr<DatabaseMySQL> DatabaseMySQLPtr;
typedef stdext::shared_object_ptr<DBResult> DBResultPtr;
typedef stdext::shared_object_ptr<MySQLResult> MySQLResultPtr;
#endif

80
src/framework/sql/mysql.cpp
View File

@ -100,7 +100,7 @@ bool DatabaseMySQL::beginTransaction()
bool DatabaseMySQL::rollback()
{
if(mysql_rollback(m_handle)) {
g_logger.error(stdext::format("DatabaseMySQL::rollback() ERROR: %s (%s)", mysql_error(m_handle), mysql_errno(m_handle)));
g_logger.error(stdext::format("[DatabaseMySQL::rollback] ERROR: %s (%s)", mysql_error(m_handle), mysql_errno(m_handle)));
return false;
}
@ -110,7 +110,7 @@ bool DatabaseMySQL::rollback()
bool DatabaseMySQL::commit()
{
if(mysql_commit(m_handle)) {
g_logger.error(stdext::format("DatabaseMySQL::commit() ERROR: %s (%s)", mysql_error(m_handle), mysql_errno(m_handle)));
g_logger.error(stdext::format("[DatabaseMySQL::commit] ERROR: %s (%s)", mysql_error(m_handle), mysql_errno(m_handle)));
return false;
}
@ -130,13 +130,11 @@ bool DatabaseMySQL::internalExecuteQuery(const std::string &query)
bool DatabaseMySQL::executeQuery(const std::string &query)
{
//LOG_ONDELAY(500);
if(internalExecuteQuery(query)) {
MYSQL_RES* m_res = mysql_store_result(m_handle);
MYSQL_RES* res = mysql_store_result(m_handle);
if(m_res) {
mysql_free_result(m_res);
if(res) {
mysql_free_result(res);
} else if(mysql_errno(m_handle) != 0) {
handleError();
}
@ -149,24 +147,22 @@ bool DatabaseMySQL::executeQuery(const std::string &query)
DBResultPtr DatabaseMySQL::storeQuery(const std::string& query)
{
//LOG_ONDELAY(500);
while(internalExecuteQuery(query)) {
MYSQL_RES* m_res = mysql_store_result(m_handle);
MYSQL_RES* res = mysql_store_result(m_handle);
if(m_res) {
DBResultPtr res = (DBResultPtr)MySQLResultPtr(new MySQLResult(m_res));
if(!verifyResult(res))
if(res) {
DBResultPtr result = (DBResultPtr)MySQLResultPtr(new MySQLResult(res));
if(!verifyResult(result))
break;
return res;
return result;
}
else if(mysql_errno(m_handle) != 0) {
if(!handleError())
break;
}
//mSleep(10);
stdext::millisleep(10);
}
return nullptr;
@ -201,81 +197,81 @@ std::string DatabaseMySQL::escapeBlob(const char* s, uint32 length)
int32 MySQLResult::getDataInt(const std::string& s)
{
listNames_t::iterator it = m_listNames.find(s);
if(it != m_listNames.end())
RowNames_t::iterator it = m_names.find(s);
if(it != m_names.end())
return m_row[it->second] ? atoi(m_row[it->second]) : 0;
g_logger.error(stdext::format("MySQLResult::getDataInt() Error: %d", s));
return 0; // Failed
g_logger.error(stdext::format("[MySQLResult::getDataInt] Error: %d", s));
return 0;
}
int64 MySQLResult::getDataLong(const std::string& s)
{
listNames_t::iterator it = m_listNames.find(s);
if(it != m_listNames.end())
RowNames_t::iterator it = m_names.find(s);
if(it != m_names.end())
return m_row[it->second] ? atoll(m_row[it->second]) : 0;
g_logger.error(stdext::format("MySQLResult::getDataLong() Error: %d", s));
return 0; // Failed
g_logger.error(stdext::format("[MySQLResult::getDataLong] Error: %d", s));
return 0;
}
std::string MySQLResult::getDataString(const std::string& s)
{
listNames_t::iterator it = m_listNames.find(s);
if(it != m_listNames.end())
RowNames_t::iterator it = m_names.find(s);
if(it != m_names.end())
return m_row[it->second] ? std::string(m_row[it->second]) : std::string();
g_logger.error(stdext::format("MySQLResult::getDataString() Error: %d", s));
return std::string(); // Failed
g_logger.error(stdext::format("[MySQLResult::getDataString] Error: %d", s));
return std::string();
}
const char* MySQLResult::getDataStream(const std::string& s, uint64& size)
{
size = 0;
listNames_t::iterator it = m_listNames.find(s);
if(it == m_listNames.end()) {
g_logger.error(stdext::format("MySQLResult::getDataStream() Error: %d", s));
RowNames_t::iterator it = m_names.find(s);
if(it == m_names.end()) {
g_logger.error(stdext::format("[MySQLResult::getDataStream] Error: %d", s));
return NULL;
}
if(!m_row[it->second])
return NULL;
size = mysql_fetch_lengths(m_resultHandle)[it->second];
size = mysql_fetch_lengths(m_result)[it->second];
return m_row[it->second];
}
void MySQLResult::free()
{
if(!m_resultHandle) {
g_logger.fatal("MySQLResult::free() Error trying to free already freed result");
if(!m_result) {
g_logger.fatal("[MySQLResult::free] Error: trying to free already freed result");
return;
}
mysql_free_result(m_resultHandle);
m_resultHandle = NULL;
mysql_free_result(m_result);
m_result = NULL;
}
bool MySQLResult::next()
{
m_row = mysql_fetch_row(m_resultHandle);
m_row = mysql_fetch_row(m_result);
return (m_row != NULL);
}
MySQLResult::~MySQLResult()
{
if(m_resultHandle)
mysql_free_result(m_resultHandle);
if(m_result)
mysql_free_result(m_result);
}
MySQLResult::MySQLResult(MYSQL_RES* result)
{
m_resultHandle = result;
m_listNames.clear();
m_result = result;
m_names.clear();
MYSQL_FIELD* field;
int32 i = 0;
while((field = mysql_fetch_field(m_resultHandle))) {
m_listNames[field->name] = i++;
while((field = mysql_fetch_field(m_result))) {
m_names[field->name] = i++;
}
}

8
src/framework/sql/mysql.h
View File

@ -73,16 +73,16 @@ class MySQLResult : public DBResult
virtual void free();
virtual bool next();
virtual int getRowCount() { return mysql_num_rows(m_resultHandle); }
virtual int getRowCount() { return mysql_num_rows(m_result); }
protected:
MySQLResult(MYSQL_RES* result);
virtual ~MySQLResult();
typedef std::map<const std::string, uint32> listNames_t;
listNames_t m_listNames;
typedef std::map<const std::string, uint32> RowNames_t;
RowNames_t m_names;
MYSQL_RES* m_resultHandle;
MYSQL_RES* m_result;
MYSQL_ROW m_row;
};