0fcda94162
Summary: Hardcoded query infrastructure - first concrete version - USEFUL FOR: POCs & pilots Test Plan: manual + jenkins Reviewers: sale, florijan Reviewed By: florijan Subscribers: pullbot, buda Differential Revision: https://phabricator.memgraph.io/D45
61 lines
1.7 KiB
C++
61 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <fstream>
|
|
|
|
// TODO: remove experimental from here once that becomes possible (C++17
|
|
// standard)
|
|
#include <experimental/filesystem>
|
|
namespace fs = std::experimental::filesystem;
|
|
|
|
namespace utils
|
|
{
|
|
|
|
/**
|
|
* Loads all file paths in the specified directory. Optionally
|
|
* the paths are filtered by extension.
|
|
*
|
|
* NOTE: the call isn't recursive
|
|
*
|
|
* @param directory a path to directory that will be scanned in order to find
|
|
* all paths
|
|
* @param extension paths will be filtered by this extension
|
|
*
|
|
* @return std::vector of paths founded in the directory
|
|
*/
|
|
inline auto LoadFilePaths(const fs::path &directory,
|
|
const std::string &extension = "")
|
|
{
|
|
// result container
|
|
std::vector<fs::path> file_paths;
|
|
|
|
for (auto &directory_entry : fs::recursive_directory_iterator(directory))
|
|
{
|
|
|
|
auto path = directory_entry.path().string();
|
|
|
|
// skip directories
|
|
if (!fs::is_regular_file(directory_entry)) continue;
|
|
|
|
// if extension isn't defined then put all file paths from the directory
|
|
// to the result set
|
|
if (!extension.empty()) {
|
|
// skip paths that don't have appropriate extension
|
|
auto file_extension = path.substr(path.find_last_of(".") + 1);
|
|
if (file_extension != extension) continue;
|
|
}
|
|
|
|
file_paths.emplace_back(path);
|
|
|
|
// skip paths that don't have appropriate extension
|
|
auto file_extension = path.substr(path.find_last_of(".") + 1);
|
|
if (file_extension != extension) continue;
|
|
|
|
// path has the right extension and can be placed in the result
|
|
// container
|
|
file_paths.emplace_back(path);
|
|
}
|
|
|
|
return file_paths;
|
|
}
|
|
}
|