mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-28 18:42:26 +00:00
e623ad041f
C-z is extermelly useful for some users (like myself), so provide a way to configure it in client and avoid it's ignorance in clickhouse-disks (I hope it is OK since it is not that known utility and it does not have it's own configuration, while cli option is useless, one should remeber about it). Honestly I've never seen any interactive client that forbids C-z, so ignoring it my default looks strange to me. Signed-off-by: Azat Khuzhin <a.khuzhin@semrush.com>
89 lines
2.6 KiB
C++
89 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <filesystem>
|
|
#include <optional>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
#include <Loggers/Loggers.h>
|
|
#include <Disks/IDisk.h>
|
|
|
|
#include <Interpreters/Context.h>
|
|
#include <boost/program_options/options_description.hpp>
|
|
#include <boost/program_options/variables_map.hpp>
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
namespace DB
|
|
{
|
|
|
|
std::vector<String> split(const String & text, const String & delimiters);
|
|
|
|
using ProgramOptionsDescription = boost::program_options::options_description;
|
|
using CommandLineOptions = boost::program_options::variables_map;
|
|
|
|
class DiskWithPath
|
|
{
|
|
public:
|
|
explicit DiskWithPath(DiskPtr disk_, std::optional<String> path_ = std::nullopt);
|
|
|
|
String getAbsolutePath(const String & any_path) const { return normalizePath(fs::path(path) / any_path); }
|
|
|
|
String getCurrentPath() const { return path; }
|
|
|
|
bool isDirectory(const String & any_path) const
|
|
{
|
|
return disk->isDirectory(getRelativeFromRoot(any_path)) || (getRelativeFromRoot(any_path).empty() && (disk->isDirectory("/")));
|
|
}
|
|
|
|
std::vector<String> listAllFilesByPath(const String & any_path) const;
|
|
|
|
std::vector<String> getAllFilesByPattern(const String & pattern) const;
|
|
|
|
DiskPtr getDisk() const { return disk; }
|
|
|
|
void setPath(const String & any_path);
|
|
|
|
String getRelativeFromRoot(const String & any_path) const { return normalizePathAndGetAsRelative(getAbsolutePath(any_path)); }
|
|
|
|
private:
|
|
static String validatePathAndGetAsRelative(const String & path);
|
|
static std::string normalizePathAndGetAsRelative(const std::string & messyPath);
|
|
static std::string normalizePath(const std::string & messyPath);
|
|
|
|
const DiskPtr disk;
|
|
String path;
|
|
};
|
|
|
|
class DisksClient
|
|
{
|
|
public:
|
|
explicit DisksClient(std::vector<std::pair<DiskPtr, std::optional<String>>> && disks_with_paths, std::optional<String> begin_disk);
|
|
|
|
const DiskWithPath & getDiskWithPath(const String & disk) const;
|
|
|
|
DiskWithPath & getDiskWithPath(const String & disk);
|
|
|
|
const DiskWithPath & getCurrentDiskWithPath() const;
|
|
|
|
DiskWithPath & getCurrentDiskWithPath();
|
|
|
|
DiskPtr getCurrentDisk() const { return getCurrentDiskWithPath().getDisk(); }
|
|
|
|
DiskPtr getDisk(const String & disk) const { return getDiskWithPath(disk).getDisk(); }
|
|
|
|
void switchToDisk(const String & disk_, const std::optional<String> & path_);
|
|
|
|
std::vector<String> getAllDiskNames() const;
|
|
|
|
std::vector<String> getAllFilesByPatternFromAllDisks(const String & pattern) const;
|
|
|
|
|
|
private:
|
|
void addDisk(DiskPtr disk_, const std::optional<String> & path_);
|
|
|
|
String current_disk;
|
|
std::unordered_map<String, DiskWithPath> disks;
|
|
};
|
|
}
|