mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-26 09:32:01 +00:00
304 lines
14 KiB
C++
304 lines
14 KiB
C++
#pragma once
|
||
|
||
#include <map>
|
||
#include <set>
|
||
|
||
#include <Poco/SharedPtr.h>
|
||
#include <Poco/Mutex.h>
|
||
|
||
#include <Yandex/logger_useful.h>
|
||
|
||
#include <DB/Core/NamesAndTypes.h>
|
||
#include <DB/IO/UncompressedCache.h>
|
||
#include <DB/Storages/MarkCache.h>
|
||
#include <DB/DataStreams/FormatFactory.h>
|
||
#include <DB/Storages/IStorage.h>
|
||
#include <DB/Functions/FunctionFactory.h>
|
||
#include <DB/AggregateFunctions/AggregateFunctionFactory.h>
|
||
#include <DB/DataTypes/DataTypeFactory.h>
|
||
#include <DB/Storages/StorageFactory.h>
|
||
#include <DB/TableFunctions/TableFunctionFactory.h>
|
||
#include <DB/Interpreters/Settings.h>
|
||
#include <DB/Interpreters/Users.h>
|
||
#include <DB/Interpreters/Quota.h>
|
||
#include <DB/Interpreters/Dictionaries.h>
|
||
#include <DB/Interpreters/ProcessList.h>
|
||
#include <DB/Interpreters/Cluster.h>
|
||
#include <DB/Client/ConnectionPool.h>
|
||
#include <statdaemons/ConfigProcessor.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
class TableFunctionFactory;
|
||
|
||
using Poco::SharedPtr;
|
||
|
||
/// имя таблицы -> таблица
|
||
typedef std::map<String, StoragePtr> Tables;
|
||
|
||
/// имя БД -> таблицы
|
||
typedef std::map<String, Tables> Databases;
|
||
|
||
/// (имя базы данных, имя таблицы)
|
||
typedef std::pair<String, String> DatabaseAndTableName;
|
||
|
||
/// таблица -> множество таблиц-вьюшек, которые селектят из нее
|
||
typedef std::map<DatabaseAndTableName, std::set<DatabaseAndTableName> > ViewDependencies;
|
||
typedef std::vector<DatabaseAndTableName> Dependencies;
|
||
|
||
|
||
/** Набор известных объектов, которые могут быть использованы в запросе.
|
||
* Разделяемая часть. Порядок членов (порядок их уничтожения) очень важен.
|
||
*/
|
||
struct ContextShared
|
||
{
|
||
Logger * log; /// Логгер.
|
||
|
||
struct AfterDestroy
|
||
{
|
||
Logger * log;
|
||
|
||
AfterDestroy(Logger * log_) : log(log_) {}
|
||
~AfterDestroy()
|
||
{
|
||
#ifndef DBMS_CLIENT
|
||
LOG_INFO(log, "Uninitialized shared context.");
|
||
#endif
|
||
}
|
||
} after_destroy;
|
||
|
||
mutable Poco::Mutex mutex; /// Для доступа и модификации разделяемых объектов.
|
||
|
||
String path; /// Путь к директории с данными, со слешем на конце.
|
||
Databases databases; /// Список БД и таблиц в них.
|
||
TableFunctionFactory table_function_factory; /// Табличные функции.
|
||
FunctionFactory function_factory; /// Обычные функции.
|
||
AggregateFunctionFactory aggregate_function_factory; /// Агрегатные функции.
|
||
DataTypeFactory data_type_factory; /// Типы данных.
|
||
StorageFactory storage_factory; /// Движки таблиц.
|
||
FormatFactory format_factory; /// Форматы.
|
||
mutable SharedPtr<Dictionaries> dictionaries; /// Словари Метрики. Инициализируются лениво.
|
||
Users users; /// Известные пользователи.
|
||
Quotas quotas; /// Известные квоты на использование ресурсов.
|
||
mutable UncompressedCachePtr uncompressed_cache; /// Кэш разжатых блоков.
|
||
mutable MarkCachePtr mark_cache; /// Кэш засечек в сжатых файлах.
|
||
ProcessList process_list; /// Исполняющиеся в данный момент запросы.
|
||
ViewDependencies view_dependencies; /// Текущие зависимости
|
||
ConfigurationPtr users_config; /// Конфиг с секциями users, profiles и quotas.
|
||
|
||
/// Кластеры для distributed таблиц
|
||
/// Создаются при создании Distributed таблиц, так как нужно дождаться пока будут выставлены Settings
|
||
Poco::SharedPtr<Clusters> clusters;
|
||
|
||
bool shutdown_called = false;
|
||
|
||
|
||
ContextShared() : log(&Logger::get("Context")), after_destroy(log) {};
|
||
|
||
~ContextShared()
|
||
{
|
||
#ifndef DBMS_CLIENT
|
||
LOG_INFO(log, "Uninitializing shared context.");
|
||
#endif
|
||
|
||
try
|
||
{
|
||
shutdown();
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
}
|
||
}
|
||
|
||
|
||
/** Выполнить сложную работу по уничтожению объектов заранее.
|
||
*/
|
||
void shutdown()
|
||
{
|
||
if (shutdown_called)
|
||
return;
|
||
shutdown_called = true;
|
||
|
||
/** В этот момент, некоторые таблицы могут иметь потоки,
|
||
* которые модифицируют список таблиц, и блокируют наш mutex (см. StorageChunkMerger).
|
||
* Чтобы корректно их завершить, скопируем текущий список таблиц,
|
||
* и попросим их всех закончить свою работу.
|
||
* Потом удалим все объекты с таблицами.
|
||
*/
|
||
|
||
Databases current_databases;
|
||
|
||
{
|
||
Poco::ScopedLock<Poco::Mutex> lock(mutex);
|
||
current_databases = databases;
|
||
}
|
||
|
||
for (Databases::iterator it = current_databases.begin(); it != current_databases.end(); ++it)
|
||
for (Tables::iterator jt = it->second.begin(); jt != it->second.end(); ++jt)
|
||
jt->second->shutdown();
|
||
|
||
{
|
||
Poco::ScopedLock<Poco::Mutex> lock(mutex);
|
||
databases.clear();
|
||
}
|
||
}
|
||
};
|
||
|
||
|
||
/** Набор известных объектов, которые могут быть использованы в запросе.
|
||
* Состоит из разделяемой части (всегда общей для всех сессий и запросов)
|
||
* и копируемой части (которая может быть своей для каждой сессии или запроса).
|
||
*
|
||
* Всё инкапсулировано для всяких проверок и блокировок.
|
||
*/
|
||
class Context
|
||
{
|
||
private:
|
||
typedef SharedPtr<ContextShared> Shared;
|
||
Shared shared;
|
||
|
||
String user; /// Текущий пользователь.
|
||
Poco::Net::IPAddress ip_address; /// IP-адрес, с которого задан запрос.
|
||
QuotaForIntervalsPtr quota; /// Текущая квота. По-умолчанию - пустая квота, которая ничего не ограничивает.
|
||
String current_database; /// Текущая БД.
|
||
String current_query_id; /// Id текущего запроса.
|
||
NamesAndTypesList columns; /// Столбцы текущей обрабатываемой таблицы.
|
||
Settings settings; /// Настройки выполнения запроса.
|
||
ProgressCallback progress_callback; /// Колбек для отслеживания прогресса выполнения запроса.
|
||
ProcessList::Element * process_list_elem; /// Для отслеживания общего количества потраченных на запрос ресурсов.
|
||
|
||
String default_format; /// Формат, используемый, если сервер сам форматирует данные, и если в запросе не задан FORMAT.
|
||
/// То есть, используется в HTTP-интерфейсе. Может быть не задан - тогда используется некоторый глобальный формат по-умолчанию.
|
||
|
||
Context * session_context; /// Контекст сессии или NULL, если его нет. (Возможно, равен this.)
|
||
Context * global_context; /// Глобальный контекст или NULL, если его нет. (Возможно, равен this.)
|
||
|
||
public:
|
||
Context() : shared(new ContextShared), quota(new QuotaForIntervals), process_list_elem(NULL), session_context(NULL), global_context(NULL) {}
|
||
|
||
String getPath() const;
|
||
void setPath(const String & path);
|
||
|
||
/** Забрать список пользователей, квот и профилей настроек из этого конфига.
|
||
* Список пользователей полностью заменяется.
|
||
* Накопленные значения у квоты не сбрасываются, если квота не удалена.
|
||
*/
|
||
void setUsersConfig(ConfigurationPtr config);
|
||
|
||
ConfigurationPtr getUsersConfig();
|
||
|
||
void setUser(const String & name, const String & password, const Poco::Net::IPAddress & address, const String & quota_key);
|
||
String getUser() const { return user; }
|
||
Poco::Net::IPAddress getIPAddress() const { return ip_address; }
|
||
|
||
void setQuota(const String & name, const String & quota_key, const String & user_name, const Poco::Net::IPAddress & address);
|
||
QuotaForIntervals & getQuota();
|
||
|
||
void addDependency(const DatabaseAndTableName & from, const DatabaseAndTableName & where);
|
||
void removeDependency(const DatabaseAndTableName & from, const DatabaseAndTableName & where);
|
||
Dependencies getDependencies(const DatabaseAndTableName & from) const;
|
||
|
||
/// Проверка существования таблицы/БД. database может быть пустой - в этом случае используется текущая БД.
|
||
bool isTableExist(const String & database_name, const String & table_name) const;
|
||
bool isDatabaseExist(const String & database_name) const;
|
||
void assertTableExists(const String & database_name, const String & table_name) const;
|
||
void assertTableDoesntExist(const String & database_name, const String & table_name) const;
|
||
void assertDatabaseExists(const String & database_name) const;
|
||
void assertDatabaseDoesntExist(const String & database_name) const;
|
||
|
||
StoragePtr getTable(const String & database_name, const String & table_name) const;
|
||
StoragePtr tryGetTable(const String & database_name, const String & table_name) const;
|
||
void addTable(const String & database_name, const String & table_name, StoragePtr table);
|
||
void addDatabase(const String & database_name);
|
||
|
||
/// Возвращает отцепленную таблицу.
|
||
StoragePtr detachTable(const String & database_name, const String & table_name);
|
||
|
||
void detachDatabase(const String & database_name);
|
||
|
||
String getCurrentDatabase() const;
|
||
String getCurrentQueryId() const;
|
||
void setCurrentDatabase(const String & name);
|
||
void setCurrentQueryId(const String & query_id);
|
||
|
||
String getDefaultFormat() const; /// Если default_format не задан - возвращается некоторый глобальный формат по-умолчанию.
|
||
void setDefaultFormat(const String & name);
|
||
|
||
Settings getSettings() const;
|
||
void setSettings(const Settings & settings_);
|
||
|
||
Limits getLimits() const;
|
||
|
||
/// Установить настройку по имени.
|
||
void setSetting(const String & name, const Field & value);
|
||
|
||
/// Установить настройку по имени. Прочитать значение в текстовом виде из строки (например, из конфига, или из параметра URL).
|
||
void setSetting(const String & name, const std::string & value);
|
||
|
||
const TableFunctionFactory & getTableFunctionFactory() const { return shared->table_function_factory; }
|
||
const FunctionFactory & getFunctionFactory() const { return shared->function_factory; }
|
||
const AggregateFunctionFactory & getAggregateFunctionFactory() const { return shared->aggregate_function_factory; }
|
||
const DataTypeFactory & getDataTypeFactory() const { return shared->data_type_factory; }
|
||
const StorageFactory & getStorageFactory() const { return shared->storage_factory; }
|
||
const FormatFactory & getFormatFactory() const { return shared->format_factory; }
|
||
const Dictionaries & getDictionaries() const;
|
||
|
||
/// Получить запрос на CREATE таблицы.
|
||
ASTPtr getCreateQuery(const String & database_name, const String & table_name) const;
|
||
|
||
/// Для методов ниже может быть необходимо захватывать mutex самостоятельно.
|
||
Poco::Mutex & getMutex() const { return shared->mutex; }
|
||
|
||
/// Метод getDatabases не потокобезопасен. При работе со списком БД и таблиц, вы должны захватить mutex.
|
||
const Databases & getDatabases() const { return shared->databases; }
|
||
Databases & getDatabases() { return shared->databases; }
|
||
|
||
/// При работе со списком столбцов, используйте локальный контекст, чтобы никто больше его не менял.
|
||
const NamesAndTypesList & getColumns() const { return columns; }
|
||
NamesAndTypesList & getColumns() { return columns; }
|
||
void setColumns(const NamesAndTypesList & columns_) { columns = columns_; }
|
||
|
||
Context & getSessionContext();
|
||
Context & getGlobalContext();
|
||
|
||
void setSessionContext(Context & context_) { session_context = &context_; }
|
||
void setGlobalContext(Context & context_) { global_context = &context_; }
|
||
|
||
const Settings & getSettingsRef() const { return settings; };
|
||
Settings & getSettingsRef() { return settings; };
|
||
|
||
void setProgressCallback(ProgressCallback callback);
|
||
/// Используется в InterpreterSelectQuery, чтобы передать его в IProfilingBlockInputStream.
|
||
ProgressCallback getProgressCallback() const;
|
||
|
||
/** Устанавливается в executeQuery и InterpreterSelectQuery. Затем используется в IProfilingBlockInputStream,
|
||
* чтобы обновлять и контролировать информацию об общем количестве потраченных на запрос ресурсов.
|
||
*/
|
||
void setProcessListElement(ProcessList::Element * elem);
|
||
/// Может вернуть NULL, если запрос не был вставлен в ProcessList.
|
||
ProcessList::Element * getProcessListElement();
|
||
|
||
/// Список всех запросов.
|
||
ProcessList & getProcessList() { return shared->process_list; }
|
||
const ProcessList & getProcessList() const { return shared->process_list; }
|
||
|
||
/// Создать кэш разжатых блоков указанного размера. Это можно сделать только один раз.
|
||
void setUncompressedCache(size_t cache_size_in_cells);
|
||
UncompressedCachePtr getUncompressedCache() const;
|
||
|
||
/// Создать кэш засечек указанного размера. Это можно сделать только один раз.
|
||
void setMarkCache(size_t cache_size_in_bytes);
|
||
MarkCachePtr getMarkCache() const;
|
||
|
||
void initClusters();
|
||
Cluster & getCluster(const std::string & cluster_name);
|
||
|
||
void shutdown() { shared->shutdown(); }
|
||
};
|
||
|
||
|
||
}
|