ClickHouse/libs/libmysqlxx/include/mysqlxx/Pool.h

242 lines
6.8 KiB
C++
Raw Normal View History

2014-04-08 06:51:53 +00:00
#pragma once
2011-03-04 20:58:19 +00:00
#include <list>
2015-10-01 14:22:01 +00:00
#include <memory>
#include <mutex>
2011-03-04 20:58:19 +00:00
#include <Poco/Exception.h>
#include <mysqlxx/Connection.h>
2011-03-17 20:00:04 +00:00
#define MYSQLXX_POOL_DEFAULT_START_CONNECTIONS 1
#define MYSQLXX_POOL_DEFAULT_MAX_CONNECTIONS 16
#define MYSQLXX_POOL_SLEEP_ON_CONNECT_FAIL 10
2011-03-04 20:58:19 +00:00
namespace mysqlxx
{
/** Пул соединений с MySQL.
* Этот класс имеет мало отношения к mysqlxx и сделан не в стиле библиотеки. (взят из старого кода)
2011-03-18 20:26:54 +00:00
* Использование:
* mysqlxx::Pool pool("mysql_params");
*
* void thread()
* {
* mysqlxx::Pool::Entry connection = pool.Get();
* std::string s = connection->query("SELECT 'Hello, world!' AS world").use().fetch()["world"].getString();
* }
*
* TODO: Упростить, используя PoolBase.
2011-03-18 20:26:54 +00:00
*/
class Pool final
2011-03-04 20:58:19 +00:00
{
protected:
/** Информация о соединении. */
2011-03-04 20:58:19 +00:00
struct Connection
{
mysqlxx::Connection conn;
int ref_count = 0;
2011-03-04 20:58:19 +00:00
};
public:
/** Соединение с базой данных. */
2011-03-04 20:58:19 +00:00
class Entry
{
public:
2014-04-08 07:31:51 +00:00
Entry() {}
2011-03-04 20:58:19 +00:00
Entry(const Entry & src)
: data(src.data), pool(src.pool)
2011-03-04 20:58:19 +00:00
{
incrementRefCount();
2011-03-04 20:58:19 +00:00
}
~Entry()
2011-03-04 20:58:19 +00:00
{
decrementRefCount();
2011-03-04 20:58:19 +00:00
}
Entry & operator= (const Entry & src)
{
pool = src.pool;
if (data)
decrementRefCount();
data = src.data;
if (data)
incrementRefCount();
2011-03-04 20:58:19 +00:00
return * this;
}
bool isNull() const
{
2014-04-08 07:31:51 +00:00
return data == nullptr;
}
2011-03-04 20:58:19 +00:00
operator mysqlxx::Connection & ()
{
forceConnected();
return data->conn;
2011-03-04 20:58:19 +00:00
}
operator const mysqlxx::Connection & () const
{
forceConnected();
return data->conn;
2011-03-04 20:58:19 +00:00
}
const mysqlxx::Connection * operator->() const
{
forceConnected();
return &data->conn;
2011-03-04 20:58:19 +00:00
}
mysqlxx::Connection * operator->()
{
forceConnected();
return &data->conn;
2011-03-04 20:58:19 +00:00
}
Entry(Pool::Connection * conn, Pool * p)
: data(conn), pool(p)
2011-03-04 20:58:19 +00:00
{
incrementRefCount();
2011-03-04 20:58:19 +00:00
}
std::string getDescription() const
{
if (pool)
return pool->getDescription();
else
return "pool is null";
}
2011-03-04 20:58:19 +00:00
friend class Pool;
private:
/** Указатель на соединение. */
2014-04-08 07:31:51 +00:00
Connection * data = nullptr;
/** Указатель на пул, которому мы принадлежим. */
2014-04-08 07:31:51 +00:00
Pool * pool = nullptr;
2011-03-04 20:58:19 +00:00
/** Переподключается к базе данных в случае необходимости. Если не удалось - подождать и попробовать снова. */
void forceConnected() const;
/** Переподключается к базе данных в случае необходимости. Если не удалось - вернуть false. */
bool tryForceConnected() const
{
return data->conn.ping();
}
void incrementRefCount();
void decrementRefCount();
2011-03-04 20:58:19 +00:00
};
Pool(const std::string & config_name,
unsigned default_connections_ = MYSQLXX_POOL_DEFAULT_START_CONNECTIONS,
unsigned max_connections_ = MYSQLXX_POOL_DEFAULT_MAX_CONNECTIONS,
const char * parent_config_name_ = nullptr)
: Pool{Poco::Util::Application::instance().config(), config_name,
default_connections_, max_connections_, parent_config_name_}
{}
2011-03-04 20:58:19 +00:00
/**
* @param config_name Имя параметра в конфигурационном файле
* @param default_connections_ Количество подключений по-умолчанию
* @param max_connections_ Максимальное количество подключений
2011-03-04 20:58:19 +00:00
*/
Pool(const Poco::Util::AbstractConfiguration & cfg, const std::string & config_name,
unsigned default_connections_ = MYSQLXX_POOL_DEFAULT_START_CONNECTIONS,
unsigned max_connections_ = MYSQLXX_POOL_DEFAULT_MAX_CONNECTIONS,
const char * parent_config_name_ = nullptr);
2015-10-01 14:22:01 +00:00
/**
* @param db_ Имя БД
* @param server_ Хост для подключения
* @param user_ Имя пользователя
* @param password_ Пароль
* @param port_ Порт для подключения
* @param default_connections_ Количество подключений по-умолчанию
* @param max_connections_ Максимальное количество подключений
*/
Pool(const std::string & db_,
const std::string & server_,
const std::string & user_ = "",
const std::string & password_ = "",
unsigned port_ = 0,
unsigned connect_timeout_ = MYSQLXX_DEFAULT_TIMEOUT,
unsigned rw_timeout_ = MYSQLXX_DEFAULT_RW_TIMEOUT,
unsigned default_connections_ = MYSQLXX_POOL_DEFAULT_START_CONNECTIONS,
unsigned max_connections_ = MYSQLXX_POOL_DEFAULT_MAX_CONNECTIONS)
: default_connections(default_connections_), max_connections(max_connections_),
db(db_), server(server_), user(user_), password(password_), port(port_),
connect_timeout(connect_timeout_), rw_timeout(rw_timeout_) {}
Pool(const Pool & other)
: default_connections{other.default_connections},
max_connections{other.max_connections},
db{other.db}, server{other.server},
user{other.user}, password{other.password},
port{other.port}, connect_timeout{other.connect_timeout},
rw_timeout{other.rw_timeout}
{}
Pool & operator=(const Pool &) = delete;
2011-03-04 20:58:19 +00:00
~Pool();
2011-03-04 20:58:19 +00:00
/** Выделяет соединение для работы. */
Entry Get();
2011-03-04 20:58:19 +00:00
/** Выделяет соединение для работы.
* Если база недоступна - возвращает пустой объект Entry.
* Если пул переполнен - кидает исключение.
*/
Entry tryGet();
/// Получить описание БД
std::string getDescription() const
{
return description;
2011-03-04 20:58:19 +00:00
}
protected:
/** Количество соединений с MySQL, создаваемых при запуске. */
unsigned default_connections;
/** Максимально возможное количество соедиений. */
unsigned max_connections;
2011-03-04 20:58:19 +00:00
private:
/** Признак того, что мы инициализированы. */
bool initialized{false};
/** Список соединений. */
using Connections = std::list<Connection *>;
/** Список соединений. */
Connections connections;
/** Замок для доступа к списку соединений. */
std::mutex mutex;
/** Описание соединения. */
std::string description;
2015-10-01 14:22:01 +00:00
/** Параметры подключения. **/
std::string db;
std::string server;
std::string user;
std::string password;
unsigned port;
unsigned connect_timeout;
unsigned rw_timeout;
2011-03-04 20:58:19 +00:00
/** Хотя бы один раз было успешное соединение. */
bool was_successful{false};
/** Выполняет инициализацию класса, если мы еще не инициализированы. */
void initialize();
2011-03-04 20:58:19 +00:00
/** Create new connection. */
Connection * allocConnection(bool dont_throw_if_failed_first_time = false);
2011-03-04 20:58:19 +00:00
};
}