risingwave_frontend/optimizer/rule/
mod.rs1use std::convert::Infallible;
18use std::ops::FromResidual;
19
20use thiserror_ext::AsReport;
21
22use super::PlanRef;
23use crate::error::RwError;
24
25pub enum ApplyResult<T> {
27 Ok(T),
29 NotApplicable,
32 Err(RwError),
35}
36
37impl<T> ApplyResult<T> {
38 pub fn unwrap(self) -> T {
40 match self {
41 ApplyResult::Ok(plan) => plan,
42 ApplyResult::NotApplicable => panic!("unwrap ApplyResult::NotApplicable"),
43 ApplyResult::Err(e) => panic!("unwrap ApplyResult::Err, error: {:?}", e.as_report()),
44 }
45 }
46}
47
48impl<T> FromResidual<Option<Infallible>> for ApplyResult<T> {
50 fn from_residual(residual: Option<Infallible>) -> Self {
51 match residual {
52 Some(i) => match i {},
53 None => Self::NotApplicable,
54 }
55 }
56}
57
58impl<T, E> FromResidual<Result<Infallible, E>> for ApplyResult<T>
60where
61 E: Into<RwError>,
62{
63 fn from_residual(residual: Result<Infallible, E>) -> Self {
64 match residual {
65 Ok(i) => match i {},
66 Err(e) => Self::Err(e.into()),
67 }
68 }
69}
70
71trait InfallibleRule<C: ConventionMarker>: Send + Sync + Description {
75 fn apply(&self, plan: PlanRef<C>) -> Option<PlanRef<C>>;
80}
81
82use InfallibleRule as Rule;
83
84pub trait FallibleRule<C: ConventionMarker>: Send + Sync + Description {
89 fn apply(&self, plan: PlanRef<C>) -> ApplyResult<PlanRef<C>>;
96}
97
98impl<C: ConventionMarker, R> FallibleRule<C> for R
99where
100 R: InfallibleRule<C>,
101{
102 fn apply(&self, plan: PlanRef<C>) -> ApplyResult<PlanRef<C>> {
103 match InfallibleRule::apply(self, plan) {
104 Some(plan) => ApplyResult::Ok(plan),
105 None => ApplyResult::NotApplicable,
106 }
107 }
108}
109
110pub trait Description {
111 fn description(&self) -> &str;
112}
113
114pub(super) type BoxedRule<C> = Box<dyn FallibleRule<C>>;
115
116mod correlated_expr_rewriter;
117mod logical_filter_expression_simplify_rule;
118pub use logical_filter_expression_simplify_rule::*;
119mod over_window_merge_rule;
120pub use over_window_merge_rule::*;
121mod project_join_merge_rule;
122pub use project_join_merge_rule::*;
123mod project_eliminate_rule;
124pub use project_eliminate_rule::*;
125mod project_merge_rule;
126pub use project_merge_rule::*;
127mod pull_up_correlated_predicate_rule;
128pub use pull_up_correlated_predicate_rule::*;
129mod pull_up_correlated_project_value_rule;
130pub use pull_up_correlated_project_value_rule::*;
131mod index_delta_join_rule;
132pub use index_delta_join_rule::*;
133mod left_deep_tree_join_ordering_rule;
134pub use left_deep_tree_join_ordering_rule::*;
135mod apply_agg_transpose_rule;
136pub use apply_agg_transpose_rule::*;
137mod apply_filter_transpose_rule;
138pub use apply_filter_transpose_rule::*;
139mod apply_project_transpose_rule;
140pub use apply_project_transpose_rule::*;
141mod apply_eliminate_rule;
142pub use apply_eliminate_rule::*;
143mod translate_apply_rule;
144pub use translate_apply_rule::*;
145mod merge_multijoin_rule;
146pub use merge_multijoin_rule::*;
147mod max_one_row_eliminate_rule;
148pub use max_one_row_eliminate_rule::*;
149mod apply_join_transpose_rule;
150pub use apply_join_transpose_rule::*;
151mod apply_to_join_rule;
152pub use apply_to_join_rule::*;
153mod distinct_agg_rule;
154pub use distinct_agg_rule::*;
155mod index_selection_rule;
156pub use index_selection_rule::*;
157mod push_calculation_of_join_rule;
158pub use push_calculation_of_join_rule::*;
159mod join_commute_rule;
160mod over_window_to_agg_and_join_rule;
161pub use over_window_to_agg_and_join_rule::*;
162mod over_window_split_rule;
163pub use over_window_split_rule::*;
164mod over_window_to_topn_rule;
165pub use join_commute_rule::*;
166pub use over_window_to_topn_rule::*;
167mod union_to_distinct_rule;
168pub use union_to_distinct_rule::*;
169mod agg_project_merge_rule;
170pub use agg_project_merge_rule::*;
171mod union_merge_rule;
172pub use union_merge_rule::*;
173mod dag_to_tree_rule;
174pub use dag_to_tree_rule::*;
175mod apply_share_eliminate_rule;
176pub use apply_share_eliminate_rule::*;
177mod top_n_on_index_rule;
178pub use top_n_on_index_rule::*;
179mod stream;
180pub use stream::bushy_tree_join_ordering_rule::*;
181pub use stream::filter_with_now_to_join_rule::*;
182pub use stream::generate_series_with_now_rule::*;
183pub use stream::separate_consecutive_join_rule::*;
184pub use stream::split_now_and_rule::*;
185pub use stream::split_now_or_rule::*;
186pub use stream::stream_project_merge_rule::*;
187mod trivial_project_to_values_rule;
188pub use trivial_project_to_values_rule::*;
189mod union_input_values_merge_rule;
190pub use union_input_values_merge_rule::*;
191mod rewrite_like_expr_rule;
192pub use rewrite_like_expr_rule::*;
193mod min_max_on_index_rule;
194pub use min_max_on_index_rule::*;
195mod always_false_filter_rule;
196pub use always_false_filter_rule::*;
197mod join_project_transpose_rule;
198pub use join_project_transpose_rule::*;
199mod limit_push_down_rule;
200pub use limit_push_down_rule::*;
201mod pull_up_hop_rule;
202pub use pull_up_hop_rule::*;
203mod apply_offset_rewriter;
204use apply_offset_rewriter::ApplyOffsetRewriter;
205mod intersect_to_semi_join_rule;
206pub use intersect_to_semi_join_rule::*;
207mod except_to_anti_join_rule;
208pub use except_to_anti_join_rule::*;
209mod intersect_merge_rule;
210pub use intersect_merge_rule::*;
211mod except_merge_rule;
212pub use except_merge_rule::*;
213mod apply_union_transpose_rule;
214pub use apply_union_transpose_rule::*;
215mod apply_dedup_transpose_rule;
216pub use apply_dedup_transpose_rule::*;
217mod project_join_separate_rule;
218pub use project_join_separate_rule::*;
219mod grouping_sets_to_expand_rule;
220pub use grouping_sets_to_expand_rule::*;
221mod apply_project_set_transpose_rule;
222pub use apply_project_set_transpose_rule::*;
223mod cross_join_eliminate_rule;
224pub use cross_join_eliminate_rule::*;
225mod table_function_to_project_set_rule;
226
227pub use table_function_to_project_set_rule::*;
228mod apply_topn_transpose_rule;
229pub use apply_topn_transpose_rule::*;
230mod apply_limit_transpose_rule;
231pub use apply_limit_transpose_rule::*;
232mod batch;
233pub use batch::batch_project_merge_rule::*;
234mod common_sub_expr_extract_rule;
235pub use common_sub_expr_extract_rule::*;
236mod apply_over_window_transpose_rule;
237pub use apply_over_window_transpose_rule::*;
238mod apply_expand_transpose_rule;
239pub use apply_expand_transpose_rule::*;
240mod expand_to_project_rule;
241pub use expand_to_project_rule::*;
242mod agg_group_by_simplify_rule;
243pub use agg_group_by_simplify_rule::*;
244mod apply_hop_window_transpose_rule;
245pub use apply_hop_window_transpose_rule::*;
246mod agg_call_merge_rule;
247pub use agg_call_merge_rule::*;
248mod unify_first_last_value_rule;
249pub use unify_first_last_value_rule::*;
250mod empty_agg_remove_rule;
251pub use empty_agg_remove_rule::*;
252mod add_logstore_rule;
253mod pull_up_correlated_predicate_agg_rule;
254mod source_to_iceberg_scan_rule;
255mod source_to_kafka_scan_rule;
256mod table_function_to_file_scan_rule;
257mod table_function_to_internal_backfill_progress;
258mod table_function_to_internal_source_backfill_progress;
259mod table_function_to_mysql_query_rule;
260mod table_function_to_postgres_query_rule;
261mod top_n_to_vector_search_rule;
262mod values_extract_project_rule;
263
264pub use add_logstore_rule::*;
265pub use batch::batch_iceberg_count_star::*;
266pub use batch::batch_iceberg_predicate_pushdown::*;
267pub use batch::batch_push_limit_to_scan_rule::*;
268pub use pull_up_correlated_predicate_agg_rule::*;
269pub use source_to_iceberg_scan_rule::*;
270pub use source_to_kafka_scan_rule::*;
271pub use table_function_to_file_scan_rule::*;
272pub use table_function_to_internal_backfill_progress::*;
273pub use table_function_to_internal_source_backfill_progress::*;
274pub use table_function_to_mysql_query_rule::*;
275pub use table_function_to_postgres_query_rule::*;
276pub use top_n_to_vector_search_rule::*;
277pub use values_extract_project_rule::*;
278
279use crate::optimizer::plan_node::ConventionMarker;
280
281#[macro_export]
282macro_rules! for_all_rules {
283 ($macro:ident) => {
284 $macro! {
285 { ApplyAggTransposeRule }
286 , { ApplyFilterTransposeRule }
287 , { ApplyProjectTransposeRule }
288 , { ApplyProjectSetTransposeRule }
289 , { ApplyEliminateRule }
290 , { ApplyJoinTransposeRule }
291 , { ApplyShareEliminateRule }
292 , { ApplyToJoinRule }
293 , { MaxOneRowEliminateRule }
294 , { DistinctAggRule }
295 , { IndexDeltaJoinRule }
296 , { MergeMultiJoinRule }
297 , { ProjectEliminateRule }
298 , { ProjectJoinMergeRule }
299 , { ProjectMergeRule }
300 , { PullUpCorrelatedPredicateRule }
301 , { PullUpCorrelatedProjectValueRule }
302 , { LeftDeepTreeJoinOrderingRule }
303 , { TranslateApplyRule }
304 , { PushCalculationOfJoinRule }
305 , { IndexSelectionRule }
306 , { OverWindowToTopNRule }
307 , { OverWindowToAggAndJoinRule }
308 , { OverWindowSplitRule }
309 , { OverWindowMergeRule }
310 , { JoinCommuteRule }
311 , { UnionToDistinctRule }
312 , { AggProjectMergeRule }
313 , { UnionMergeRule }
314 , { DagToTreeRule }
315 , { SplitNowAndRule }
316 , { SplitNowOrRule }
317 , { FilterWithNowToJoinRule }
318 , { GenerateSeriesWithNowRule }
319 , { TopNOnIndexRule }
320 , { TrivialProjectToValuesRule }
321 , { UnionInputValuesMergeRule }
322 , { RewriteLikeExprRule }
323 , { MinMaxOnIndexRule }
324 , { AlwaysFalseFilterRule }
325 , { BushyTreeJoinOrderingRule }
326 , { StreamProjectMergeRule }
327 , { SeparateConsecutiveJoinRule }
328 , { LogicalFilterExpressionSimplifyRule }
329 , { JoinProjectTransposeRule }
330 , { LimitPushDownRule }
331 , { PullUpHopRule }
332 , { IntersectToSemiJoinRule }
333 , { ExceptToAntiJoinRule }
334 , { IntersectMergeRule }
335 , { ExceptMergeRule }
336 , { ApplyUnionTransposeRule }
337 , { ApplyDedupTransposeRule }
338 , { ProjectJoinSeparateRule }
339 , { GroupingSetsToExpandRule }
340 , { CrossJoinEliminateRule }
341 , { ApplyTopNTransposeRule }
342 , { TableFunctionToProjectSetRule }
343 , { TableFunctionToFileScanRule }
344 , { TableFunctionToPostgresQueryRule }
345 , { TableFunctionToMySqlQueryRule }
346 , { TableFunctionToInternalBackfillProgressRule }
347 , { TableFunctionToInternalSourceBackfillProgressRule }
348 , { ApplyLimitTransposeRule }
349 , { CommonSubExprExtractRule }
350 , { BatchProjectMergeRule }
351 , { ApplyOverWindowTransposeRule }
352 , { ApplyExpandTransposeRule }
353 , { ExpandToProjectRule }
354 , { AggGroupBySimplifyRule }
355 , { ApplyHopWindowTransposeRule }
356 , { AggCallMergeRule }
357 , { UnifyFirstLastValueRule }
358 , { ValuesExtractProjectRule }
359 , { BatchPushLimitToScanRule }
360 , { BatchIcebergPredicatePushDownRule }
361 , { BatchIcebergCountStar }
362 , { PullUpCorrelatedPredicateAggRule }
363 , { SourceToKafkaScanRule }
364 , { SourceToIcebergScanRule }
365 , { AddLogstoreRule }
366 , { EmptyAggRemoveRule }
367 , { TopNToVectorSearchRule }
368 }
369 };
370}
371
372macro_rules! impl_description {
373 ($( { $name:ident }),*) => {
374 paste::paste!{
375 $(impl Description for [<$name>] {
376 fn description(&self) -> &str {
377 stringify!([<$name>])
378 }
379 })*
380 }
381 }
382}
383
384for_all_rules! {impl_description}
385
386mod prelude {
387 pub(super) use crate::optimizer::plan_node::{Logical, LogicalPlanRef as PlanRef};
388 pub(super) use crate::optimizer::rule::Rule;
389
390 pub(super) type BoxedRule = crate::optimizer::rule::BoxedRule<Logical>;
391}