Merge

dbms/include/DB/Common/Macros.h

@@ -0,0 +1,27 @@
+#pragma once
+
+#include <DB/Core/Types.h>
+#include <Poco/Util/AbstractConfiguration.h>
+#include <map>
+
+namespace DB
+{
+
+/** Раскрывает в строке макросы из конфига.
+  */
+class Macros
+{
+public:
+	Macros();
+	Macros(const Poco::Util::AbstractConfiguration & config, const String & key);
+
+	/// Заменить в строке подстроки вида {macro_name} на значение для macro_name, полученное из конфига.
+	String expand(const String & s) const;
+
+private:
+	typedef std::map<String, String> MacroMap;
+
+	MacroMap macros;
+};
+
+}

dbms/include/DB/Interpreters/Context.h

@@ -9,6 +9,7 @@
 #include <Yandex/logger_useful.h>
 
 #include <DB/Core/NamesAndTypes.h>
+#include <DB/Common/Macros.h>
 #include <DB/IO/UncompressedCache.h>
 #include <DB/Storages/MarkCache.h>
 #include <DB/DataStreams/FormatFactory.h>
@@ -95,8 +96,9 @@ struct ContextShared
 	ViewDependencies view_dependencies;						/// Текущие зависимости
 	ConfigurationPtr users_config;							/// Конфиг с секциями users, profiles и quotas.
 	InterserverIOHandler interserver_io_handler;			/// Обработчик для межсерверной передачи данных.
-	String default_replica_name;							/// Имя реплики из конфига.
+	String default_replica_name;							/// Имя реплики из конфига. DEPRECATED
 	BackgroundProcessingPoolPtr background_pool;			/// Пул потоков для фоновой работы, выполняемой таблицами.
+	Macros macros;											/// Подстановки из конфига.
 
 	/// Кластеры для distributed таблиц
 	/// Создаются при создании Distributed таблиц, так как нужно дождаться пока будут выставлены Settings
@@ -242,6 +244,9 @@ public:
 	String getDefaultReplicaName() const;
 	void setDefaultReplicaName(const String & name);
 
+	const Macros & getMacros() const;
+	void setMacros(Macros && macros);
+
 	Settings getSettings() const;
 	void setSettings(const Settings & settings_);
 

dbms/src/Common/Macros.cpp

@@ -0,0 +1,60 @@
+#include <DB/Common/Macros.h>
+#include <DB/Core/Exception.h>
+#include <DB/Core/ErrorCodes.h>
+
+namespace DB
+{
+
+Macros::Macros() {}
+
+Macros::Macros(const Poco::Util::AbstractConfiguration & config, const String & root_key)
+{
+	Poco::Util::AbstractConfiguration::Keys keys;
+	config.keys(root_key, keys);
+	for (const String & key : keys)
+	{
+		macros[key] = config.getString(root_key + "." + key);
+	}
+}
+
+String Macros::expand(const String & s) const
+{
+	if (s.find('{') == String::npos)
+		return s;
+
+	String res;
+	size_t pos = 0;
+	while (true)
+	{
+		size_t begin = s.find('{', pos);
+
+		if (begin == String::npos)
+		{
+			res.append(s, pos, String::npos);
+			break;
+		}
+		else
+		{
+			res.append(s, pos, begin - pos);
+		}
+
+		++begin;
+		size_t end = s.find('}', begin);
+		if (end == String::npos)
+			throw Exception("Unbalanced { and } in string with macros: \"" + s + "\"", ErrorCodes::SYNTAX_ERROR);
+
+		String macro_name = s.substr(begin, end - begin);
+
+		auto it = macros.find(macro_name);
+		if (it == macros.end())
+			throw Exception("No macro " + macro_name + " in config", ErrorCodes::SYNTAX_ERROR);
+
+		res += it->second;
+
+		pos = end + 1;
+	}
+
+	return res;
+}
+
+}

dbms/src/Interpreters/Context.cpp

@@ -459,6 +459,17 @@ void Context::setDefaultReplicaName(const String & name)
 	shared->default_replica_name = name;
 }
 
+const Macros& Context::getMacros() const
+{
+	return shared->macros;
+}
+
+void Context::setMacros(Macros && macros)
+{
+	/// Полагаемся, что это присваивание происходит один раз при старте сервера. Если это не так, нужно использовать мьютекс.
+	shared->macros = macros;
+}
+
 
 Context & Context::getSessionContext()
 {

dbms/src/Server/Server.cpp

@@ -373,6 +373,9 @@ int Server::main(const std::vector<std::string> & args)
 	if (config().has("replica_name"))
 		global_context->setDefaultReplicaName(config().getString("replica_name"));
 
+	if (config().has("macros"))
+		global_context->setMacros(Macros(config(), "macros"));
+
 	std::string users_config_path = config().getString("users_config", config().getString("config-file", "config.xml"));
 	auto users_config_reloader = stdext::make_unique<UsersConfigReloader>(users_config_path, global_context.get());
 

dbms/src/Storages/StorageReplicatedMergeTree.cpp

@@ -45,8 +45,9 @@ StorageReplicatedMergeTree::StorageReplicatedMergeTree(
 	const MergeTreeSettings & settings_)
 	:
 	context(context_), zookeeper(context.getZooKeeper()), database_name(database_name_),
-	table_name(name_), full_path(path_ + escapeForFileName(table_name) + '/'), zookeeper_path(zookeeper_path_),
-	replica_name(replica_name_),
+	table_name(name_), full_path(path_ + escapeForFileName(table_name) + '/'),
+	zookeeper_path(context.getMacros().expand(zookeeper_path_)),
+	replica_name(context.getMacros().expand(replica_name_)),
 	data(	full_path, columns_, context_, primary_expr_ast_, date_column_name_, sampling_expression_,
 			index_granularity_, mode_, sign_column_, settings_, database_name_ + "." + table_name, true,
 			std::bind(&StorageReplicatedMergeTree::enqueuePartForCheck, this, std::placeholders::_1)),