| 1 | #pragma once |
| 2 | |
| 3 | #include <Core/QueryProcessingStage.h> |
| 4 | #include <DataStreams/BlockIO.h> |
| 5 | |
| 6 | #include <Processors/QueryPipeline.h> |
| 7 | |
| 8 | namespace DB |
| 9 | { |
| 10 | |
| 11 | class ReadBuffer; |
| 12 | class WriteBuffer; |
| 13 | class Context; |
| 14 | |
| 15 | |
| 16 | /// Parse and execute a query. |
| 17 | void executeQuery( |
| 18 | ReadBuffer & istr, /// Where to read query from (and data for INSERT, if present). |
| 19 | WriteBuffer & ostr, /// Where to write query output to. |
| 20 | bool allow_into_outfile, /// If true and the query contains INTO OUTFILE section, redirect output to that file. |
| 21 | Context & context, /// DB, tables, data types, storage engines, functions, aggregate functions... |
| 22 | std::function<void(const String &)> set_content_type, /// If non-empty callback is passed, it will be called with the Content-Type of the result. |
| 23 | std::function<void(const String &)> set_query_id /// If non-empty callback is passed, it will be called with the query id. |
| 24 | ); |
| 25 | |
| 26 | |
| 27 | /// More low-level function for server-to-server interaction. |
| 28 | /// Prepares a query for execution but doesn't execute it. |
| 29 | /// Returns a pair of block streams which, when used, will result in query execution. |
| 30 | /// This means that the caller can to the extent control the query execution pipeline. |
| 31 | /// |
| 32 | /// To execute: |
| 33 | /// * if present, write INSERT data into BlockIO::out |
| 34 | /// * then read the results from BlockIO::in. |
| 35 | /// |
| 36 | /// If the query doesn't involve data insertion or returning of results, out and in respectively |
| 37 | /// will be equal to nullptr. |
| 38 | /// |
| 39 | /// Correctly formatting the results (according to INTO OUTFILE and FORMAT sections) |
| 40 | /// must be done separately. |
| 41 | BlockIO executeQuery( |
| 42 | const String & query, /// Query text without INSERT data. The latter must be written to BlockIO::out. |
| 43 | Context & context, /// DB, tables, data types, storage engines, functions, aggregate functions... |
| 44 | bool internal = false, /// If true, this query is caused by another query and thus needn't be registered in the ProcessList. |
| 45 | QueryProcessingStage::Enum stage = QueryProcessingStage::Complete, /// To which stage the query must be executed. |
| 46 | bool may_have_embedded_data = false, /// If insert query may have embedded data |
| 47 | bool allow_processors = true /// If can use processors pipeline |
| 48 | ); |
| 49 | |
| 50 | |
| 51 | QueryPipeline executeQueryWithProcessors( |
| 52 | const String & query, /// Query text without INSERT data. The latter must be written to BlockIO::out. |
| 53 | Context & context, /// DB, tables, data types, storage engines, functions, aggregate functions... |
| 54 | bool internal = false, /// If true, this query is caused by another query and thus needn't be registered in the ProcessList. |
| 55 | QueryProcessingStage::Enum stage = QueryProcessingStage::Complete, /// To which stage the query must be executed. |
| 56 | bool may_have_embedded_data = false /// If insert query may have embedded data |
| 57 | ); |
| 58 | |
| 59 | } |
| 60 | |