2017-04-01 09:19:00 +00:00
|
|
|
#include <Columns/ColumnString.h>
|
|
|
|
#include <Columns/ColumnsNumber.h>
|
|
|
|
#include <DataTypes/DataTypeString.h>
|
|
|
|
#include <DataTypes/DataTypeDateTime.h>
|
|
|
|
#include <DataStreams/OneBlockInputStream.h>
|
|
|
|
#include <Storages/System/StorageSystemTables.h>
|
2017-11-20 05:22:54 +00:00
|
|
|
#include <Storages/VirtualColumnUtils.h>
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Databases/IDatabase.h>
|
|
|
|
#include <Interpreters/Context.h>
|
2012-05-08 11:19:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2015-01-21 04:30:33 +00:00
|
|
|
StorageSystemTables::StorageSystemTables(const std::string & name_)
|
2017-04-01 07:20:54 +00:00
|
|
|
: name(name_),
|
|
|
|
columns
|
|
|
|
{
|
2017-10-11 14:41:49 +00:00
|
|
|
{"database", std::make_shared<DataTypeString>()},
|
|
|
|
{"name", std::make_shared<DataTypeString>()},
|
|
|
|
{"engine", std::make_shared<DataTypeString>()},
|
|
|
|
{"metadata_modification_time", std::make_shared<DataTypeDateTime>()}
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
2012-05-08 11:19:00 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2015-01-21 04:30:33 +00:00
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
static ColumnPtr getFilteredDatabases(const ASTPtr & query, const Context & context)
|
2015-01-21 04:30:33 +00:00
|
|
|
{
|
2017-12-16 00:49:03 +00:00
|
|
|
MutableColumnPtr column = ColumnString::create();
|
2017-04-01 07:20:54 +00:00
|
|
|
for (const auto & db : context.getDatabases())
|
2017-12-16 00:49:03 +00:00
|
|
|
column->insert(db.first);
|
2016-08-04 23:35:07 +00:00
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
Block block { ColumnWithTypeAndName( std::move(column), std::make_shared<DataTypeString>(), "database" ) };
|
2017-04-01 07:20:54 +00:00
|
|
|
VirtualColumnUtils::filterBlockWithQuery(query, block, context);
|
2017-12-16 00:49:03 +00:00
|
|
|
return block.getByPosition(0).column;
|
2013-02-06 11:26:35 +00:00
|
|
|
}
|
|
|
|
|
2012-05-08 11:19:00 +00:00
|
|
|
|
|
|
|
BlockInputStreams StorageSystemTables::read(
|
2017-04-01 07:20:54 +00:00
|
|
|
const Names & column_names,
|
2017-07-15 03:48:36 +00:00
|
|
|
const SelectQueryInfo & query_info,
|
2017-04-01 07:20:54 +00:00
|
|
|
const Context & context,
|
|
|
|
QueryProcessingStage::Enum & processed_stage,
|
2017-12-02 02:47:12 +00:00
|
|
|
const size_t /*max_block_size*/,
|
|
|
|
const unsigned /*num_streams*/)
|
2012-05-08 11:19:00 +00:00
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
check(column_names);
|
|
|
|
processed_stage = QueryProcessingStage::FetchColumns;
|
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
MutableColumns res_columns = getSampleBlock().cloneEmptyColumns();
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
ColumnPtr filtered_databases_column = getFilteredDatabases(query_info.query, context);
|
2017-04-01 07:20:54 +00:00
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
for (size_t row_number = 0; row_number < filtered_databases_column->size(); ++row_number)
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
2017-12-16 00:49:03 +00:00
|
|
|
std::string database_name = filtered_databases_column->getDataAt(row_number).toString();
|
2017-04-01 07:20:54 +00:00
|
|
|
auto database = context.tryGetDatabase(database_name);
|
|
|
|
|
|
|
|
if (!database)
|
|
|
|
{
|
|
|
|
/// Database was deleted just now.
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2017-09-11 12:39:01 +00:00
|
|
|
for (auto iterator = database->getIterator(context); iterator->isValid(); iterator->next())
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
auto table_name = iterator->name();
|
2017-12-16 00:49:03 +00:00
|
|
|
res_columns[0]->insert(database_name);
|
|
|
|
res_columns[1]->insert(table_name);
|
|
|
|
res_columns[2]->insert(iterator->table()->getName());
|
|
|
|
res_columns[3]->insert(static_cast<UInt64>(database->getTableMetadataModificationTime(context, table_name)));
|
2017-04-01 07:20:54 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-12-16 00:49:03 +00:00
|
|
|
return BlockInputStreams(1, std::make_shared<OneBlockInputStream>(getSampleBlock().cloneWithColumns(std::move(res_columns))));
|
2012-05-08 11:19:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|