ClickHouse/dbms/include/DB/Interpreters/Context.h

149 lines
6.9 KiB
C
Raw Normal View History

2011-08-09 19:19:25 +00:00
#pragma once
#include <map>
2011-08-18 20:33:20 +00:00
#include <set>
2011-08-09 19:19:25 +00:00
2011-08-19 18:31:14 +00:00
#include <Poco/SharedPtr.h>
#include <Poco/Mutex.h>
#include <Yandex/logger_useful.h>
2011-08-09 19:19:25 +00:00
#include <DB/Core/NamesAndTypes.h>
2012-03-11 08:52:56 +00:00
#include <DB/DataStreams/FormatFactory.h>
2011-08-18 20:33:20 +00:00
#include <DB/Storages/IStorage.h>
#include <DB/Functions/FunctionFactory.h>
2011-09-19 01:42:16 +00:00
#include <DB/AggregateFunctions/AggregateFunctionFactory.h>
2011-08-18 20:33:20 +00:00
#include <DB/DataTypes/DataTypeFactory.h>
2011-10-31 17:30:44 +00:00
#include <DB/Storages/StorageFactory.h>
2012-03-05 00:09:41 +00:00
#include <DB/Interpreters/Settings.h>
2012-12-19 20:15:15 +00:00
#include <DB/Interpreters/Dictionaries.h>
2011-08-09 19:19:25 +00:00
namespace DB
{
2011-08-19 18:31:14 +00:00
using Poco::SharedPtr;
2011-08-18 20:33:20 +00:00
/// имя таблицы -> таблица
typedef std::map<String, StoragePtr> Tables;
/// имя БД -> таблицы
typedef std::map<String, Tables> Databases;
/// имя БД -> dropper
typedef std::map<String, DatabaseDropperPtr> DatabaseDroppers;
2011-08-18 20:33:20 +00:00
2011-08-09 19:19:25 +00:00
/** Набор известных объектов, которые могут быть использованы в запросе.
2012-08-02 17:33:31 +00:00
* Разделяемая часть.
2011-08-09 19:19:25 +00:00
*/
2012-08-02 17:33:31 +00:00
struct ContextShared
2011-08-09 19:19:25 +00:00
{
2011-09-19 01:42:16 +00:00
String path; /// Путь к директории с данными, со слешем на конце.
2012-08-02 17:33:31 +00:00
Databases databases; /// Список БД и таблиц в них.
DatabaseDroppers database_droppers; /// Reference counter'ы для ленивого удаления БД.
FunctionFactory function_factory; /// Обычные функции.
2012-08-02 17:33:31 +00:00
AggregateFunctionFactory aggregate_function_factory; /// Агрегатные функции.
DataTypeFactory data_type_factory; /// Типы данных.
StorageFactory storage_factory; /// Движки таблиц.
2012-08-02 17:33:31 +00:00
FormatFactory format_factory; /// Форматы.
mutable SharedPtr<Dictionaries> dictionaries; /// Словари Метрики. Инициализируются лениво.
2012-03-09 03:56:12 +00:00
Logger * log; /// Логгер.
2011-08-19 18:31:14 +00:00
2012-08-02 17:33:31 +00:00
mutable Poco::Mutex mutex; /// Для доступа и модификации разделяемых объектов.
2012-06-18 06:19:13 +00:00
ContextShared() : log(&Logger::get("Context")) {};
2012-08-02 17:33:31 +00:00
};
2012-06-18 07:49:19 +00:00
2012-08-02 17:33:31 +00:00
/** Набор известных объектов, которые могут быть использованы в запросе.
* Состоит из разделяемой части (всегда общей для всех сессий и запросов)
* и копируемой части (которая может быть своей для каждой сессии или запроса).
*
* Всё инкапсулировано для всяких проверок и блокировок.
*/
class Context
{
private:
typedef SharedPtr<ContextShared> Shared;
Shared shared;
2012-06-18 07:49:19 +00:00
2012-08-02 17:33:31 +00:00
String current_database; /// Текущая БД.
NamesAndTypesList columns; /// Столбцы текущей обрабатываемой таблицы.
Settings settings; /// Настройки выполнения запроса.
ProgressCallback progress_callback; /// Колбек для отслеживания прогресса выполнения запроса.
2012-08-02 17:33:31 +00:00
Context * session_context; /// Контекст сессии или NULL, если его нет. (Возможно, равен this.)
Context * global_context; /// Глобальный контекст или NULL, если его нет. (Возможно, равен this.)
2012-06-18 07:49:19 +00:00
2012-08-02 17:33:31 +00:00
public:
Context() : shared(new ContextShared), session_context(NULL), global_context(NULL) {}
2012-08-02 18:05:06 +00:00
2012-08-02 17:33:31 +00:00
String getPath() const;
void setPath(const String & path);
/// Проверка существования таблицы/БД. 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;
void addTable(const String & database_name, const String & table_name, StoragePtr table);
void addDatabase(const String & database_name);
void detachTable(const String & database_name, const String & table_name);
void detachDatabase(const String & database_name);
String getCurrentDatabase() const;
void setCurrentDatabase(const String & name);
DatabaseDropperPtr getDatabaseDropper(const String & name);
2012-08-02 17:33:31 +00:00
Settings getSettings() const;
void setSettings(const Settings & settings_);
2012-08-02 19:03:32 +00:00
/// Установить настройку по имени.
void setSetting(const String & name, const Field & value);
2012-08-02 17:33:31 +00:00
const FunctionFactory & getFunctionFactory() const { return shared->function_factory; }
const AggregateFunctionFactory & getAggregateFunctionFactory() const { return shared->aggregate_function_factory; }
2012-08-02 17:33:31 +00:00
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;
2012-08-02 17:33:31 +00:00
2012-08-17 19:53:11 +00:00
/// Получить запрос на CREATE таблицы.
ASTPtr getCreateQuery(const String & database_name, const String & table_name) const;
2012-08-02 17:33:31 +00:00
/// Для методов ниже может быть необходимо захватывать 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;
2011-08-09 19:19:25 +00:00
};
}