| 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 |  | 
|---|