mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-09 18:10:48 +00:00
75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
#include "HierarchiesProvider.h"
|
|
|
|
#include <IO/ReadBufferFromFile.h>
|
|
#include <Poco/DirectoryIterator.h>
|
|
#include <Poco/Exception.h>
|
|
#include <Poco/Util/Application.h>
|
|
#include "HierarchyFormatReader.h"
|
|
|
|
|
|
bool RegionsHierarchyDataSource::isModified() const
|
|
{
|
|
return updates_tracker.isModified();
|
|
}
|
|
|
|
IRegionsHierarchyReaderPtr RegionsHierarchyDataSource::createReader()
|
|
{
|
|
updates_tracker.fixCurrentVersion();
|
|
auto file_reader = std::make_shared<DB::ReadBufferFromFile>(path);
|
|
return std::make_unique<RegionsHierarchyFormatReader>(std::move(file_reader));
|
|
}
|
|
|
|
|
|
RegionsHierarchiesDataProvider::RegionsHierarchiesDataProvider(const std::string & path_) : path(path_)
|
|
{
|
|
discoverFilesWithCustomHierarchies();
|
|
}
|
|
|
|
void RegionsHierarchiesDataProvider::discoverFilesWithCustomHierarchies()
|
|
{
|
|
std::string basename = Poco::Path(path).getBaseName();
|
|
|
|
Poco::Path dir_path = Poco::Path(path).absolute().parent();
|
|
|
|
Poco::DirectoryIterator dir_end;
|
|
for (Poco::DirectoryIterator dir_it(dir_path); dir_it != dir_end; ++dir_it)
|
|
{
|
|
std::string candidate_basename = dir_it.path().getBaseName();
|
|
|
|
if (candidate_basename.starts_with(basename)
|
|
&& (candidate_basename.size() > basename.size() + 1)
|
|
&& (candidate_basename[basename.size()] == '_'))
|
|
{
|
|
const std::string suffix = candidate_basename.substr(basename.size() + 1);
|
|
hierarchy_files.emplace(suffix, dir_it->path());
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<std::string> RegionsHierarchiesDataProvider::listCustomHierarchies() const
|
|
{
|
|
std::vector<std::string> names;
|
|
names.reserve(hierarchy_files.size());
|
|
for (const auto & it : hierarchy_files)
|
|
names.push_back(it.first);
|
|
return names;
|
|
}
|
|
|
|
IRegionsHierarchyDataSourcePtr RegionsHierarchiesDataProvider::getDefaultHierarchySource() const
|
|
{
|
|
return std::make_shared<RegionsHierarchyDataSource>(path);
|
|
}
|
|
|
|
IRegionsHierarchyDataSourcePtr RegionsHierarchiesDataProvider::getHierarchySource(const std::string & name) const
|
|
{
|
|
auto found = hierarchy_files.find(name);
|
|
|
|
if (found != hierarchy_files.end())
|
|
{
|
|
const auto & hierarchy_file_path = found->second;
|
|
return std::make_shared<RegionsHierarchyDataSource>(hierarchy_file_path);
|
|
}
|
|
|
|
throw Poco::Exception("Regions hierarchy '" + name + "' not found");
|
|
}
|