2012-12-18 20:06:36 +00:00
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <DB/Functions/IFunction.h>
|
2014-12-17 11:53:17 +00:00
|
|
|
|
#include <Yandex/singleton.h>
|
2012-12-18 20:06:36 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
class Context;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Позволяет получить функцию по имени.
|
|
|
|
|
* Функция при создании также может использовать для инициализации (например, захватить SharedPtr)
|
|
|
|
|
* какие-нибудь справочники, находящиеся в Context-е.
|
|
|
|
|
*/
|
2014-12-17 11:53:17 +00:00
|
|
|
|
class FunctionFactory : public Singleton<FunctionFactory>
|
2012-12-18 20:06:36 +00:00
|
|
|
|
{
|
2014-08-22 00:57:20 +00:00
|
|
|
|
private:
|
2014-08-22 15:48:48 +00:00
|
|
|
|
typedef IFunction* (*Creator)(const Context & context); /// Не std::function, так как меньше indirection и размер объекта.
|
2014-08-22 00:57:20 +00:00
|
|
|
|
std::unordered_map<String, Creator> functions;
|
|
|
|
|
|
2012-12-18 20:06:36 +00:00
|
|
|
|
public:
|
2014-08-22 00:57:20 +00:00
|
|
|
|
FunctionFactory();
|
|
|
|
|
|
|
|
|
|
FunctionPtr get(const String & name, const Context & context) const;
|
|
|
|
|
|
2014-11-12 17:23:26 +00:00
|
|
|
|
template <typename F> void registerFunction()
|
2014-08-22 00:57:20 +00:00
|
|
|
|
{
|
2014-11-12 17:23:26 +00:00
|
|
|
|
static_assert(std::is_same<decltype(&F::create), Creator>::value, "F::create has incorrect type");
|
|
|
|
|
functions[F::name] = &F::create;
|
2014-08-22 00:57:20 +00:00
|
|
|
|
}
|
2012-12-18 20:06:36 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
}
|