mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 21:24:28 +00:00
67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
#pragma once
|
|
|
|
#include <Client/ClientBase.h>
|
|
#include <Client/LocalConnection.h>
|
|
|
|
#include <Common/ProgressIndication.h>
|
|
#include <Common/StatusFile.h>
|
|
#include <Common/InterruptListener.h>
|
|
#include <loggers/Loggers.h>
|
|
#include <Core/Settings.h>
|
|
#include <Interpreters/Context.h>
|
|
|
|
#include <filesystem>
|
|
#include <memory>
|
|
#include <optional>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/// Lightweight Application for clickhouse-local
|
|
/// No networking, no extra configs and working directories, no pid and status files, no dictionaries, no logging.
|
|
/// Quiet mode by default
|
|
class LocalServer : public ClientBase, public Loggers
|
|
{
|
|
public:
|
|
LocalServer() = default;
|
|
|
|
void initialize(Poco::Util::Application & self) override;
|
|
|
|
int main(const std::vector<String> & /*args*/) override;
|
|
|
|
protected:
|
|
bool executeMultiQuery(const String & all_queries_text) override;
|
|
|
|
void connect() override;
|
|
void processError(const String & query) const override;
|
|
String getName() const override { return "local"; }
|
|
|
|
String getQueryTextPrefix() override;
|
|
void printHelpMessage(const OptionsDescription & options_description) override;
|
|
|
|
void addAndCheckOptions(OptionsDescription & options_description, po::variables_map & options, Arguments & arguments) override;
|
|
void processOptions(const OptionsDescription & options_description, const CommandLineOptions & options,
|
|
const std::vector<Arguments> &) override;
|
|
void processConfig() override;
|
|
|
|
private:
|
|
/** Composes CREATE subquery based on passed arguments (--structure --file --table and --input-format)
|
|
* This query will be executed first, before queries passed through --query argument
|
|
* Returns empty string if it cannot compose that query.
|
|
*/
|
|
std::string getInitialCreateTableQuery();
|
|
|
|
void tryInitPath();
|
|
void setupUsers();
|
|
void cleanup();
|
|
|
|
void applyCmdOptions(ContextMutablePtr context);
|
|
void applyCmdSettings(ContextMutablePtr context);
|
|
|
|
std::optional<StatusFile> status;
|
|
std::optional<std::filesystem::path> temporary_directory_to_delete;
|
|
};
|
|
|
|
}
|