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