mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-23 18:20:50 +00:00
117 lines
3.5 KiB
C++
117 lines
3.5 KiB
C++
#pragma once
|
||
|
||
#include <list>
|
||
#include <Poco/SharedPtr.h>
|
||
#include <Poco/Mutex.h>
|
||
#include <Poco/Condition.h>
|
||
#include <statdaemons/Stopwatch.h>
|
||
#include <DB/Core/Defines.h>
|
||
#include <DB/Core/Exception.h>
|
||
#include <DB/Core/ErrorCodes.h>
|
||
#include <DB/IO/WriteHelpers.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Список исполняющихся в данный момент запросов.
|
||
* Также реализует ограничение на их количество.
|
||
*/
|
||
|
||
class ProcessList
|
||
{
|
||
friend class Entry;
|
||
public:
|
||
/// Запрос и таймер его выполнения.
|
||
struct Element
|
||
{
|
||
std::string query;
|
||
Stopwatch watch;
|
||
|
||
Element(const std::string & query_) : query(query_) {}
|
||
};
|
||
|
||
/// list, чтобы итераторы не инвалидировались. NOTE: можно заменить на cyclic buffer, но почти незачем.
|
||
typedef std::list<Element> Containter;
|
||
|
||
private:
|
||
Containter cont;
|
||
size_t cur_size; /// В C++03 std::list::size не O(1).
|
||
size_t max_size; /// Если 0 - не ограничено. Иначе, если пытаемся добавить больше - кидается исключение.
|
||
mutable Poco::FastMutex mutex;
|
||
mutable Poco::Condition have_space; /// Количество одновременно выполняющихся запросов стало меньше максимального.
|
||
|
||
/// Держит итератор на список, и удаляет элемент из списка в деструкторе.
|
||
class Entry
|
||
{
|
||
private:
|
||
ProcessList & parent;
|
||
Containter::iterator it;
|
||
public:
|
||
Entry(ProcessList & parent_, Containter::iterator it_)
|
||
: parent(parent_), it(it_) {}
|
||
|
||
~Entry()
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(parent.mutex);
|
||
parent.cont.erase(it);
|
||
--parent.cur_size;
|
||
parent.have_space.signal();
|
||
}
|
||
};
|
||
|
||
public:
|
||
ProcessList(size_t max_size_ = 0) : cur_size(0), max_size(max_size_) {}
|
||
|
||
typedef Poco::SharedPtr<Entry> EntryPtr;
|
||
|
||
/** Зарегистрировать выполняющийся запрос. Возвращает refcounted объект, который удаляет запрос из списка при уничтожении.
|
||
* Если выполняющихся запросов сейчас слишком много - ждать не более указанного времени.
|
||
* Если времени не хватило - кинуть исключение.
|
||
*/
|
||
EntryPtr insert(const std::string & query_, size_t max_wait_milliseconds = DEFAULT_QUERIES_QUEUE_WAIT_TIME_MS)
|
||
{
|
||
EntryPtr res;
|
||
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
|
||
if (max_size && cur_size >= max_size && (!max_wait_milliseconds || !have_space.tryWait(mutex, max_wait_milliseconds)))
|
||
throw Exception("Too much simultaneous queries. Maximum: " + toString(max_size), ErrorCodes::TOO_MUCH_SIMULTANEOUS_QUERIES);
|
||
|
||
++cur_size;
|
||
res = new Entry(*this, cont.insert(cont.end(), Element(query_)));
|
||
}
|
||
|
||
return res;
|
||
}
|
||
|
||
/// Количество одновременно выполняющихся запросов.
|
||
size_t size() const
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
return cur_size;
|
||
}
|
||
|
||
/// Получить текущее состояние (копию) списка запросов.
|
||
Containter get() const
|
||
{
|
||
Containter res;
|
||
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
res = cont;
|
||
}
|
||
|
||
return res;
|
||
}
|
||
|
||
void setMaxSize(size_t max_size_)
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
max_size = max_size_;
|
||
}
|
||
};
|
||
|
||
}
|