mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 09:22:05 +00:00
91 lines
3.0 KiB
C++
91 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#include <QueryPipeline/BlockIO.h>
|
|
#include <Interpreters/IInterpreter.h>
|
|
#include <Parsers/ASTInsertQuery.h>
|
|
#include <Storages/StorageInMemoryMetadata.h>
|
|
#include <Common/ThreadStatus.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class Chain;
|
|
class ThreadStatus;
|
|
|
|
struct ThreadStatusesHolder;
|
|
using ThreadStatusesHolderPtr = std::shared_ptr<ThreadStatusesHolder>;
|
|
|
|
/** Interprets the INSERT query.
|
|
*/
|
|
class InterpreterInsertQuery : public IInterpreter, WithContext
|
|
{
|
|
public:
|
|
InterpreterInsertQuery(
|
|
const ASTPtr & query_ptr_,
|
|
ContextPtr context_,
|
|
bool allow_materialized_ = false,
|
|
bool no_squash_ = false,
|
|
bool no_destination_ = false,
|
|
bool async_insert_ = false);
|
|
|
|
/** Prepare a request for execution. Return block streams
|
|
* - the stream into which you can write data to execute the query, if INSERT;
|
|
* - the stream from which you can read the result of the query, if SELECT and similar;
|
|
* Or nothing if the request INSERT SELECT (self-sufficient query - does not accept the input data, does not return the result).
|
|
*/
|
|
BlockIO execute() override;
|
|
|
|
StorageID getDatabaseTable() const;
|
|
|
|
/// Return explicitly specified column names to insert.
|
|
/// It not explicit names were specified, return nullopt.
|
|
std::optional<Names> getInsertColumnNames() const;
|
|
|
|
Chain buildChain(
|
|
const StoragePtr & table,
|
|
const StorageMetadataPtr & metadata_snapshot,
|
|
const Names & columns,
|
|
ThreadStatusesHolderPtr thread_status_holder = {},
|
|
std::atomic_uint64_t * elapsed_counter_ms = nullptr);
|
|
|
|
static void extendQueryLogElemImpl(QueryLogElement & elem, ContextPtr context_);
|
|
|
|
void extendQueryLogElemImpl(QueryLogElement & elem, const ASTPtr & ast, ContextPtr context_) const override;
|
|
|
|
StoragePtr getTable(ASTInsertQuery & query);
|
|
Block getSampleBlock(const ASTInsertQuery & query, const StoragePtr & table, const StorageMetadataPtr & metadata_snapshot) const;
|
|
|
|
bool supportsTransactions() const override { return true; }
|
|
|
|
void addBuffer(std::unique_ptr<ReadBuffer> buffer) { owned_buffers.push_back(std::move(buffer)); }
|
|
|
|
bool shouldAddSquashingFroStorage(const StoragePtr & table) const;
|
|
|
|
private:
|
|
Block getSampleBlock(const Names & names, const StoragePtr & table, const StorageMetadataPtr & metadata_snapshot) const;
|
|
|
|
ASTPtr query_ptr;
|
|
const bool allow_materialized;
|
|
const bool no_squash;
|
|
const bool no_destination;
|
|
const bool async_insert;
|
|
|
|
std::vector<std::unique_ptr<ReadBuffer>> owned_buffers;
|
|
|
|
Chain buildSink(
|
|
const StoragePtr & table,
|
|
const StorageMetadataPtr & metadata_snapshot,
|
|
ThreadStatusesHolderPtr thread_status_holder,
|
|
ThreadGroupPtr running_group,
|
|
std::atomic_uint64_t * elapsed_counter_ms);
|
|
|
|
Chain buildPreSinkChain(
|
|
const Block & subsequent_header,
|
|
const StoragePtr & table,
|
|
const StorageMetadataPtr & metadata_snapshot,
|
|
const Block & query_sample_block);
|
|
};
|
|
|
|
|
|
}
|