ClickHouse/libs/libmysqlxx/src/Connection.cpp

#include <mysqlxx/Connection.h>
#include <mysqlxx/Exception.h>


namespace mysqlxx
{

Connection::Connection()
{
	is_connected = false;
}

Connection::Connection(
	const char* db,
	const char* server,
	const char* user,
	const char* password,
	unsigned int port)
{
	is_connected = false;
	connect(db, server, user, password, port);
}

Connection::~Connection()
{
	disconnect();
}

void Connection::connect(const char* db,
	const char* server,
	const char* user,
	const char* password,
	unsigned int port)
{
	if (is_connected)
		disconnect();

	/// Инициализация библиотеки.
	LibrarySingleton::instance();

	if (!mysql_init(&driver))
		throw ConnectionFailed(mysql_error(&driver), mysql_errno(&driver));

	/// Установим таймауты
	unsigned int timeout = MYSQLXX_TIMEOUT;
	if (mysql_options(&driver, MYSQL_OPT_CONNECT_TIMEOUT, reinterpret_cast<const char *>(&timeout)))
		throw ConnectionFailed(mysql_error(&driver), mysql_errno(&driver));

	/** Включаем возможность использовать запрос LOAD DATA LOCAL INFILE с серверами,
	  *  которые были скомпилированы без опции --enable-local-infile.
	  */
	if (mysql_options(&driver, MYSQL_OPT_LOCAL_INFILE, NULL))
		throw ConnectionFailed(mysql_error(&driver), mysql_errno(&driver));

	if (!mysql_real_connect(&driver, server, user, password, db, port, NULL, driver.client_flag))
		throw ConnectionFailed(mysql_error(&driver), mysql_errno(&driver));

	/// Установим кодировки по-умолчанию - UTF-8.
	if (mysql_set_character_set(&driver, "UTF8"))
		throw ConnectionFailed(mysql_error(&driver), mysql_errno(&driver));

	is_connected = true;
}

bool Connection::connected() const
{
	return is_connected;
}

void Connection::disconnect()
{
	if (!is_connected)
		return;

	mysql_close(&driver);
	memset(&driver, 0, sizeof(driver));
	is_connected = false;
}

bool Connection::ping()
{
	return is_connected && !mysql_ping(&driver);
}

Query Connection::query(const std::string & str)
{
	return Query(this, str);
}

MYSQL * Connection::getDriver()
{
	return &driver;
}

}