mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 14:11:58 +00:00
69 lines
2.6 KiB
C++
69 lines
2.6 KiB
C++
#include <Parsers/ASTInsertQuery.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Interpreters/InterpreterSetQuery.h>
|
|
#include <IO/ConcatReadBuffer.h>
|
|
#include <IO/ReadBufferFromMemory.h>
|
|
#include <DataStreams/BlockIO.h>
|
|
#include <DataStreams/InputStreamFromASTInsertQuery.h>
|
|
#include <DataStreams/AddingDefaultsBlockInputStream.h>
|
|
#include <Storages/ColumnsDescription.h>
|
|
#include <Storages/IStorage.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
extern const int INVALID_USAGE_OF_INPUT;
|
|
}
|
|
|
|
|
|
InputStreamFromASTInsertQuery::InputStreamFromASTInsertQuery(
|
|
const ASTPtr & ast, ReadBuffer * input_buffer_tail_part, const Block & header, const Context & context, const ASTPtr & input_function)
|
|
{
|
|
const auto * ast_insert_query = ast->as<ASTInsertQuery>();
|
|
|
|
if (!ast_insert_query)
|
|
throw Exception("Logical error: query requires data to insert, but it is not INSERT query", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
String format = ast_insert_query->format;
|
|
if (format.empty())
|
|
{
|
|
if (input_function)
|
|
throw Exception("FORMAT must be specified for function input()", ErrorCodes::INVALID_USAGE_OF_INPUT);
|
|
format = "Values";
|
|
}
|
|
|
|
/// Data could be in parsed (ast_insert_query.data) and in not parsed yet (input_buffer_tail_part) part of query.
|
|
|
|
input_buffer_ast_part = std::make_unique<ReadBufferFromMemory>(
|
|
ast_insert_query->data, ast_insert_query->data ? ast_insert_query->end - ast_insert_query->data : 0);
|
|
|
|
ConcatReadBuffer::ReadBuffers buffers;
|
|
if (ast_insert_query->data)
|
|
buffers.push_back(input_buffer_ast_part.get());
|
|
|
|
if (input_buffer_tail_part)
|
|
buffers.push_back(input_buffer_tail_part);
|
|
|
|
/** NOTE Must not read from 'input_buffer_tail_part' before read all between 'ast_insert_query.data' and 'ast_insert_query.end'.
|
|
* - because 'query.data' could refer to memory piece, used as buffer for 'input_buffer_tail_part'.
|
|
*/
|
|
|
|
input_buffer_contacenated = std::make_unique<ConcatReadBuffer>(buffers);
|
|
|
|
res_stream = context.getInputFormat(format, *input_buffer_contacenated, header, context.getSettings().max_insert_block_size);
|
|
|
|
if (context.getSettingsRef().input_format_defaults_for_omitted_fields && !ast_insert_query->table.empty() && !input_function)
|
|
{
|
|
StoragePtr storage = context.getTable(ast_insert_query->database, ast_insert_query->table);
|
|
auto column_defaults = storage->getColumns().getDefaults();
|
|
if (!column_defaults.empty())
|
|
res_stream = std::make_shared<AddingDefaultsBlockInputStream>(res_stream, column_defaults, context);
|
|
}
|
|
}
|
|
|
|
}
|