2014-01-28 16:45:10 +00:00
|
|
|
#pragma once
|
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <TableFunctions/ITableFunction.h>
|
2014-01-28 16:45:10 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2017-05-23 18:44:36 +00:00
|
|
|
/* merge (db_name, tables_regexp) - creates a temporary StorageMerge.
|
2017-04-16 15:00:33 +00:00
|
|
|
* The structure of the table is taken from the first table that came up, suitable for regexp.
|
|
|
|
* If there is no such table, an exception is thrown.
|
2014-01-28 16:45:10 +00:00
|
|
|
*/
|
2017-06-10 09:04:31 +00:00
|
|
|
class TableFunctionMerge : public ITableFunction
|
2014-01-28 16:45:10 +00:00
|
|
|
{
|
|
|
|
public:
|
2017-06-10 09:04:31 +00:00
|
|
|
static constexpr auto name = "merge";
|
|
|
|
std::string getName() const override { return name; }
|
2021-07-01 14:07:55 +00:00
|
|
|
|
2018-03-02 05:03:28 +00:00
|
|
|
private:
|
2021-04-10 23:33:54 +00:00
|
|
|
StoragePtr executeImpl(const ASTPtr & ast_function, ContextPtr context, const std::string & table_name, ColumnsDescription cached_columns) const override;
|
2020-04-06 05:19:40 +00:00
|
|
|
const char * getStorageTypeName() const override { return "Merge"; }
|
2020-10-14 12:19:29 +00:00
|
|
|
|
2021-07-02 02:11:28 +00:00
|
|
|
using TableSet = std::set<String>;
|
2021-09-25 02:48:24 +00:00
|
|
|
using DBToTableSetMap = std::map<String, TableSet>;
|
|
|
|
const DBToTableSetMap & getSourceDatabasesAndTables(ContextPtr context) const;
|
2021-04-10 23:33:54 +00:00
|
|
|
ColumnsDescription getActualTableStructure(ContextPtr context) const override;
|
2023-02-26 13:45:10 +00:00
|
|
|
std::vector<size_t> skipAnalysisForArguments(const QueryTreeNodePtr & query_node_table_function, ContextPtr context) const override;
|
2021-04-10 23:33:54 +00:00
|
|
|
void parseArguments(const ASTPtr & ast_function, ContextPtr context) override;
|
2021-07-02 02:11:28 +00:00
|
|
|
static TableSet getMatchedTablesWithAccess(const String & database_name, const String & table_regexp, const ContextPtr & context);
|
2020-10-14 12:19:29 +00:00
|
|
|
|
2021-06-25 13:51:17 +00:00
|
|
|
String source_database_name_or_regexp;
|
2020-12-10 20:16:53 +00:00
|
|
|
String source_table_regexp;
|
2021-06-25 13:51:17 +00:00
|
|
|
bool database_is_regexp = false;
|
2021-09-25 02:48:24 +00:00
|
|
|
mutable std::optional<DBToTableSetMap> source_databases_and_tables;
|
2014-01-28 16:45:10 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
}
|