2015-01-14 10:06:30 +00:00
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <DB/Client/Connection.h>
|
2015-01-15 18:33:20 +00:00
|
|
|
|
#include <DB/Client/ConnectionPool.h>
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
|
{
|
2015-01-15 16:55:50 +00:00
|
|
|
|
/**
|
|
|
|
|
* Множество реплик одного шарда.
|
|
|
|
|
*/
|
2015-01-15 18:03:59 +00:00
|
|
|
|
class ShardReplicas final
|
2015-01-15 16:55:50 +00:00
|
|
|
|
{
|
2015-01-14 10:06:30 +00:00
|
|
|
|
public:
|
2015-01-16 15:22:12 +00:00
|
|
|
|
ShardReplicas(std::vector<ConnectionPool::Entry> & entries, const Settings & settings_);
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
2015-01-15 18:03:59 +00:00
|
|
|
|
~ShardReplicas() = default;
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
2015-01-15 18:03:59 +00:00
|
|
|
|
ShardReplicas(const ShardReplicas &) = delete;
|
|
|
|
|
ShardReplicas & operator=(const ShardReplicas &) = delete;
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Отправить на реплики всё содержимое внешних таблиц.
|
|
|
|
|
void sendExternalTablesData(std::vector<ExternalTablesData> & data);
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Отправить запрос на реплики.
|
2015-01-16 15:22:12 +00:00
|
|
|
|
void sendQuery(const String & query, const String & query_id = "",
|
|
|
|
|
UInt64 stage = QueryProcessingStage::Complete, bool with_pending_data = false);
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Получить пакет от какой-нибудь реплики.
|
|
|
|
|
Connection::Packet receivePacket();
|
|
|
|
|
|
|
|
|
|
/// Разорвать соединения к репликам
|
2015-01-15 12:09:26 +00:00
|
|
|
|
void disconnect();
|
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Отменить запросы к репликам
|
2015-01-15 12:09:26 +00:00
|
|
|
|
void sendCancel();
|
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Для каждой реплики получить оставшиеся пакеты после отмена запроса.
|
|
|
|
|
Connection::Packet drain();
|
2015-01-15 12:09:26 +00:00
|
|
|
|
|
2015-01-22 21:54:16 +00:00
|
|
|
|
/// Получить адреса реплик в виде строки.
|
2015-01-15 12:09:26 +00:00
|
|
|
|
std::string dumpAddresses() const;
|
|
|
|
|
|
2015-01-15 16:55:50 +00:00
|
|
|
|
/// Возвращает количесто реплик.
|
2015-01-22 21:54:16 +00:00
|
|
|
|
size_t size() const { return replica_hash.size(); }
|
2015-01-15 15:05:03 +00:00
|
|
|
|
|
2015-01-14 10:06:30 +00:00
|
|
|
|
private:
|
2015-01-18 15:52:09 +00:00
|
|
|
|
/// Проверить, есть ли данные, которые можно прочитать на каких-нибудь репликах.
|
|
|
|
|
/// Возвращает соединение на реплику, с которой можно прочитать данные, если такая есть.
|
2015-01-22 21:54:16 +00:00
|
|
|
|
Connection ** waitForReadEvent();
|
2015-01-15 16:55:50 +00:00
|
|
|
|
|
|
|
|
|
private:
|
2015-01-18 15:52:09 +00:00
|
|
|
|
/// Реплики хэшированные по id сокета
|
|
|
|
|
using ReplicaHash = std::unordered_map<int, Connection *>;
|
2015-01-14 10:06:30 +00:00
|
|
|
|
|
|
|
|
|
private:
|
2015-01-16 15:22:12 +00:00
|
|
|
|
const Settings & settings;
|
2015-01-15 16:55:50 +00:00
|
|
|
|
ReplicaHash replica_hash;
|
2015-01-22 21:54:16 +00:00
|
|
|
|
size_t active_connection_count = 0;
|
|
|
|
|
bool sent_query = false;
|
|
|
|
|
bool cancelled = false;
|
2015-01-14 10:06:30 +00:00
|
|
|
|
};
|
|
|
|
|
}
|