1 | #pragma once |
2 | |
3 | #include <Core/Types.h> |
4 | #include <Core/Names.h> |
5 | |
6 | #include <map> |
7 | |
8 | |
9 | |
10 | namespace Poco |
11 | { |
12 | namespace Util |
13 | { |
14 | class AbstractConfiguration; |
15 | } |
16 | } |
17 | |
18 | |
19 | namespace DB |
20 | { |
21 | |
22 | /** Apply substitutions from the macros in config to the string. |
23 | */ |
24 | class Macros |
25 | { |
26 | public: |
27 | Macros(); |
28 | Macros(const Poco::Util::AbstractConfiguration & config, const String & key); |
29 | |
30 | /** Replace the substring of the form {macro_name} with the value for macro_name, obtained from the config file. |
31 | * If {database} and {table} macros aren`t defined explicitly, expand them as database_name and table_name respectively. |
32 | * level - the level of recursion. |
33 | */ |
34 | String expand(const String & s, size_t level = 0, const String & database_name = "" , const String & table_name = "" ) const; |
35 | |
36 | String expand(const String & s, const String & database_name, const String & table_name) const; |
37 | |
38 | |
39 | /** Apply expand for the list. |
40 | */ |
41 | Names expand(const Names & source_names, size_t level = 0) const; |
42 | |
43 | using MacroMap = std::map<String, String>; |
44 | const MacroMap getMacroMap() const { return macros; } |
45 | |
46 | String getValue(const String & key) const; |
47 | |
48 | private: |
49 | MacroMap macros; |
50 | }; |
51 | |
52 | |
53 | } |
54 | |