2013-09-05 11:44:10 +00:00
|
|
|
|
#include <mysqlxx/PoolWithFailover.h>
|
|
|
|
|
|
|
|
|
|
using namespace mysqlxx;
|
|
|
|
|
|
|
|
|
|
PoolWithFailover::PoolWithFailover(const std::string & config_name, unsigned default_connections,
|
|
|
|
|
unsigned max_connections, size_t max_tries_)
|
|
|
|
|
: max_tries(max_tries_)
|
|
|
|
|
{
|
|
|
|
|
Poco::Util::Application & app = Poco::Util::Application::instance();
|
|
|
|
|
Poco::Util::AbstractConfiguration & cfg = app.config();
|
|
|
|
|
|
|
|
|
|
if (cfg.has(config_name + ".replica"))
|
|
|
|
|
{
|
|
|
|
|
Poco::Util::AbstractConfiguration::Keys replica_keys;
|
|
|
|
|
cfg.keys(config_name, replica_keys);
|
|
|
|
|
for (Poco::Util::AbstractConfiguration::Keys::const_iterator it = replica_keys.begin(); it != replica_keys.end(); ++it)
|
|
|
|
|
{
|
|
|
|
|
if (!(*it == "port" || *it == "user" || *it == "password" || *it == "db"))
|
|
|
|
|
{
|
|
|
|
|
if (it->size() < std::string("replica").size() || it->substr(0, std::string("replica").size()) != "replica")
|
|
|
|
|
throw Poco::Exception("Unknown element in config: " + *it + ", expected replica");
|
|
|
|
|
std::string replica_name = config_name + "." + *it;
|
2013-10-29 09:27:10 +00:00
|
|
|
|
Replica replica(new Pool(replica_name, default_connections, max_connections, config_name.c_str()),
|
2013-09-05 11:44:10 +00:00
|
|
|
|
cfg.getInt(replica_name + ".priority", 0));
|
|
|
|
|
replicas_by_priority[replica.priority].push_back(replica);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
replicas_by_priority[0].push_back(Replica(new Pool(config_name, default_connections, max_connections), 0));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PoolWithFailover::Entry PoolWithFailover::Get()
|
|
|
|
|
{
|
|
|
|
|
Poco::ScopedLock<Poco::FastMutex> locker(mutex);
|
|
|
|
|
Poco::Util::Application & app = Poco::Util::Application::instance();
|
|
|
|
|
|
|
|
|
|
/// Если к какой-то реплике не подключились, потому что исчерпан лимит соединений, можно подождать и подключиться к ней.
|
|
|
|
|
Replica * full_pool = NULL;
|
|
|
|
|
|
|
|
|
|
for (size_t try_no = 0; try_no < max_tries; ++try_no)
|
|
|
|
|
{
|
|
|
|
|
full_pool = NULL;
|
|
|
|
|
|
|
|
|
|
for (ReplicasByPriority::iterator it = replicas_by_priority.begin(); it != replicas_by_priority.end(); ++it)
|
|
|
|
|
{
|
|
|
|
|
Replicas & replicas = it->second;
|
|
|
|
|
for (size_t i = 0; i < replicas.size(); ++i)
|
|
|
|
|
{
|
|
|
|
|
Replica & replica = replicas[i];
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Entry entry = replica.pool->tryGet();
|
|
|
|
|
|
|
|
|
|
if (!entry.isNull())
|
|
|
|
|
{
|
|
|
|
|
/// Переместим все пройденные реплики в конец очереди.
|
|
|
|
|
/// Пройденные реплики с другим приоритетом перемещать незачем.
|
|
|
|
|
std::rotate(replicas.begin(), replicas.begin() + i + 1, replicas.end());
|
|
|
|
|
|
|
|
|
|
return entry;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (const Poco::Exception & e)
|
|
|
|
|
{
|
|
|
|
|
if (e.displayText() == "mysqlxx::Pool is full")
|
|
|
|
|
{
|
|
|
|
|
full_pool = &replica;
|
|
|
|
|
}
|
|
|
|
|
|
2013-10-24 17:49:14 +00:00
|
|
|
|
app.logger().warning("Connection to " + replica.pool->getDescription() + " failed: " + e.displayText());
|
2013-09-05 11:44:10 +00:00
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2013-10-24 17:49:14 +00:00
|
|
|
|
app.logger().warning("Connection to " + replica.pool->getDescription() + " failed.");
|
2013-09-05 11:44:10 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
app.logger().error("Connection to all replicas failed " + Poco::NumberFormatter::format(try_no + 1) + " times");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (full_pool)
|
|
|
|
|
{
|
|
|
|
|
app.logger().error("All connections failed, trying to wait on a full pool " + full_pool->pool->getDescription());
|
|
|
|
|
return full_pool->pool->Get();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::stringstream message;
|
|
|
|
|
message << "Connections to all replicas failed: ";
|
|
|
|
|
for (ReplicasByPriority::const_iterator it = replicas_by_priority.begin(); it != replicas_by_priority.end(); ++it)
|
|
|
|
|
for (Replicas::const_iterator jt = it->second.begin(); jt != it->second.end(); ++jt)
|
|
|
|
|
message << (it == replicas_by_priority.begin() && jt == it->second.begin() ? "" : ", ") << jt->pool->getDescription();
|
|
|
|
|
|
|
|
|
|
throw Poco::Exception(message.str());
|
|
|
|
|
}
|