tibia-client/src/framework/sql/mysql.cpp

271 lines
6.9 KiB
C++
Raw Normal View History

/*
* Copyright (c) 2010-2012 OTClient <https://github.com/edubart/otclient>
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#ifdef WIN32
#include <winsock2.h>
#endif
#include "mysql.h"
#include <mysql/errmsg.h>
#include <framework/core/clock.h>
#include <framework/core/logger.h>
DatabaseMySQL::DatabaseMySQL() : m_running(false)
{
if(!mysql_init(&m_mysqlHandle))
g_logger.fatal("Failed to initialize MySQL connection handle.");
my_bool reconnect = true;
mysql_options(&m_mysqlHandle, MYSQL_OPT_RECONNECT, &reconnect);
}
DatabaseMySQL::~DatabaseMySQL()
{
mysql_close(&m_mysqlHandle);
}
void DatabaseMySQL::connect(const std::string& host, const std::string& user, const std::string& pass,
const std::string& db, uint16_t port, const std::string& unix_socket)
{
if(!mysql_real_connect(&m_mysqlHandle,
host.c_str(),
user.c_str(),
pass.c_str(),
db.c_str(),
port,
unix_socket.empty() ? NULL : unix_socket.c_str(), 0)) {
g_logger.error(stdext::format("Failed to connect to database. MYSQL ERROR: %s", mysql_error(&m_mysqlHandle)));
}
}
bool DatabaseMySQL::handleError()
{
unsigned int error = mysql_errno(&m_mysqlHandle);
g_logger.error(stdext::format("MYSQL error code = %d, message: %s", error, mysql_error(&m_mysqlHandle)));
if(error == CR_SOCKET_CREATE_ERROR ||
error == CR_CONNECTION_ERROR ||
error == CR_CONN_HOST_ERROR ||
error == CR_IPSOCK_ERROR ||
error == CR_UNKNOWN_HOST ||
error == CR_SERVER_GONE_ERROR ||
error == CR_SERVER_LOST ||
error == CR_SERVER_HANDSHAKE_ERR) {
g_logger.error("MYSQL connection lost, trying to reconnect...");
//int64_t startTime = g_clock.millis();
/*while(true) {
bool connected = (mysql_ping(&m_mysqlHandle) == 0);
uint32_t diffTime = (mTime() - startTime);
if(connected) {
g_logger.info(stdext::format("MySQL reconneted in %d ms", diffTime));
return true;
}
mSleep(100);
}*/
}
return false;
}
bool DatabaseMySQL::beginTransaction()
{
return executeQuery("BEGIN");
}
bool DatabaseMySQL::rollback()
{
if(mysql_rollback(&m_mysqlHandle) != 0) {
g_logger.error(mysql_error(&m_mysqlHandle));
return false;
}
return true;
}
bool DatabaseMySQL::commit()
{
if(mysql_commit(&m_mysqlHandle) != 0) {
g_logger.error(mysql_error(&m_mysqlHandle));
return false;
}
return true;
}
bool DatabaseMySQL::internalExecuteQuery(const std::string &query)
{
while(mysql_real_query(&m_mysqlHandle, query.c_str(), query.length()) != 0) {
if(!handleError()) {
return false;
}
}
return true;
}
bool DatabaseMySQL::executeQuery(const std::string &query)
{
//LOG_ONDELAY(500);
if(internalExecuteQuery(query)) {
MYSQL_RES *m_res = mysql_store_result(&m_mysqlHandle);
if(m_res) {
mysql_free_result(m_res);
} else if(mysql_errno(&m_mysqlHandle) != 0) {
handleError();
}
return true;
}
return false;
}
2012-08-03 13:44:11 +02:00
DBResultPtr DatabaseMySQL::storeQuery(const std::string &query)
{
//LOG_ONDELAY(500);
while(internalExecuteQuery(query)) {
MYSQL_RES *m_res = mysql_store_result(&m_mysqlHandle);
if(m_res) {
2012-08-03 13:44:11 +02:00
DBResultPtr res = DBResultPtr(new DBResult(m_res));
if(res->next()) {
return res;
} else {
2012-08-03 13:44:11 +02:00
//delete res;
break;
}
} else if(mysql_errno(&m_mysqlHandle) != 0) {
if(!handleError()) {
break;
}
}
//mSleep(10);
}
return NULL;
}
uint64_t DatabaseMySQL::getLastInsertedRowID()
{
return (uint64_t)mysql_insert_id(&m_mysqlHandle);
}
std::string DatabaseMySQL::escapeString(const std::string &s)
{
return escapeBlob( s.c_str(), s.length() );
}
std::string DatabaseMySQL::escapeBlob(const char* s, uint32_t length)
{
if(!s)
return std::string("''");
char* output = new char[length * 2 + 1];
mysql_real_escape_string(&m_mysqlHandle, output, s, length);
std::string r = "'";
r += output;
r += "'";
delete[] output;
return r;
}
void DatabaseMySQL::freeResult(DBResult* res)
{
delete res;
}
DBResult::DBResult(MYSQL_RES* res)
{
m_res = res;
m_listNames.clear();
MYSQL_FIELD* field;
int32_t i = 0;
while((field = mysql_fetch_field(m_res))) {
m_listNames[field->name] = i;
i++;
}
}
DBResult::~DBResult()
{
mysql_free_result(m_res);
}
int32_t DBResult::getDataInt(const std::string &s)
{
ListNames::iterator it = m_listNames.find(s);
if(it != m_listNames.end() ) {
if(m_row[it->second] == NULL) {
return 0;
}
else {
return atoi(m_row[it->second]);
}
}
g_logger.error(stdext::format("error during getDataInt(%s).", s));
return 0;
}
int64_t DBResult::getDataLong(const std::string &s)
{
ListNames::iterator it = m_listNames.find(s);
if(it != m_listNames.end()) {
if(m_row[it->second] == NULL) {
return 0;
}
else {
return atoll(m_row[it->second]);
}
}
g_logger.error(stdext::format("error during getDataLong(%s).", s));
return 0;
}
std::string DBResult::getDataString(const std::string &s)
{
ListNames::iterator it = m_listNames.find(s);
if(it != m_listNames.end() ) {
if(m_row[it->second] == NULL)
return std::string("");
else
return std::string(m_row[it->second]);
}
g_logger.error(stdext::format("error during getDataString(%s).", s));
return std::string("");
}
bool DBResult::next()
{
m_row = mysql_fetch_row(m_res);
return m_row != NULL;
}