mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-11-22 21:47:24 +00:00
76 lines
2.3 KiB
C++
76 lines
2.3 KiB
C++
|
#include "Database.h"
|
||
|
|
#include "Game.h"
|
||
|
|
#include "dConfig.h"
|
||
|
|
#include "dLogger.h"
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
#pragma warning (disable:4251) //Disables SQL warnings
|
||
|
|
|
||
|
|
sql::Driver * Database::driver;
|
||
|
|
sql::Connection * Database::con;
|
||
|
|
|
||
|
|
void Database::Connect(const string& host, const string& database, const string& username, const string& password) {
|
||
|
|
driver = get_driver_instance();
|
||
|
|
|
||
|
|
//To bypass debug issues:
|
||
|
|
std::string newHost = "tcp://" + host;
|
||
|
|
const char* szHost = newHost.c_str();
|
||
|
|
const char* szDatabase = database.c_str();
|
||
|
|
const char* szUsername = username.c_str();
|
||
|
|
const char* szPassword = password.c_str();
|
||
|
|
|
||
|
|
con = driver->connect(szHost, szUsername, szPassword);
|
||
|
|
con->setSchema(szDatabase);
|
||
|
|
|
||
|
|
bool myTrue = true;
|
||
|
|
con->setClientOption("MYSQL_OPT_RECONNECT", &myTrue);
|
||
|
|
} //Connect
|
||
|
|
|
||
|
|
void Database::Destroy() {
|
||
|
|
if (!con) return;
|
||
|
|
cout << "Destroying MySQL connection!" << endl;
|
||
|
|
con->close();
|
||
|
|
delete con;
|
||
|
|
} //Destroy
|
||
|
|
|
||
|
|
sql::Statement* Database::CreateStmt() {
|
||
|
|
sql::Statement* toReturn = con->createStatement();
|
||
|
|
return toReturn;
|
||
|
|
} //CreateStmt
|
||
|
|
|
||
|
|
sql::PreparedStatement* Database::CreatePreppedStmt(const std::string& query) {
|
||
|
|
const char* test = query.c_str();
|
||
|
|
size_t size = query.length();
|
||
|
|
sql::SQLString str(test, size);
|
||
|
|
|
||
|
|
if (!con) {
|
||
|
|
//Connect to the MySQL Database
|
||
|
|
std::string mysql_host = Game::config->GetValue("mysql_host");
|
||
|
|
std::string mysql_database = Game::config->GetValue("mysql_database");
|
||
|
|
std::string mysql_username = Game::config->GetValue("mysql_username");
|
||
|
|
std::string mysql_password = Game::config->GetValue("mysql_password");
|
||
|
|
|
||
|
|
Connect(mysql_host, mysql_database, mysql_username, mysql_password);
|
||
|
|
Game::logger->Log("Database", "Trying to reconnect to MySQL\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!con->isValid() || con->isClosed())
|
||
|
|
{
|
||
|
|
delete con;
|
||
|
|
|
||
|
|
con = nullptr;
|
||
|
|
|
||
|
|
//Connect to the MySQL Database
|
||
|
|
std::string mysql_host = Game::config->GetValue("mysql_host");
|
||
|
|
std::string mysql_database = Game::config->GetValue("mysql_database");
|
||
|
|
std::string mysql_username = Game::config->GetValue("mysql_username");
|
||
|
|
std::string mysql_password = Game::config->GetValue("mysql_password");
|
||
|
|
|
||
|
|
Connect(mysql_host, mysql_database, mysql_username, mysql_password);
|
||
|
|
Game::logger->Log("Database", "Trying to reconnect to MySQL from invalid or closed connection\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
auto* stmt = con->prepareStatement(str);
|
||
|
|
|
||
|
|
return stmt;
|
||
|
|
} //CreatePreppedStmt
|