mirror of
https://github.com/EQEmu/Server.git
synced 2025-12-11 16:51:29 +00:00
95249889a6
This adds support for using prepared statements for MySQL queries. It is intended for use in a database quest API but it can be used in source with some caveats: - It uses exceptions for error handling instead of returning a fake result that needs checked. Usage must be wrapped in try/catch. - DBcore has a connection mutex which indicates the connection might be shared with other threads. This mutex is locked for certain stmt operations in an attempt to make it safe to use with multi threaded connections. - Prepared statements should only be used on the main thread since the internal logging is not synchronized. - Unlike the current query API which retrieves all results as strings, results are stored in buffers that represent the db field type. Getter functions are available to retrieve values as desired types.
108 lines
2.4 KiB
C++
108 lines
2.4 KiB
C++
#ifndef DBCORE_H
|
|
#define DBCORE_H
|
|
|
|
#ifdef _WINDOWS
|
|
#include <winsock2.h>
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
#include "../common/mutex.h"
|
|
#include "../common/mysql_request_result.h"
|
|
#include "../common/types.h"
|
|
|
|
#include <mysql.h>
|
|
#include <string.h>
|
|
#include <mutex>
|
|
|
|
#define CR_SERVER_GONE_ERROR 2006
|
|
#define CR_SERVER_LOST 2013
|
|
|
|
namespace mysql { class PreparedStmt; }
|
|
|
|
class DBcore {
|
|
public:
|
|
enum eStatus {
|
|
Closed, Connected, Error
|
|
};
|
|
|
|
DBcore();
|
|
~DBcore();
|
|
eStatus GetStatus() { return pStatus; }
|
|
MySQLRequestResult QueryDatabase(const char *query, uint32 querylen, bool retryOnFailureOnce = true);
|
|
MySQLRequestResult QueryDatabase(const std::string& query, bool retryOnFailureOnce = true);
|
|
MySQLRequestResult QueryDatabaseMulti(const std::string &query);
|
|
void TransactionBegin();
|
|
void TransactionCommit();
|
|
void TransactionRollback();
|
|
std::string Escape(const std::string& s);
|
|
uint32 DoEscapeString(char *tobuf, const char *frombuf, uint32 fromlen);
|
|
void ping();
|
|
|
|
const std::string &GetOriginHost() const;
|
|
void SetOriginHost(const std::string &origin_host);
|
|
|
|
bool DoesTableExist(const std::string& table_name);
|
|
|
|
void SetMySQL(const DBcore &o)
|
|
{
|
|
mysql = o.mysql;
|
|
mysqlOwner = false;
|
|
}
|
|
void SetMutex(Mutex *mutex);
|
|
|
|
// only safe on connections shared with other threads if results buffered
|
|
// unsafe to use off main thread due to internal server logging
|
|
// throws std::runtime_error on failure
|
|
mysql::PreparedStmt Prepare(std::string query);
|
|
|
|
protected:
|
|
bool Open(
|
|
const char *iHost,
|
|
const char *iUser,
|
|
const char *iPassword,
|
|
const char *iDatabase,
|
|
uint32 iPort,
|
|
uint32 *errnum = 0,
|
|
char *errbuf = 0,
|
|
bool iCompress = false,
|
|
bool iSSL = false
|
|
);
|
|
|
|
private:
|
|
bool Open(uint32 *errnum = nullptr, char *errbuf = nullptr);
|
|
|
|
MYSQL* mysql;
|
|
bool mysqlOwner;
|
|
Mutex *m_mutex;
|
|
eStatus pStatus;
|
|
|
|
std::mutex m_query_lock{};
|
|
|
|
std::string origin_host;
|
|
|
|
char *pHost;
|
|
char *pUser;
|
|
char *pPassword;
|
|
char *pDatabase;
|
|
bool pCompress;
|
|
uint32 pPort;
|
|
bool pSSL;
|
|
|
|
// allows multiple queries to be executed within the same query
|
|
// do not use this under normal operation
|
|
// we use this during database migrations only currently
|
|
void SetMultiStatementsOn()
|
|
{
|
|
mysql_set_server_option(mysql, MYSQL_OPTION_MULTI_STATEMENTS_ON);
|
|
}
|
|
|
|
// disables multiple statements to be executed in one query
|
|
void SetMultiStatementsOff()
|
|
{
|
|
mysql_set_server_option(mysql, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
|
|
}
|
|
};
|
|
|
|
|
|
#endif
|