mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-16 04:32:33 +00:00
63b95c7451
* Added a new type of Exception for better recognition of connection failures * Added more logging for mysql communication * Retries on connection loss during query. Make MySQL dictionary source resilient to unexpected loss of connection during querying MySQL. This behavior is controlled with ".fail_on_connection_loss" config parameter, which defaults to false. * Stripped some debugging leftover garbage * Minor followup corrections * Corrections after PR comments * Yet more fixes
106 lines
2.2 KiB
C++
106 lines
2.2 KiB
C++
#if __has_include(<mysql.h>)
|
|
#include <errmsg.h>
|
|
#include <mysql.h>
|
|
#else
|
|
#include <mysql/errmsg.h>
|
|
#include <mysql/mysql.h>
|
|
#endif
|
|
|
|
#include <Poco/Logger.h>
|
|
|
|
#include <mysqlxx/Connection.h>
|
|
#include <mysqlxx/Query.h>
|
|
#include <mysqlxx/Types.h>
|
|
|
|
|
|
namespace mysqlxx
|
|
{
|
|
|
|
Query::Query(Connection * conn_, const std::string & query_string) : conn(conn_)
|
|
{
|
|
/// Важно в случае, если Query используется не из того же потока, что Connection.
|
|
mysql_thread_init();
|
|
|
|
if (!query_string.empty())
|
|
query_buf << query_string;
|
|
|
|
query_buf.imbue(std::locale::classic());
|
|
}
|
|
|
|
Query::Query(const Query & other) : conn(other.conn)
|
|
{
|
|
/// Важно в случае, если Query используется не из того же потока, что Connection.
|
|
mysql_thread_init();
|
|
|
|
query_buf.imbue(std::locale::classic());
|
|
|
|
*this << other.str();
|
|
}
|
|
|
|
Query & Query::operator= (const Query & other)
|
|
{
|
|
if (this == &other)
|
|
return *this;
|
|
|
|
conn = other.conn;
|
|
|
|
query_buf.str(other.str());
|
|
|
|
return *this;
|
|
}
|
|
|
|
Query::~Query()
|
|
{
|
|
mysql_thread_end();
|
|
}
|
|
|
|
void Query::reset()
|
|
{
|
|
query_buf.str({});
|
|
}
|
|
|
|
void Query::executeImpl()
|
|
{
|
|
std::string query_string = query_buf.str();
|
|
|
|
MYSQL* mysql_driver = conn->getDriver();
|
|
|
|
auto & logger = Poco::Logger::get("mysqlxx::Query");
|
|
logger.trace("Running MySQL query using connection %lu", mysql_thread_id(mysql_driver));
|
|
if (mysql_real_query(mysql_driver, query_string.data(), query_string.size()))
|
|
{
|
|
const auto err_no = mysql_errno(mysql_driver);
|
|
switch (err_no)
|
|
{
|
|
case CR_SERVER_GONE_ERROR:
|
|
[[fallthrough]];
|
|
case CR_SERVER_LOST:
|
|
throw ConnectionLost(errorMessage(mysql_driver), err_no);
|
|
default:
|
|
throw BadQuery(errorMessage(mysql_driver), err_no);
|
|
}
|
|
}
|
|
}
|
|
|
|
UseQueryResult Query::use()
|
|
{
|
|
executeImpl();
|
|
MYSQL_RES * res = mysql_use_result(conn->getDriver());
|
|
if (!res)
|
|
onError(conn->getDriver());
|
|
|
|
return UseQueryResult(res, conn, this);
|
|
}
|
|
|
|
void Query::execute()
|
|
{
|
|
executeImpl();
|
|
}
|
|
|
|
UInt64 Query::insertID()
|
|
{
|
|
return mysql_insert_id(conn->getDriver());
|
|
}
|
|
|
|
}
|