Fix build

This commit is contained in:
kssenii 2021-08-25 21:51:43 +00:00
parent 3cd56266c7
commit 0b66581de2
5 changed files with 32 additions and 4 deletions

View File

@ -9,7 +9,7 @@
#include <IO/WriteHelpers.h>
#include <IO/Operators.h>
#include <Formats/FormatFactory.h>
#include <Common/getInsertQuery.h>
#include <Parsers/getInsertQuery.h>
namespace DB

View File

@ -1,4 +1,4 @@
#include <Common/getInsertQuery.h>
#include <Parsers/getInsertQuery.h>
#include <Parsers/ASTInsertQuery.h>
#include <Parsers/ASTIdentifier.h>

View File

@ -0,0 +1,28 @@
#include <Parsers/getInsertQuery.h>
#include <Parsers/ASTInsertQuery.h>
#include <Parsers/ASTIdentifier.h>
#include <Parsers/ASTExpressionList.h>
#include <IO/WriteBufferFromString.h>
namespace DB
{
std::string getInsertQuery(const std::string & db_name, const std::string & table_name, const ColumnsWithTypeAndName & columns, IdentifierQuotingStyle quoting)
{
ASTInsertQuery query;
query.table_id.database_name = db_name;
query.table_id.table_name = table_name;
query.columns = std::make_shared<ASTExpressionList>(',');
query.children.push_back(query.columns);
for (const auto & column : columns)
query.columns->children.emplace_back(std::make_shared<ASTIdentifier>(column.name));
WriteBufferFromOwnString buf;
IAST::FormatSettings settings(buf, true);
settings.always_quote_identifiers = true;
settings.identifier_quoting_style = quoting;
query.IAST::format(settings);
return buf.str();
}
}

View File

@ -29,7 +29,7 @@
#include <Processors/Pipe.h>
#include <Processors/Sinks/SinkToStorage.h>
#include <IO/WriteHelpers.h>
#include <Common/getInsertQuery.h>
#include <Parsers/getInsertQuery.h>
#include <IO/Operators.h>
@ -292,7 +292,7 @@ private:
pqxx::connection & connection;
pqxx::work tx;
Inserter(pqxx::connection & connection_)
explicit Inserter(pqxx::connection & connection_)
: connection(connection_)
, tx(connection) {}