mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-27 10:02:01 +00:00
119 lines
3.4 KiB
C++
119 lines
3.4 KiB
C++
#pragma once
|
||
|
||
#include <Poco/Net/NetException.h>
|
||
|
||
#include <DB/Client/ConnectionPool.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Пул соединений с отказоустойчивостью.
|
||
* Инициализируется несколькими другими IConnectionPool-ами.
|
||
* При получении соединения, пытается создать или выбрать живое соединение из какого-нибудь пула,
|
||
* перебирая их в некотором порядке, используя не более указанного количества попыток.
|
||
* Предпочитаются пулы с меньшим количеством ошибок;
|
||
* пулы с одинаковым количеством ошибок пробуются в случайном порядке.
|
||
*
|
||
* Замечание: если один из вложенных пулов заблокируется из-за переполнения, то этот пул тоже заблокируется.
|
||
*/
|
||
class ConnectionPoolWithFailover : public IConnectionPool
|
||
{
|
||
public:
|
||
typedef detail::ConnectionPoolEntry Entry;
|
||
|
||
|
||
ConnectionPoolWithFailover(ConnectionPools & nested_pools_, size_t max_tries_ = DBMS_CONNECTION_POOL_WITH_FAILOVER_DEFAULT_MAX_TRIES)
|
||
: nested_pools(nested_pools_.begin(), nested_pools_.end()), max_tries(max_tries_),
|
||
log(&Logger::get("ConnectionPoolWithFailover"))
|
||
{
|
||
}
|
||
|
||
|
||
/** Выделяет соединение для работы. */
|
||
Entry get()
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
|
||
for (size_t i = 0, size = nested_pools.size(); i < size; ++i)
|
||
nested_pools[i].randomize();
|
||
|
||
std::sort(nested_pools.begin(), nested_pools.end());
|
||
|
||
std::stringstream fail_messages;
|
||
|
||
for (size_t try_no = 0; try_no < max_tries; ++try_no)
|
||
{
|
||
for (size_t i = 0, size = nested_pools.size(); i < size; ++i)
|
||
{
|
||
std::stringstream fail_message;
|
||
|
||
try
|
||
{
|
||
Entry res = nested_pools[i].pool->get();
|
||
res->forceConnected();
|
||
return res;
|
||
}
|
||
catch (const Exception & e)
|
||
{
|
||
if (e.code() != ErrorCodes::NETWORK_ERROR && e.code() != ErrorCodes::SOCKET_TIMEOUT)
|
||
throw;
|
||
|
||
fail_message << "Code: " << e.code() << ", e.displayText() = " << e.displayText() << ", e.what() = " << e.what();
|
||
}
|
||
|
||
LOG_WARNING(log, "Connection failed at try №"
|
||
<< (try_no + 1) << ", reason: " << fail_message.str());
|
||
|
||
fail_messages << fail_message.str() << std::endl;
|
||
|
||
++nested_pools[i].error_count;
|
||
}
|
||
}
|
||
|
||
throw Exception("All connection tries failed. Log: \n\n" + fail_messages.str() + "\n",
|
||
ErrorCodes::ALL_CONNECTION_TRIES_FAILED);
|
||
}
|
||
|
||
private:
|
||
struct PoolWithErrorCount
|
||
{
|
||
ConnectionPoolPtr pool;
|
||
UInt64 error_count;
|
||
UInt32 random;
|
||
drand48_data rand_state;
|
||
|
||
PoolWithErrorCount(const ConnectionPoolPtr & pool_)
|
||
: pool(pool_), error_count(0), random(0)
|
||
{
|
||
/// Инициализация плохая, но это не важно.
|
||
srand48_r(reinterpret_cast<ptrdiff_t>(this), &rand_state);
|
||
}
|
||
|
||
void randomize()
|
||
{
|
||
long int rand_res;
|
||
lrand48_r(&rand_state, &rand_res);
|
||
random = rand_res;
|
||
}
|
||
|
||
bool operator< (const PoolWithErrorCount & rhs) const
|
||
{
|
||
return error_count < rhs.error_count
|
||
|| (error_count == rhs.error_count && random < rhs.random);
|
||
}
|
||
};
|
||
|
||
Poco::FastMutex mutex;
|
||
|
||
typedef std::vector<PoolWithErrorCount> PoolsWithErrorCount;
|
||
PoolsWithErrorCount nested_pools;
|
||
|
||
size_t max_tries;
|
||
|
||
Logger * log;
|
||
};
|
||
|
||
|
||
}
|