add storage part

Author: lanlou1554

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

@@ -46,8 +46,8 @@ impl std::fmt::Display for PhysicalNodeType {
 pub enum PredicateType {
     List,
     Constant(ConstantType),
-    ColumnRef,
-    ExternColumnRef,
+    AttributeRef,
+    ExternAttributeRef,
     UnOp(UnOpType),
     BinOp(BinOpType),
     LogOp(LogOpType),

optd-cost-model/src/cost_model.rs

@@ -12,18 +12,22 @@ use crate::{
         nodes::{ArcPredicateNode, PhysicalNodeType},
         types::ExprId,
     },
+    storage::CostModelStorageManager,
     ComputeCostContext, CostModel, CostModelResult, EstimatedStatistic,
 };
 
 pub struct CostModelImpl<CMSL: CostModelStorageLayer> {
-    backend_manager: Arc<CMSL>,
+    storage_manager: CostModelStorageManager<CMSL>,
     default_catalog_source: CatalogSource,
 }
 
 impl<CMSL: CostModelStorageLayer> CostModelImpl<CMSL> {
-    pub fn new(backend_manager: Arc<CMSL>, default_catalog_source: CatalogSource) -> Self {
+    pub fn new(
+        storage_manager: CostModelStorageManager<CMSL>,
+        default_catalog_source: CatalogSource,
+    ) -> Self {
         Self {
-            backend_manager,
+            storage_manager,
             default_catalog_source,
         }
     }

optd-cost-model/src/lib.rs

@@ -7,6 +7,7 @@ use optd_persistent::cost_model::interface::{Stat, StatType};
 pub mod common;
 pub mod cost;
 pub mod cost_model;
+pub mod storage;
 
 pub enum StatValue {
     Int(i64),

optd-cost-model/src/storage.rs

@@ -0,0 +1,14 @@
+use std::sync::Arc;
+
+use optd_persistent::CostModelStorageLayer;
+
+pub struct CostModelStorageManager<CMSL: CostModelStorageLayer> {
+    pub backend_manager: Arc<CMSL>,
+    // TODO: in-memory cache
+}
+
+impl<CMSL: CostModelStorageLayer> CostModelStorageManager<CMSL> {
+    pub fn new(backend_manager: Arc<CMSL>) -> Self {
+        Self { backend_manager }
+    }
+}