2012-12-18 20:06:36 +00:00
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <DB/Functions/IFunction.h>
|
2015-09-29 19:19:54 +00:00
|
|
|
|
#include <common/singleton.h>
|
2012-12-18 20:06:36 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
class Context;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Позволяет получить функцию по имени.
|
2016-05-28 17:31:50 +00:00
|
|
|
|
* Функция при создании также может использовать для инициализации (например, захватить shared_ptr)
|
2012-12-18 20:06:36 +00:00
|
|
|
|
* какие-нибудь справочники, находящиеся в Context-е.
|
|
|
|
|
*/
|
2014-12-17 11:53:17 +00:00
|
|
|
|
class FunctionFactory : public Singleton<FunctionFactory>
|
2012-12-18 20:06:36 +00:00
|
|
|
|
{
|
2015-04-24 15:49:30 +00:00
|
|
|
|
friend class StorageSystemFunctions;
|
|
|
|
|
|
2014-08-22 00:57:20 +00:00
|
|
|
|
private:
|
2016-05-28 15:42:22 +00:00
|
|
|
|
typedef FunctionPtr (*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();
|
|
|
|
|
|
2015-11-29 08:06:29 +00:00
|
|
|
|
FunctionPtr get(const String & name, const Context & context) const; /// Кидает исключение, если не нашлось.
|
|
|
|
|
FunctionPtr tryGet(const String & name, const Context & context) const; /// Возвращает nullptr, если не нашлось.
|
2014-08-22 00:57:20 +00:00
|
|
|
|
|
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
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
}
|