DarkflameServer/dDatabase/Database.cpp

#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(std::string source) {
	if (!con) return;
	if (source != "") Game::logger->Log("Database", "Destroying MySQL connection from %s!\n", source.c_str());
	else Game::logger->Log("Database", "Destroying MySQL connection!\n");
	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