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

#pragma once

#include <map>
#include <set>

#include <Poco/SharedPtr.h>
#include <Poco/Mutex.h>

#include <DB/Core/NamesAndTypes.h>
#include <DB/DataStreams/FormatFactory.h>
#include <DB/Storages/IStorage.h>
#include <DB/Functions/FunctionsLibrary.h>
#include <DB/AggregateFunctions/AggregateFunctionFactory.h>
#include <DB/DataTypes/DataTypeFactory.h>
#include <DB/Storages/StorageFactory.h>
#include <DB/Interpreters/Settings.h>


namespace DB
{

using Poco::SharedPtr;

/// имя таблицы -> таблица
typedef std::map<String, StoragePtr> Tables;

/// имя БД -> таблицы
typedef std::map<String, Tables> Databases;


/** Набор известных объектов, которые могут быть использованы в запросе.
  */
struct Context
{
	String path;											/// Путь к директории с данными, со слешем на конце.
	SharedPtr<Databases> databases;							/// Список БД и таблиц в них.
	String current_database;								/// Текущая БД.
	SharedPtr<Functions> functions;							/// Обычные функции.
	AggregateFunctionFactoryPtr aggregate_function_factory; /// Агрегатные функции.
	DataTypeFactoryPtr data_type_factory;					/// Типы данных.
	StorageFactoryPtr storage_factory;						/// Движки таблиц.
	FormatFactoryPtr format_factory;						/// Форматы.
	NamesAndTypesList columns;								/// Столбцы текущей обрабатываемой таблицы.
	Settings settings;										/// Настройки выполнения запроса.
	Logger * log;											/// Логгер.

	mutable SharedPtr<Poco::FastMutex> mutex;				/// Для доступа и модификации разделяемых объектов.

	Context() : databases(new Databases), functions(new Functions), log(&Logger::get("Context")), mutex(new Poco::FastMutex) {}

	/** В сервере есть глобальный контекст.
	  * При соединении, он копируется в контекст сессии.
	  * Для каждого запроса, контекст сессии копируется в контекст запроса.
	  * Блокировка нужна, так как запрос может модифицировать глобальный контекст (SET GLOBAL ...).
	  */
	Context(const Context & rhs)
	{
		Poco::ScopedLock<Poco::FastMutex> lock(*rhs.mutex);

		path 						= rhs.path;
		databases 					= rhs.databases;
		current_database 			= rhs.current_database;
		functions 					= rhs.functions;
		aggregate_function_factory 	= rhs.aggregate_function_factory;
		data_type_factory			= rhs.data_type_factory;
		storage_factory				= rhs.storage_factory;
		columns						= rhs.columns;
		settings					= rhs.settings;
		log							= rhs.log;
		mutex						= rhs.mutex;
	}
};


}