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

#pragma once

#include <memory>
#include <boost/noncopyable.hpp>

#include <Poco/Util/Application.h>

#include <common/singleton.h>

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

#define MYSQLXX_DEFAULT_TIMEOUT 60
#define MYSQLXX_DEFAULT_RW_TIMEOUT 1800


namespace mysqlxx
{


/** Для корректной инициализации и деинициализации MySQL библиотеки.
  * Обеспечивает единственный и thread-safe вызов mysql_library_init().
  * Использование:
  *     LibrarySingleton::instance();
  */
class LibrarySingleton : public Singleton<LibrarySingleton>
{
friend class Singleton<LibrarySingleton>;
private:
    LibrarySingleton();
    ~LibrarySingleton();
};


/** Соединение с MySQL.
  * Использование:
  *     mysqlxx::Connection connection("Test", "127.0.0.1", "root", "qwerty", 3306);
  *        std::cout << connection.query("SELECT 'Hello, World!'").store().at(0).at(0).getString() << std::endl;
  *
  * Или так, если вы используете конфигурацию из библиотеки Poco:
  *        mysqlxx::Connection connection("mysql_params");
  *
  * Внимание! Рекомендуется использовать соединение в том потоке, в котором оно создано.
  * Если вы используете соединение, созданное в другом потоке, то вы должны перед использованием
  *  вызвать функцию MySQL C API mysql_thread_init(), а после использования - mysql_thread_end().
  */
class Connection : private boost::noncopyable
{
public:
    /// Для отложенной инициализации.
    Connection();

    /// Создать соединение.
    Connection(
        const char* db,
        const char* server,
        const char* user = 0,
        const char* password = 0,
        unsigned port = 0,
        unsigned timeout = MYSQLXX_DEFAULT_TIMEOUT,
        unsigned rw_timeout = MYSQLXX_DEFAULT_RW_TIMEOUT);

    /** Конструктор-помошник. Создать соединение, считав все параметры из секции config_name конфигурации.
      * Можно использовать, если вы используете Poco::Util::Application из библиотеки Poco.
      */
    Connection(const std::string & config_name);

    virtual ~Connection();

    /// Для отложенной инициализации или для того, чтобы подключиться с другими параметрами.
    virtual void connect(const char * db,
        const char * server,
        const char * user,
        const char * password,
        unsigned port,
        unsigned timeout = MYSQLXX_DEFAULT_TIMEOUT,
        unsigned rw_timeout = MYSQLXX_DEFAULT_RW_TIMEOUT);

    void connect(const std::string & config_name)
    {
        Poco::Util::LayeredConfiguration & cfg = Poco::Util::Application::instance().config();

        std::string db             = cfg.getString(config_name + ".db", "");
        std::string server         = cfg.getString(config_name + ".host");
        std::string user         = cfg.getString(config_name + ".user");
        std::string password    = cfg.getString(config_name + ".password");
        unsigned port            = cfg.getInt(config_name + ".port");

        unsigned timeout =
            cfg.getInt(config_name + ".connect_timeout",
                cfg.getInt("mysql_connect_timeout",
                    MYSQLXX_DEFAULT_TIMEOUT));

        unsigned rw_timeout =
            cfg.getInt(config_name + ".rw_timeout",
                cfg.getInt("mysql_rw_timeout",
                    MYSQLXX_DEFAULT_RW_TIMEOUT));

        connect(db.c_str(), server.c_str(), user.c_str(), password.c_str(), port, timeout, rw_timeout);
    }

    /// Было ли произведено соединение с MySQL.
    bool connected() const;

    /// Отсоединиться от MySQL.
    void disconnect();

    /// Если соединение утеряно - попытаться восстановить его. true - если после вызова соединение есть.
    bool ping();

    /// Создать запрос. Вы можете сразу указать его, передав строку, или заполнить потом.
    Query query(const std::string & str = "");

    /// Получить объект MYSQL из C API.
    MYSQL * getDriver();

private:
    std::unique_ptr<MYSQL> driver;
    bool is_connected;
};


}