optd_core/
rules.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
mod ir;

use std::collections::HashMap;
use std::fmt::{Display, Formatter};
use std::sync::Arc;

use crate::{
    optimizer::Optimizer,
    rel_node::{RelNode, RelNodeTyp},
};

pub use ir::RuleMatcher;

#[derive(Clone, Copy, Debug, PartialEq)]
pub enum OptimizeType {
    Cascades,
    Heuristics,
}

impl Display for OptimizeType {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Cascades => write!(f, "cascades"),
            Self::Heuristics => write!(f, "heuristics"),
        }
    }
}

pub struct RuleWrapper<T: RelNodeTyp, O: Optimizer<T>> {
    pub rule: Arc<dyn Rule<T, O>>,
    pub optimize_type: OptimizeType,
}

impl<T: RelNodeTyp, O: Optimizer<T>> RuleWrapper<T, O> {
    pub fn new(rule: Arc<dyn Rule<T, O>>, optimizer_type: OptimizeType) -> Self {
        Self {
            rule,
            optimize_type: optimizer_type,
        }
    }
    pub fn new_cascades(rule: Arc<dyn Rule<T, O>>) -> Arc<Self> {
        Arc::new(Self {
            rule,
            optimize_type: OptimizeType::Cascades,
        })
    }
    pub fn new_heuristic(rule: Arc<dyn Rule<T, O>>) -> Arc<Self> {
        Arc::new(Self {
            rule,
            optimize_type: OptimizeType::Heuristics,
        })
    }
    pub fn rule(&self) -> Arc<dyn Rule<T, O>> {
        self.rule.clone()
    }
    pub fn optimize_type(&self) -> OptimizeType {
        self.optimize_type
    }
}

pub trait Rule<T: RelNodeTyp, O: Optimizer<T>>: 'static + Send + Sync {
    fn matcher(&self) -> &RuleMatcher<T>;
    fn apply(&self, optimizer: &O, input: HashMap<usize, RelNode<T>>) -> Vec<RelNode<T>>;
    fn name(&self) -> &'static str;
    fn is_impl_rule(&self) -> bool {
        false
    }
}