First draft of migration

Author: lanlou1554

optd-cost-model/Cargo.lock

@@ -0,0 +1,13 @@
+[package]
+name = "optd-cost-model"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+optd-persistent = { path = "../optd-persistent", version = "0.1" }
+serde = { version = "1.0", features = ["derive"] }
+arrow-schema = "53.2.0"
+datafusion-expr = "32.0.0"
+ordered-float = "4.0"
+chrono = "0.4"
+

optd-cost-model/Cargo.toml

@@ -0,0 +1,4 @@
+pub mod nodes;
+pub mod predicates;
+pub mod types;
+pub mod values;

optd-cost-model/src/common/mod.rs

@@ -0,0 +1,92 @@
+use std::sync::Arc;
+
+use arrow_schema::DataType;
+
+use super::{
+    predicates::{
+        bin_op_pred::BinOpType, constant_pred::ConstantType, func_pred::FuncType,
+        log_op_pred::LogOpType, sort_order_pred::SortOrderType, un_op_pred::UnOpType,
+    },
+    values::Value,
+};
+
+#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
+pub enum JoinType {
+    Inner = 1,
+    FullOuter,
+    LeftOuter,
+    RightOuter,
+    Cross,
+    LeftSemi,
+    RightSemi,
+    LeftAnti,
+    RightAnti,
+}
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
+pub enum PhysicalNodeType {
+    PhysicalProjection,
+    PhysicalFilter,
+    PhysicalScan,
+    PhysicalSort,
+    PhysicalAgg,
+    PhysicalHashJoin(JoinType),
+    PhysicalNestedLoopJoin(JoinType),
+    PhysicalEmptyRelation,
+    PhysicalLimit,
+}
+
+impl std::fmt::Display for PhysicalNodeType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}
+
+#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+pub enum PredicateType {
+    List,
+    Constant(ConstantType),
+    ColumnRef,
+    ExternColumnRef,
+    UnOp(UnOpType),
+    BinOp(BinOpType),
+    LogOp(LogOpType),
+    Func(FuncType),
+    SortOrder(SortOrderType),
+    Between,
+    Cast,
+    Like,
+    DataType(DataType),
+    InList,
+}
+
+impl std::fmt::Display for PredicateType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}
+
+pub type ArcPredicateNode = Arc<PredicateNode>;
+
+#[derive(Clone, Debug, Hash, PartialEq, Eq)]
+pub struct PredicateNode {
+    /// A generic predicate node type
+    pub typ: PredicateType,
+    /// Child predicate nodes, always materialized
+    pub children: Vec<PredicateNode>,
+    /// Data associated with the predicate, if any
+    pub data: Option<Value>,
+}
+
+impl std::fmt::Display for PredicateNode {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "({}", self.typ)?;
+        for child in &self.children {
+            write!(f, " {}", child)?;
+        }
+        if let Some(data) = &self.data {
+            write!(f, " {}", data)?;
+        }
+        write!(f, ")")
+    }
+}

optd-cost-model/src/common/nodes.rs

@@ -0,0 +1,39 @@
+#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
+pub enum BinOpType {
+    // numerical
+    Add,
+    Sub,
+    Mul,
+    Div,
+    Mod,
+
+    // comparison
+    Eq,
+    Neq,
+    Gt,
+    Lt,
+    Geq,
+    Leq,
+}
+
+impl std::fmt::Display for BinOpType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}
+
+impl BinOpType {
+    pub fn is_numerical(&self) -> bool {
+        matches!(
+            self,
+            Self::Add | Self::Sub | Self::Mul | Self::Div | Self::Mod
+        )
+    }
+
+    pub fn is_comparison(&self) -> bool {
+        matches!(
+            self,
+            Self::Eq | Self::Neq | Self::Gt | Self::Lt | Self::Geq | Self::Leq
+        )
+    }
+}

optd-cost-model/src/common/predicates/bin_op_pred.rs

@@ -0,0 +1,20 @@
+use serde::{Deserialize, Serialize};
+
+#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, Serialize, Deserialize)]
+pub enum ConstantType {
+    Bool,
+    Utf8String,
+    UInt8,
+    UInt16,
+    UInt32,
+    UInt64,
+    Int8,
+    Int16,
+    Int32,
+    Int64,
+    Float64,
+    Date,
+    IntervalMonthDateNano,
+    Decimal,
+    Binary,
+}

optd-cost-model/src/common/predicates/constant_pred.rs

@@ -0,0 +1,22 @@
+#[derive(Clone, PartialEq, Eq, Hash, Debug)]
+pub enum FuncType {
+    Scalar(datafusion_expr::BuiltinScalarFunction),
+    Agg(datafusion_expr::AggregateFunction),
+    Case,
+}
+
+impl std::fmt::Display for FuncType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}
+
+impl FuncType {
+    pub fn new_scalar(func_id: datafusion_expr::BuiltinScalarFunction) -> Self {
+        FuncType::Scalar(func_id)
+    }
+
+    pub fn new_agg(func_id: datafusion_expr::AggregateFunction) -> Self {
+        FuncType::Agg(func_id)
+    }
+}

optd-cost-model/src/common/predicates/func_pred.rs

@@ -0,0 +1,13 @@
+use std::fmt::Display;
+
+#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
+pub enum LogOpType {
+    And,
+    Or,
+}
+
+impl Display for LogOpType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}

optd-cost-model/src/common/predicates/log_op_pred.rs

@@ -0,0 +1,6 @@
+pub mod bin_op_pred;
+pub mod constant_pred;
+pub mod func_pred;
+pub mod log_op_pred;
+pub mod sort_order_pred;
+pub mod un_op_pred;

optd-cost-model/src/common/predicates/mod.rs

@@ -0,0 +1,13 @@
+use std::fmt::Display;
+
+#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
+pub enum SortOrderType {
+    Asc,
+    Desc,
+}
+
+impl Display for SortOrderType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}", self)
+    }
+}