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 | |