risingwave_pb/
batch_plan.rs

1// This file is @generated by prost-build.
2#[derive(prost_helpers::AnyPB)]
3#[derive(Clone, PartialEq, ::prost::Message)]
4pub struct RowSeqScanNode {
5    #[prost(message, optional, tag = "1")]
6    pub table_desc: ::core::option::Option<super::plan_common::StorageTableDesc>,
7    #[prost(int32, repeated, tag = "2")]
8    pub column_ids: ::prost::alloc::vec::Vec<i32>,
9    /// All the ranges need to be read. i.e., they are OR'ed.
10    ///
11    /// Empty `scan_ranges` means full table scan.
12    #[prost(message, repeated, tag = "3")]
13    pub scan_ranges: ::prost::alloc::vec::Vec<ScanRange>,
14    /// The partition to read for scan tasks.
15    ///
16    /// Will be filled by the scheduler.
17    #[prost(message, optional, tag = "4")]
18    pub vnode_bitmap: ::core::option::Option<super::common::Buffer>,
19    /// Whether the order on output columns should be preserved.
20    #[prost(bool, tag = "5")]
21    pub ordered: bool,
22    /// The pushed down `batch_limit`. Max rows needed to return.
23    #[prost(uint64, optional, tag = "6")]
24    pub limit: ::core::option::Option<u64>,
25    #[prost(message, optional, tag = "7")]
26    pub as_of: ::core::option::Option<super::plan_common::AsOf>,
27}
28#[derive(prost_helpers::AnyPB)]
29#[derive(Clone, PartialEq, ::prost::Message)]
30pub struct SysRowSeqScanNode {
31    #[prost(uint32, tag = "1")]
32    pub table_id: u32,
33    #[prost(message, repeated, tag = "2")]
34    pub column_descs: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
35}
36/// The range to scan, which specifies a consecutive range of the PK
37/// and can represent: (Suppose there are N columns in the PK)
38/// - full table scan: Should not occur. Use an empty `Vec<ScanRange>` instead.
39/// - index range scan: `eq_conds` includes i (between 0 and N-1, inclusive) values,
40///      and `lower_bound` & `upper_bound` is the range for the (i+1)th column
41/// - index point get: `eq_conds` includes N values, and `lower_bound` & `upper_bound` are `None`
42#[derive(prost_helpers::AnyPB)]
43#[derive(Clone, PartialEq, ::prost::Message)]
44pub struct ScanRange {
45    /// The i-th element represents the value of the i-th PK column.
46    #[prost(bytes = "vec", repeated, tag = "1")]
47    pub eq_conds: ::prost::alloc::vec::Vec<::prost::alloc::vec::Vec<u8>>,
48    ///   The lower bound of the next PK column subsequent to those in `eq_conds`.
49    #[prost(message, optional, tag = "2")]
50    pub lower_bound: ::core::option::Option<scan_range::Bound>,
51    ///   The upper bound of the next PK column subsequent to those in `eq_conds`.
52    #[prost(message, optional, tag = "3")]
53    pub upper_bound: ::core::option::Option<scan_range::Bound>,
54}
55/// Nested message and enum types in `ScanRange`.
56pub mod scan_range {
57    /// `None` represent unbounded.
58    #[derive(prost_helpers::AnyPB)]
59    #[derive(Clone, PartialEq, ::prost::Message)]
60    pub struct Bound {
61        #[prost(bytes = "vec", repeated, tag = "1")]
62        pub value: ::prost::alloc::vec::Vec<::prost::alloc::vec::Vec<u8>>,
63        #[prost(bool, tag = "2")]
64        pub inclusive: bool,
65    }
66}
67#[derive(prost_helpers::AnyPB)]
68#[derive(Clone, PartialEq, ::prost::Message)]
69pub struct SourceNode {
70    #[prost(uint32, tag = "1")]
71    pub source_id: u32,
72    #[prost(message, repeated, tag = "2")]
73    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnCatalog>,
74    #[prost(btree_map = "string, string", tag = "3")]
75    pub with_properties: ::prost::alloc::collections::BTreeMap<
76        ::prost::alloc::string::String,
77        ::prost::alloc::string::String,
78    >,
79    #[prost(bytes = "vec", repeated, tag = "4")]
80    pub split: ::prost::alloc::vec::Vec<::prost::alloc::vec::Vec<u8>>,
81    #[prost(message, optional, tag = "5")]
82    pub info: ::core::option::Option<super::catalog::StreamSourceInfo>,
83    #[prost(btree_map = "string, message", tag = "6")]
84    pub secret_refs: ::prost::alloc::collections::BTreeMap<
85        ::prost::alloc::string::String,
86        super::secret::SecretRef,
87    >,
88}
89#[derive(prost_helpers::AnyPB)]
90#[derive(Clone, PartialEq, ::prost::Message)]
91pub struct IcebergScanNode {
92    #[prost(message, repeated, tag = "1")]
93    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnCatalog>,
94    #[prost(btree_map = "string, string", tag = "2")]
95    pub with_properties: ::prost::alloc::collections::BTreeMap<
96        ::prost::alloc::string::String,
97        ::prost::alloc::string::String,
98    >,
99    #[prost(bytes = "vec", repeated, tag = "3")]
100    pub split: ::prost::alloc::vec::Vec<::prost::alloc::vec::Vec<u8>>,
101    #[prost(btree_map = "string, message", tag = "4")]
102    pub secret_refs: ::prost::alloc::collections::BTreeMap<
103        ::prost::alloc::string::String,
104        super::secret::SecretRef,
105    >,
106    #[prost(enumeration = "iceberg_scan_node::IcebergScanType", tag = "5")]
107    pub iceberg_scan_type: i32,
108}
109/// Nested message and enum types in `IcebergScanNode`.
110pub mod iceberg_scan_node {
111    #[derive(prost_helpers::AnyPB)]
112    #[derive(
113        Clone,
114        Copy,
115        Debug,
116        PartialEq,
117        Eq,
118        Hash,
119        PartialOrd,
120        Ord,
121        ::prost::Enumeration
122    )]
123    #[repr(i32)]
124    pub enum IcebergScanType {
125        Unspecified = 0,
126        DataScan = 1,
127        EqualityDeleteScan = 2,
128        PositionDeleteScan = 3,
129        CountStar = 4,
130    }
131    impl IcebergScanType {
132        /// String value of the enum field names used in the ProtoBuf definition.
133        ///
134        /// The values are not transformed in any way and thus are considered stable
135        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
136        pub fn as_str_name(&self) -> &'static str {
137            match self {
138                Self::Unspecified => "ICEBERG_SCAN_TYPE_UNSPECIFIED",
139                Self::DataScan => "ICEBERG_SCAN_TYPE_DATA_SCAN",
140                Self::EqualityDeleteScan => "ICEBERG_SCAN_TYPE_EQUALITY_DELETE_SCAN",
141                Self::PositionDeleteScan => "ICEBERG_SCAN_TYPE_POSITION_DELETE_SCAN",
142                Self::CountStar => "ICEBERG_SCAN_TYPE_COUNT_STAR",
143            }
144        }
145        /// Creates an enum from field names used in the ProtoBuf definition.
146        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
147            match value {
148                "ICEBERG_SCAN_TYPE_UNSPECIFIED" => Some(Self::Unspecified),
149                "ICEBERG_SCAN_TYPE_DATA_SCAN" => Some(Self::DataScan),
150                "ICEBERG_SCAN_TYPE_EQUALITY_DELETE_SCAN" => {
151                    Some(Self::EqualityDeleteScan)
152                }
153                "ICEBERG_SCAN_TYPE_POSITION_DELETE_SCAN" => {
154                    Some(Self::PositionDeleteScan)
155                }
156                "ICEBERG_SCAN_TYPE_COUNT_STAR" => Some(Self::CountStar),
157                _ => None,
158            }
159        }
160    }
161}
162#[derive(prost_helpers::AnyPB)]
163#[derive(Clone, PartialEq, ::prost::Message)]
164pub struct FileScanNode {
165    #[prost(message, repeated, tag = "1")]
166    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
167    #[prost(enumeration = "file_scan_node::FileFormat", tag = "2")]
168    pub file_format: i32,
169    #[prost(enumeration = "file_scan_node::StorageType", tag = "3")]
170    pub storage_type: i32,
171    #[prost(string, tag = "4")]
172    pub s3_region: ::prost::alloc::string::String,
173    #[prost(string, tag = "5")]
174    pub s3_access_key: ::prost::alloc::string::String,
175    #[prost(string, tag = "6")]
176    pub s3_secret_key: ::prost::alloc::string::String,
177    #[prost(string, repeated, tag = "7")]
178    pub file_location: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
179    #[prost(string, tag = "8")]
180    pub s3_endpoint: ::prost::alloc::string::String,
181}
182/// Nested message and enum types in `FileScanNode`.
183pub mod file_scan_node {
184    #[derive(prost_helpers::AnyPB)]
185    #[derive(
186        Clone,
187        Copy,
188        Debug,
189        PartialEq,
190        Eq,
191        Hash,
192        PartialOrd,
193        Ord,
194        ::prost::Enumeration
195    )]
196    #[repr(i32)]
197    pub enum FileFormat {
198        Unspecified = 0,
199        Parquet = 1,
200    }
201    impl FileFormat {
202        /// String value of the enum field names used in the ProtoBuf definition.
203        ///
204        /// The values are not transformed in any way and thus are considered stable
205        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
206        pub fn as_str_name(&self) -> &'static str {
207            match self {
208                Self::Unspecified => "FILE_FORMAT_UNSPECIFIED",
209                Self::Parquet => "PARQUET",
210            }
211        }
212        /// Creates an enum from field names used in the ProtoBuf definition.
213        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
214            match value {
215                "FILE_FORMAT_UNSPECIFIED" => Some(Self::Unspecified),
216                "PARQUET" => Some(Self::Parquet),
217                _ => None,
218            }
219        }
220    }
221    #[derive(prost_helpers::AnyPB)]
222    #[derive(
223        Clone,
224        Copy,
225        Debug,
226        PartialEq,
227        Eq,
228        Hash,
229        PartialOrd,
230        Ord,
231        ::prost::Enumeration
232    )]
233    #[repr(i32)]
234    pub enum StorageType {
235        Unspecified = 0,
236        S3 = 1,
237        Gcs = 2,
238        Azblob = 3,
239    }
240    impl StorageType {
241        /// String value of the enum field names used in the ProtoBuf definition.
242        ///
243        /// The values are not transformed in any way and thus are considered stable
244        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
245        pub fn as_str_name(&self) -> &'static str {
246            match self {
247                Self::Unspecified => "STORAGE_TYPE_UNSPECIFIED",
248                Self::S3 => "S3",
249                Self::Gcs => "GCS",
250                Self::Azblob => "AZBLOB",
251            }
252        }
253        /// Creates an enum from field names used in the ProtoBuf definition.
254        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
255            match value {
256                "STORAGE_TYPE_UNSPECIFIED" => Some(Self::Unspecified),
257                "S3" => Some(Self::S3),
258                "GCS" => Some(Self::Gcs),
259                "AZBLOB" => Some(Self::Azblob),
260                _ => None,
261            }
262        }
263    }
264}
265#[derive(prost_helpers::AnyPB)]
266#[derive(Clone, PartialEq, ::prost::Message)]
267pub struct GcsFileScanNode {
268    #[prost(message, repeated, tag = "1")]
269    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
270    #[prost(enumeration = "gcs_file_scan_node::FileFormat", tag = "2")]
271    pub file_format: i32,
272    #[prost(string, tag = "3")]
273    pub credential: ::prost::alloc::string::String,
274    #[prost(string, repeated, tag = "4")]
275    pub file_location: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
276}
277/// Nested message and enum types in `GcsFileScanNode`.
278pub mod gcs_file_scan_node {
279    #[derive(prost_helpers::AnyPB)]
280    #[derive(
281        Clone,
282        Copy,
283        Debug,
284        PartialEq,
285        Eq,
286        Hash,
287        PartialOrd,
288        Ord,
289        ::prost::Enumeration
290    )]
291    #[repr(i32)]
292    pub enum FileFormat {
293        Unspecified = 0,
294        Parquet = 1,
295    }
296    impl FileFormat {
297        /// String value of the enum field names used in the ProtoBuf definition.
298        ///
299        /// The values are not transformed in any way and thus are considered stable
300        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
301        pub fn as_str_name(&self) -> &'static str {
302            match self {
303                Self::Unspecified => "FILE_FORMAT_UNSPECIFIED",
304                Self::Parquet => "PARQUET",
305            }
306        }
307        /// Creates an enum from field names used in the ProtoBuf definition.
308        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
309            match value {
310                "FILE_FORMAT_UNSPECIFIED" => Some(Self::Unspecified),
311                "PARQUET" => Some(Self::Parquet),
312                _ => None,
313            }
314        }
315    }
316}
317#[derive(prost_helpers::AnyPB)]
318#[derive(Clone, PartialEq, ::prost::Message)]
319pub struct AzblobFileScanNode {
320    #[prost(message, repeated, tag = "1")]
321    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
322    #[prost(enumeration = "azblob_file_scan_node::FileFormat", tag = "2")]
323    pub file_format: i32,
324    #[prost(string, tag = "3")]
325    pub account_name: ::prost::alloc::string::String,
326    #[prost(string, tag = "4")]
327    pub account_key: ::prost::alloc::string::String,
328    #[prost(string, tag = "5")]
329    pub endpoint: ::prost::alloc::string::String,
330    #[prost(string, repeated, tag = "6")]
331    pub file_location: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
332}
333/// Nested message and enum types in `AzblobFileScanNode`.
334pub mod azblob_file_scan_node {
335    #[derive(prost_helpers::AnyPB)]
336    #[derive(
337        Clone,
338        Copy,
339        Debug,
340        PartialEq,
341        Eq,
342        Hash,
343        PartialOrd,
344        Ord,
345        ::prost::Enumeration
346    )]
347    #[repr(i32)]
348    pub enum FileFormat {
349        Unspecified = 0,
350        Parquet = 1,
351    }
352    impl FileFormat {
353        /// String value of the enum field names used in the ProtoBuf definition.
354        ///
355        /// The values are not transformed in any way and thus are considered stable
356        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
357        pub fn as_str_name(&self) -> &'static str {
358            match self {
359                Self::Unspecified => "FILE_FORMAT_UNSPECIFIED",
360                Self::Parquet => "PARQUET",
361            }
362        }
363        /// Creates an enum from field names used in the ProtoBuf definition.
364        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
365            match value {
366                "FILE_FORMAT_UNSPECIFIED" => Some(Self::Unspecified),
367                "PARQUET" => Some(Self::Parquet),
368                _ => None,
369            }
370        }
371    }
372}
373/// NOTE(kwannoel): This will only be used in batch mode. We can change the definition as needed.
374#[derive(prost_helpers::AnyPB)]
375#[derive(Clone, PartialEq, ::prost::Message)]
376pub struct PostgresQueryNode {
377    #[prost(message, repeated, tag = "1")]
378    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
379    #[prost(string, tag = "2")]
380    pub hostname: ::prost::alloc::string::String,
381    #[prost(string, tag = "3")]
382    pub port: ::prost::alloc::string::String,
383    #[prost(string, tag = "4")]
384    pub username: ::prost::alloc::string::String,
385    #[prost(string, tag = "5")]
386    pub password: ::prost::alloc::string::String,
387    #[prost(string, tag = "6")]
388    pub database: ::prost::alloc::string::String,
389    #[prost(string, tag = "7")]
390    pub query: ::prost::alloc::string::String,
391}
392/// NOTE(kwannoel): This will only be used in batch mode. We can change the definition as needed.
393#[derive(prost_helpers::AnyPB)]
394#[derive(Clone, PartialEq, ::prost::Message)]
395pub struct MySqlQueryNode {
396    #[prost(message, repeated, tag = "1")]
397    pub columns: ::prost::alloc::vec::Vec<super::plan_common::ColumnDesc>,
398    #[prost(string, tag = "2")]
399    pub hostname: ::prost::alloc::string::String,
400    #[prost(string, tag = "3")]
401    pub port: ::prost::alloc::string::String,
402    #[prost(string, tag = "4")]
403    pub username: ::prost::alloc::string::String,
404    #[prost(string, tag = "5")]
405    pub password: ::prost::alloc::string::String,
406    #[prost(string, tag = "6")]
407    pub database: ::prost::alloc::string::String,
408    #[prost(string, tag = "7")]
409    pub query: ::prost::alloc::string::String,
410}
411#[derive(prost_helpers::AnyPB)]
412#[derive(Clone, PartialEq, ::prost::Message)]
413pub struct ProjectNode {
414    #[prost(message, repeated, tag = "1")]
415    pub select_list: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
416}
417#[derive(prost_helpers::AnyPB)]
418#[derive(Clone, PartialEq, ::prost::Message)]
419pub struct FilterNode {
420    #[prost(message, optional, tag = "1")]
421    pub search_condition: ::core::option::Option<super::expr::ExprNode>,
422}
423#[derive(prost_helpers::AnyPB)]
424#[derive(Clone, PartialEq, ::prost::Message)]
425pub struct LogRowSeqScanNode {
426    #[prost(message, optional, tag = "1")]
427    pub table_desc: ::core::option::Option<super::plan_common::StorageTableDesc>,
428    /// This records the mandatory column_ids of the original table, excluding op
429    #[prost(int32, repeated, tag = "2")]
430    pub column_ids: ::prost::alloc::vec::Vec<i32>,
431    /// The partition to read for scan tasks.
432    ///
433    /// Will be filled by the scheduler.
434    #[prost(message, optional, tag = "3")]
435    pub vnode_bitmap: ::core::option::Option<super::common::Buffer>,
436    #[prost(message, optional, tag = "4")]
437    pub old_epoch: ::core::option::Option<super::common::BatchQueryEpoch>,
438    #[prost(message, optional, tag = "5")]
439    pub new_epoch: ::core::option::Option<super::common::BatchQueryEpoch>,
440    #[prost(bool, tag = "6")]
441    pub ordered: bool,
442}
443#[derive(prost_helpers::AnyPB)]
444#[derive(Clone, PartialEq, ::prost::Message)]
445pub struct InsertNode {
446    /// Id of the table to perform inserting.
447    #[prost(uint32, tag = "1")]
448    pub table_id: u32,
449    /// Version of the table.
450    #[prost(uint64, tag = "5")]
451    pub table_version_id: u64,
452    #[prost(uint32, repeated, tag = "2")]
453    pub column_indices: ::prost::alloc::vec::Vec<u32>,
454    #[prost(message, optional, tag = "6")]
455    pub default_columns: ::core::option::Option<super::plan_common::DefaultColumns>,
456    /// An optional field and will be `None` for tables without user-defined pk.
457    /// The `BatchInsertExecutor` should add a column with NULL value which will
458    /// be filled in streaming.
459    #[prost(uint32, optional, tag = "3")]
460    pub row_id_index: ::core::option::Option<u32>,
461    #[prost(bool, tag = "4")]
462    pub returning: bool,
463    /// Session id is used to ensure that dml data from the same session should be sent to a fixed worker node and channel.
464    #[prost(uint32, tag = "7")]
465    pub session_id: u32,
466}
467#[derive(prost_helpers::AnyPB)]
468#[derive(Clone, Copy, PartialEq, ::prost::Message)]
469pub struct DeleteNode {
470    /// Id of the table to perform deleting.
471    #[prost(uint32, tag = "1")]
472    pub table_id: u32,
473    /// Version of the table.
474    #[prost(uint64, tag = "3")]
475    pub table_version_id: u64,
476    #[prost(bool, tag = "2")]
477    pub returning: bool,
478    /// Session id is used to ensure that dml data from the same session should be sent to a fixed worker node and channel.
479    #[prost(uint32, tag = "4")]
480    pub session_id: u32,
481}
482#[derive(prost_helpers::AnyPB)]
483#[derive(Clone, PartialEq, ::prost::Message)]
484pub struct UpdateNode {
485    /// Id of the table to perform updating.
486    #[prost(uint32, tag = "1")]
487    pub table_id: u32,
488    /// Version of the table.
489    #[prost(uint64, tag = "2")]
490    pub table_version_id: u64,
491    /// Expressions to generate `U-` records.
492    #[prost(message, repeated, tag = "3")]
493    pub old_exprs: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
494    /// Expressions to generate `U+` records.
495    #[prost(message, repeated, tag = "4")]
496    pub new_exprs: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
497    #[prost(bool, tag = "5")]
498    pub returning: bool,
499    /// Session id is used to ensure that dml data from the same session should be sent to a fixed worker node and channel.
500    #[prost(uint32, tag = "6")]
501    pub session_id: u32,
502}
503#[derive(prost_helpers::AnyPB)]
504#[derive(Clone, PartialEq, ::prost::Message)]
505pub struct ValuesNode {
506    #[prost(message, repeated, tag = "1")]
507    pub tuples: ::prost::alloc::vec::Vec<values_node::ExprTuple>,
508    #[prost(message, repeated, tag = "2")]
509    pub fields: ::prost::alloc::vec::Vec<super::plan_common::Field>,
510}
511/// Nested message and enum types in `ValuesNode`.
512pub mod values_node {
513    #[derive(prost_helpers::AnyPB)]
514    #[derive(Clone, PartialEq, ::prost::Message)]
515    pub struct ExprTuple {
516        #[prost(message, repeated, tag = "1")]
517        pub cells: ::prost::alloc::vec::Vec<super::super::expr::ExprNode>,
518    }
519}
520#[derive(prost_helpers::AnyPB)]
521#[derive(Clone, PartialEq, ::prost::Message)]
522pub struct SortNode {
523    #[prost(message, repeated, tag = "1")]
524    pub column_orders: ::prost::alloc::vec::Vec<super::common::ColumnOrder>,
525}
526#[derive(prost_helpers::AnyPB)]
527#[derive(Clone, PartialEq, ::prost::Message)]
528pub struct TopNNode {
529    #[prost(message, repeated, tag = "1")]
530    pub column_orders: ::prost::alloc::vec::Vec<super::common::ColumnOrder>,
531    #[prost(uint64, tag = "2")]
532    pub limit: u64,
533    #[prost(uint64, tag = "3")]
534    pub offset: u64,
535    #[prost(bool, tag = "4")]
536    pub with_ties: bool,
537}
538#[derive(prost_helpers::AnyPB)]
539#[derive(Clone, PartialEq, ::prost::Message)]
540pub struct GroupTopNNode {
541    #[prost(message, repeated, tag = "1")]
542    pub column_orders: ::prost::alloc::vec::Vec<super::common::ColumnOrder>,
543    #[prost(uint64, tag = "2")]
544    pub limit: u64,
545    #[prost(uint64, tag = "3")]
546    pub offset: u64,
547    #[prost(uint32, repeated, tag = "4")]
548    pub group_key: ::prost::alloc::vec::Vec<u32>,
549    #[prost(bool, tag = "5")]
550    pub with_ties: bool,
551}
552#[derive(prost_helpers::AnyPB)]
553#[derive(Clone, Copy, PartialEq, ::prost::Message)]
554pub struct LimitNode {
555    #[prost(uint64, tag = "1")]
556    pub limit: u64,
557    #[prost(uint64, tag = "2")]
558    pub offset: u64,
559}
560#[derive(prost_helpers::AnyPB)]
561#[derive(Clone, PartialEq, ::prost::Message)]
562pub struct NestedLoopJoinNode {
563    #[prost(enumeration = "super::plan_common::JoinType", tag = "1")]
564    pub join_type: i32,
565    #[prost(message, optional, tag = "2")]
566    pub join_cond: ::core::option::Option<super::expr::ExprNode>,
567    #[prost(uint32, repeated, tag = "3")]
568    pub output_indices: ::prost::alloc::vec::Vec<u32>,
569}
570#[derive(prost_helpers::AnyPB)]
571#[derive(Clone, PartialEq, ::prost::Message)]
572pub struct HashAggNode {
573    #[prost(uint32, repeated, tag = "1")]
574    pub group_key: ::prost::alloc::vec::Vec<u32>,
575    #[prost(message, repeated, tag = "2")]
576    pub agg_calls: ::prost::alloc::vec::Vec<super::expr::AggCall>,
577}
578#[derive(prost_helpers::AnyPB)]
579#[derive(Clone, PartialEq, ::prost::Message)]
580pub struct ExpandNode {
581    #[prost(message, repeated, tag = "1")]
582    pub column_subsets: ::prost::alloc::vec::Vec<expand_node::Subset>,
583}
584/// Nested message and enum types in `ExpandNode`.
585pub mod expand_node {
586    #[derive(prost_helpers::AnyPB)]
587    #[derive(Clone, PartialEq, ::prost::Message)]
588    pub struct Subset {
589        #[prost(uint32, repeated, tag = "1")]
590        pub column_indices: ::prost::alloc::vec::Vec<u32>,
591    }
592}
593#[derive(prost_helpers::AnyPB)]
594#[derive(Clone, PartialEq, ::prost::Message)]
595pub struct ProjectSetNode {
596    #[prost(message, repeated, tag = "1")]
597    pub select_list: ::prost::alloc::vec::Vec<super::expr::ProjectSetSelectItem>,
598}
599#[derive(prost_helpers::AnyPB)]
600#[derive(Clone, PartialEq, ::prost::Message)]
601pub struct SortAggNode {
602    #[prost(message, repeated, tag = "1")]
603    pub group_key: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
604    #[prost(message, repeated, tag = "2")]
605    pub agg_calls: ::prost::alloc::vec::Vec<super::expr::AggCall>,
606}
607#[derive(prost_helpers::AnyPB)]
608#[derive(Clone, PartialEq, ::prost::Message)]
609pub struct HashJoinNode {
610    #[prost(enumeration = "super::plan_common::JoinType", tag = "1")]
611    pub join_type: i32,
612    #[prost(int32, repeated, tag = "2")]
613    pub left_key: ::prost::alloc::vec::Vec<i32>,
614    #[prost(int32, repeated, tag = "3")]
615    pub right_key: ::prost::alloc::vec::Vec<i32>,
616    #[prost(message, optional, tag = "4")]
617    pub condition: ::core::option::Option<super::expr::ExprNode>,
618    #[prost(uint32, repeated, tag = "5")]
619    pub output_indices: ::prost::alloc::vec::Vec<u32>,
620    /// Null safe means it treats `null = null` as true.
621    /// Each key pair can be null safe independently. (left_key, right_key, null_safe)
622    #[prost(bool, repeated, tag = "6")]
623    pub null_safe: ::prost::alloc::vec::Vec<bool>,
624    #[prost(message, optional, tag = "7")]
625    pub asof_desc: ::core::option::Option<super::plan_common::AsOfJoinDesc>,
626}
627#[derive(prost_helpers::AnyPB)]
628#[derive(Clone, PartialEq, ::prost::Message)]
629pub struct SortMergeJoinNode {
630    #[prost(enumeration = "super::plan_common::JoinType", tag = "1")]
631    pub join_type: i32,
632    #[prost(int32, repeated, tag = "2")]
633    pub left_key: ::prost::alloc::vec::Vec<i32>,
634    #[prost(int32, repeated, tag = "3")]
635    pub right_key: ::prost::alloc::vec::Vec<i32>,
636    #[prost(enumeration = "super::common::Direction", tag = "4")]
637    pub direction: i32,
638    #[prost(uint32, repeated, tag = "5")]
639    pub output_indices: ::prost::alloc::vec::Vec<u32>,
640}
641#[derive(prost_helpers::AnyPB)]
642#[derive(Clone, PartialEq, ::prost::Message)]
643pub struct HopWindowNode {
644    #[prost(uint32, tag = "1")]
645    pub time_col: u32,
646    #[prost(message, optional, tag = "2")]
647    pub window_slide: ::core::option::Option<super::data::Interval>,
648    #[prost(message, optional, tag = "3")]
649    pub window_size: ::core::option::Option<super::data::Interval>,
650    #[prost(uint32, repeated, tag = "4")]
651    pub output_indices: ::prost::alloc::vec::Vec<u32>,
652    #[prost(message, repeated, tag = "5")]
653    pub window_start_exprs: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
654    #[prost(message, repeated, tag = "6")]
655    pub window_end_exprs: ::prost::alloc::vec::Vec<super::expr::ExprNode>,
656}
657#[derive(prost_helpers::AnyPB)]
658#[derive(Clone, PartialEq, ::prost::Message)]
659pub struct TableFunctionNode {
660    #[prost(message, optional, tag = "1")]
661    pub table_function: ::core::option::Option<super::expr::TableFunction>,
662}
663/// Task is a running instance of Stage.
664#[derive(prost_helpers::AnyPB)]
665#[derive(Clone, PartialEq, ::prost::Message)]
666pub struct TaskId {
667    #[prost(string, tag = "1")]
668    pub query_id: ::prost::alloc::string::String,
669    #[prost(uint32, tag = "2")]
670    pub stage_id: u32,
671    #[prost(uint64, tag = "3")]
672    pub task_id: u64,
673}
674/// Every task will create N buffers (channels) for parent operators to fetch results from,
675/// where N is the parallelism of parent stage.
676#[derive(prost_helpers::AnyPB)]
677#[derive(Clone, PartialEq, ::prost::Message)]
678pub struct TaskOutputId {
679    #[prost(message, optional, tag = "1")]
680    pub task_id: ::core::option::Option<TaskId>,
681    /// The id of output channel to fetch from
682    #[prost(uint64, tag = "2")]
683    pub output_id: u64,
684}
685#[derive(prost_helpers::AnyPB)]
686#[derive(Clone, PartialEq, ::prost::Message)]
687pub struct LocalExecutePlan {
688    #[prost(message, optional, tag = "1")]
689    pub plan: ::core::option::Option<PlanFragment>,
690    #[prost(message, optional, tag = "2")]
691    pub epoch: ::core::option::Option<super::common::BatchQueryEpoch>,
692    #[prost(map = "string, string", tag = "3")]
693    pub tracing_context: ::std::collections::HashMap<
694        ::prost::alloc::string::String,
695        ::prost::alloc::string::String,
696    >,
697}
698/// ExchangeSource describes where to read results from children operators
699#[derive(prost_helpers::AnyPB)]
700#[derive(Clone, PartialEq, ::prost::Message)]
701pub struct ExchangeSource {
702    #[prost(message, optional, tag = "1")]
703    pub task_output_id: ::core::option::Option<TaskOutputId>,
704    #[prost(message, optional, tag = "2")]
705    pub host: ::core::option::Option<super::common::HostAddress>,
706    #[prost(oneof = "exchange_source::LocalExecutePlan", tags = "3")]
707    pub local_execute_plan: ::core::option::Option<exchange_source::LocalExecutePlan>,
708}
709/// Nested message and enum types in `ExchangeSource`.
710pub mod exchange_source {
711    #[derive(prost_helpers::AnyPB)]
712    #[derive(Clone, PartialEq, ::prost::Oneof)]
713    pub enum LocalExecutePlan {
714        #[prost(message, tag = "3")]
715        Plan(super::LocalExecutePlan),
716    }
717}
718#[derive(prost_helpers::AnyPB)]
719#[derive(Clone, PartialEq, ::prost::Message)]
720pub struct ExchangeNode {
721    #[prost(message, repeated, tag = "1")]
722    pub sources: ::prost::alloc::vec::Vec<ExchangeSource>,
723    /// sequential means each tasks of the exchange node will be executed sequentially.
724    #[prost(bool, tag = "2")]
725    pub sequential: bool,
726    #[prost(message, repeated, tag = "3")]
727    pub input_schema: ::prost::alloc::vec::Vec<super::plan_common::Field>,
728}
729#[derive(prost_helpers::AnyPB)]
730#[derive(Clone, PartialEq, ::prost::Message)]
731pub struct MergeSortExchangeNode {
732    #[prost(message, optional, tag = "1")]
733    pub exchange: ::core::option::Option<ExchangeNode>,
734    #[prost(message, repeated, tag = "2")]
735    pub column_orders: ::prost::alloc::vec::Vec<super::common::ColumnOrder>,
736}
737#[derive(prost_helpers::AnyPB)]
738#[derive(Clone, PartialEq, ::prost::Message)]
739pub struct LocalLookupJoinNode {
740    #[prost(enumeration = "super::plan_common::JoinType", tag = "1")]
741    pub join_type: i32,
742    #[prost(message, optional, tag = "2")]
743    pub condition: ::core::option::Option<super::expr::ExprNode>,
744    #[prost(uint32, repeated, tag = "3")]
745    pub outer_side_key: ::prost::alloc::vec::Vec<u32>,
746    #[prost(uint32, repeated, tag = "4")]
747    pub inner_side_key: ::prost::alloc::vec::Vec<u32>,
748    #[prost(uint32, tag = "5")]
749    pub lookup_prefix_len: u32,
750    #[prost(message, optional, tag = "6")]
751    pub inner_side_table_desc: ::core::option::Option<
752        super::plan_common::StorageTableDesc,
753    >,
754    #[prost(uint64, repeated, tag = "7")]
755    pub inner_side_vnode_mapping: ::prost::alloc::vec::Vec<u64>,
756    #[prost(int32, repeated, tag = "8")]
757    pub inner_side_column_ids: ::prost::alloc::vec::Vec<i32>,
758    #[prost(uint32, repeated, tag = "9")]
759    pub output_indices: ::prost::alloc::vec::Vec<u32>,
760    #[prost(message, repeated, tag = "10")]
761    pub worker_nodes: ::prost::alloc::vec::Vec<super::common::WorkerNode>,
762    /// Null safe means it treats `null = null` as true.
763    /// Each key pair can be null safe independently. (left_key, right_key, null_safe)
764    #[prost(bool, repeated, tag = "11")]
765    pub null_safe: ::prost::alloc::vec::Vec<bool>,
766    #[prost(message, optional, tag = "12")]
767    pub as_of: ::core::option::Option<super::plan_common::AsOf>,
768    #[prost(message, optional, tag = "13")]
769    pub asof_desc: ::core::option::Option<super::plan_common::AsOfJoinDesc>,
770}
771/// RFC: A new schedule way for distributed lookup join
772/// <https://github.com/risingwavelabs/rfcs/pull/6>
773#[derive(prost_helpers::AnyPB)]
774#[derive(Clone, PartialEq, ::prost::Message)]
775pub struct DistributedLookupJoinNode {
776    #[prost(enumeration = "super::plan_common::JoinType", tag = "1")]
777    pub join_type: i32,
778    #[prost(message, optional, tag = "2")]
779    pub condition: ::core::option::Option<super::expr::ExprNode>,
780    #[prost(uint32, repeated, tag = "3")]
781    pub outer_side_key: ::prost::alloc::vec::Vec<u32>,
782    #[prost(uint32, repeated, tag = "4")]
783    pub inner_side_key: ::prost::alloc::vec::Vec<u32>,
784    #[prost(uint32, tag = "5")]
785    pub lookup_prefix_len: u32,
786    #[prost(message, optional, tag = "6")]
787    pub inner_side_table_desc: ::core::option::Option<
788        super::plan_common::StorageTableDesc,
789    >,
790    #[prost(int32, repeated, tag = "7")]
791    pub inner_side_column_ids: ::prost::alloc::vec::Vec<i32>,
792    #[prost(uint32, repeated, tag = "8")]
793    pub output_indices: ::prost::alloc::vec::Vec<u32>,
794    /// Null safe means it treats `null = null` as true.
795    /// Each key pair can be null safe independently. (left_key, right_key, null_safe)
796    #[prost(bool, repeated, tag = "9")]
797    pub null_safe: ::prost::alloc::vec::Vec<bool>,
798    #[prost(message, optional, tag = "10")]
799    pub as_of: ::core::option::Option<super::plan_common::AsOf>,
800    #[prost(message, optional, tag = "11")]
801    pub asof_desc: ::core::option::Option<super::plan_common::AsOfJoinDesc>,
802}
803#[derive(prost_helpers::AnyPB)]
804#[derive(Clone, Copy, PartialEq, ::prost::Message)]
805pub struct UnionNode {}
806#[derive(prost_helpers::AnyPB)]
807#[derive(Clone, PartialEq, ::prost::Message)]
808pub struct SortOverWindowNode {
809    #[prost(message, repeated, tag = "1")]
810    pub calls: ::prost::alloc::vec::Vec<super::expr::WindowFunction>,
811    #[prost(uint32, repeated, tag = "2")]
812    pub partition_by: ::prost::alloc::vec::Vec<u32>,
813    #[prost(message, repeated, tag = "3")]
814    pub order_by: ::prost::alloc::vec::Vec<super::common::ColumnOrder>,
815}
816#[derive(prost_helpers::AnyPB)]
817#[derive(Clone, Copy, PartialEq, ::prost::Message)]
818pub struct MaxOneRowNode {}
819#[derive(prost_helpers::AnyPB)]
820#[derive(Clone, PartialEq, ::prost::Message)]
821pub struct PlanNode {
822    #[prost(message, repeated, tag = "1")]
823    pub children: ::prost::alloc::vec::Vec<PlanNode>,
824    #[prost(string, tag = "24")]
825    pub identity: ::prost::alloc::string::String,
826    #[prost(
827        oneof = "plan_node::NodeBody",
828        tags = "2, 3, 4, 5, 7, 8, 9, 10, 11, 14, 15, 16, 17, 18, 19, 21, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 100, 101"
829    )]
830    pub node_body: ::core::option::Option<plan_node::NodeBody>,
831}
832/// Nested message and enum types in `PlanNode`.
833pub mod plan_node {
834    #[derive(prost_helpers::AnyPB)]
835    #[derive(::enum_as_inner::EnumAsInner, ::strum::Display, ::strum::EnumDiscriminants)]
836    #[derive(Clone, PartialEq, ::prost::Oneof)]
837    pub enum NodeBody {
838        #[prost(message, tag = "2")]
839        Insert(super::InsertNode),
840        #[prost(message, tag = "3")]
841        Delete(super::DeleteNode),
842        #[prost(message, tag = "4")]
843        Update(super::UpdateNode),
844        #[prost(message, tag = "5")]
845        Project(super::ProjectNode),
846        #[prost(message, tag = "7")]
847        HashAgg(super::HashAggNode),
848        #[prost(message, tag = "8")]
849        Filter(super::FilterNode),
850        #[prost(message, tag = "9")]
851        Exchange(super::ExchangeNode),
852        #[prost(message, tag = "10")]
853        Sort(super::SortNode),
854        #[prost(message, tag = "11")]
855        NestedLoopJoin(super::NestedLoopJoinNode),
856        #[prost(message, tag = "14")]
857        TopN(super::TopNNode),
858        #[prost(message, tag = "15")]
859        SortAgg(super::SortAggNode),
860        #[prost(message, tag = "16")]
861        RowSeqScan(super::RowSeqScanNode),
862        #[prost(message, tag = "17")]
863        Limit(super::LimitNode),
864        #[prost(message, tag = "18")]
865        Values(super::ValuesNode),
866        #[prost(message, tag = "19")]
867        HashJoin(super::HashJoinNode),
868        #[prost(message, tag = "21")]
869        MergeSortExchange(super::MergeSortExchangeNode),
870        #[prost(message, tag = "25")]
871        HopWindow(super::HopWindowNode),
872        #[prost(message, tag = "26")]
873        TableFunction(super::TableFunctionNode),
874        #[prost(message, tag = "27")]
875        SysRowSeqScan(super::SysRowSeqScanNode),
876        #[prost(message, tag = "28")]
877        Expand(super::ExpandNode),
878        #[prost(message, tag = "29")]
879        LocalLookupJoin(super::LocalLookupJoinNode),
880        #[prost(message, tag = "30")]
881        ProjectSet(super::ProjectSetNode),
882        #[prost(message, tag = "31")]
883        Union(super::UnionNode),
884        #[prost(message, tag = "32")]
885        GroupTopN(super::GroupTopNNode),
886        #[prost(message, tag = "33")]
887        DistributedLookupJoin(super::DistributedLookupJoinNode),
888        #[prost(message, tag = "34")]
889        Source(super::SourceNode),
890        #[prost(message, tag = "35")]
891        SortOverWindow(super::SortOverWindowNode),
892        #[prost(message, tag = "36")]
893        MaxOneRow(super::MaxOneRowNode),
894        #[prost(message, tag = "37")]
895        LogRowSeqScan(super::LogRowSeqScanNode),
896        #[prost(message, tag = "38")]
897        FileScan(super::FileScanNode),
898        #[prost(message, tag = "39")]
899        IcebergScan(super::IcebergScanNode),
900        #[prost(message, tag = "40")]
901        PostgresQuery(super::PostgresQueryNode),
902        #[prost(message, tag = "41")]
903        MysqlQuery(super::MySqlQueryNode),
904        #[prost(message, tag = "42")]
905        GcsFileScan(super::GcsFileScanNode),
906        #[prost(message, tag = "43")]
907        AzblobFileScan(super::AzblobFileScanNode),
908        /// The following nodes are used for testing.
909        #[prost(bool, tag = "100")]
910        BlockExecutor(bool),
911        #[prost(bool, tag = "101")]
912        BusyLoopExecutor(bool),
913    }
914}
915/// ExchangeInfo determines how to distribute results to tasks of next stage.
916///
917/// Note that the fragment itself does not know the where are the receivers. Instead, it prepares results in
918/// N buffers and wait for parent operators (`Exchange` nodes) to pull data from a specified buffer
919#[derive(prost_helpers::AnyPB)]
920#[derive(Clone, PartialEq, ::prost::Message)]
921pub struct ExchangeInfo {
922    #[prost(enumeration = "exchange_info::DistributionMode", tag = "1")]
923    pub mode: i32,
924    #[prost(oneof = "exchange_info::Distribution", tags = "2, 3, 4")]
925    pub distribution: ::core::option::Option<exchange_info::Distribution>,
926}
927/// Nested message and enum types in `ExchangeInfo`.
928pub mod exchange_info {
929    #[derive(prost_helpers::AnyPB)]
930    #[derive(Clone, Copy, PartialEq, ::prost::Message)]
931    pub struct BroadcastInfo {
932        #[prost(uint32, tag = "1")]
933        pub count: u32,
934    }
935    #[derive(prost_helpers::AnyPB)]
936    #[derive(Clone, PartialEq, ::prost::Message)]
937    pub struct HashInfo {
938        #[prost(uint32, tag = "1")]
939        pub output_count: u32,
940        #[prost(uint32, repeated, tag = "3")]
941        pub key: ::prost::alloc::vec::Vec<u32>,
942    }
943    #[derive(prost_helpers::AnyPB)]
944    #[derive(Clone, PartialEq, ::prost::Message)]
945    pub struct ConsistentHashInfo {
946        /// `vmap` maps virtual node to down stream task id
947        #[prost(uint32, repeated, tag = "1")]
948        pub vmap: ::prost::alloc::vec::Vec<u32>,
949        #[prost(uint32, repeated, tag = "2")]
950        pub key: ::prost::alloc::vec::Vec<u32>,
951    }
952    #[derive(prost_helpers::AnyPB)]
953    #[derive(
954        Clone,
955        Copy,
956        Debug,
957        PartialEq,
958        Eq,
959        Hash,
960        PartialOrd,
961        Ord,
962        ::prost::Enumeration
963    )]
964    #[repr(i32)]
965    pub enum DistributionMode {
966        /// No partitioning at all, used for root segment which aggregates query results
967        Unspecified = 0,
968        Single = 1,
969        Broadcast = 2,
970        Hash = 3,
971        ConsistentHash = 4,
972    }
973    impl DistributionMode {
974        /// String value of the enum field names used in the ProtoBuf definition.
975        ///
976        /// The values are not transformed in any way and thus are considered stable
977        /// (if the ProtoBuf definition does not change) and safe for programmatic use.
978        pub fn as_str_name(&self) -> &'static str {
979            match self {
980                Self::Unspecified => "UNSPECIFIED",
981                Self::Single => "SINGLE",
982                Self::Broadcast => "BROADCAST",
983                Self::Hash => "HASH",
984                Self::ConsistentHash => "CONSISTENT_HASH",
985            }
986        }
987        /// Creates an enum from field names used in the ProtoBuf definition.
988        pub fn from_str_name(value: &str) -> ::core::option::Option<Self> {
989            match value {
990                "UNSPECIFIED" => Some(Self::Unspecified),
991                "SINGLE" => Some(Self::Single),
992                "BROADCAST" => Some(Self::Broadcast),
993                "HASH" => Some(Self::Hash),
994                "CONSISTENT_HASH" => Some(Self::ConsistentHash),
995                _ => None,
996            }
997        }
998    }
999    #[derive(prost_helpers::AnyPB)]
1000    #[derive(Clone, PartialEq, ::prost::Oneof)]
1001    pub enum Distribution {
1002        #[prost(message, tag = "2")]
1003        BroadcastInfo(BroadcastInfo),
1004        #[prost(message, tag = "3")]
1005        HashInfo(HashInfo),
1006        #[prost(message, tag = "4")]
1007        ConsistentHashInfo(ConsistentHashInfo),
1008    }
1009}
1010#[derive(prost_helpers::AnyPB)]
1011#[derive(Clone, PartialEq, ::prost::Message)]
1012pub struct PlanFragment {
1013    #[prost(message, optional, tag = "1")]
1014    pub root: ::core::option::Option<PlanNode>,
1015    #[prost(message, optional, tag = "2")]
1016    pub exchange_info: ::core::option::Option<ExchangeInfo>,
1017}
risingwave_pb/batch_plan.rs

risingwave_pb/
batch_plan.rs
