implement get_attribute

Author: xx01cyx

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

@@ -49,3 +49,33 @@ impl Display for EpochId {
         write!(f, "Epoch#{}", self.0)
     }
 }
+
+impl From<GroupId> for i32 {
+    fn from(id: GroupId) -> i32 {
+        id.0 as i32
+    }
+}
+
+impl From<ExprId> for i32 {
+    fn from(id: ExprId) -> i32 {
+        id.0 as i32
+    }
+}
+
+impl From<TableId> for i32 {
+    fn from(id: TableId) -> i32 {
+        id.0 as i32
+    }
+}
+
+impl From<AttrId> for i32 {
+    fn from(id: AttrId) -> i32 {
+        id.0 as i32
+    }
+}
+
+impl From<EpochId> for i32 {
+    fn from(id: EpochId) -> i32 {
+        id.0 as i32
+    }
+}

optd-cost-model/src/lib.rs

@@ -40,7 +40,7 @@ pub enum SemanticError {
     // TODO: Add more error types
     UnknownStatisticType,
     VersionedStatisticNotFound,
-    AttributeNotFound,
+    AttributeNotFound(TableId, i32), // (table_id, attribute_base_index)
 }
 
 #[derive(Debug)]
@@ -50,6 +50,12 @@ pub enum CostModelError {
     SemanticError(SemanticError),
 }
 
+impl From<BackendError> for CostModelError {
+    fn from(err: BackendError) -> Self {
+        CostModelError::ORMError(err)
+    }
+}
+
 pub trait CostModel: 'static + Send + Sync {
     /// TODO: documentation
     fn compute_operation_cost(

optd-cost-model/src/storage.rs

@@ -1,16 +1,52 @@
+#![allow(unused_variables)]
 use std::sync::Arc;
 
-use optd_persistent::CostModelStorageLayer;
+use optd_persistent::{cost_model::interface::Attr, BackendManager, CostModelStorageLayer};
+
+use crate::{
+    common::types::TableId, stats::AttributeCombValueStats, CostModelError, CostModelResult,
+    SemanticError,
+};
 
 /// TODO: documentation
 pub struct CostModelStorageManager<S: CostModelStorageLayer> {
     pub backend_manager: Arc<S>,
     // TODO: in-memory cache
 }
 
-impl<S: CostModelStorageLayer> CostModelStorageManager<S> {
-    /// TODO: documentation
-    pub fn new(backend_manager: Arc<S>) -> Self {
+impl CostModelStorageManager<BackendManager> {
+    pub fn new(backend_manager: Arc<BackendManager>) -> Self {
         Self { backend_manager }
     }
+
+    /// TODO: documentation
+    /// TODO: if we have memory cache,
+    /// we should add the reference. (&Field)
+    pub async fn get_attribute_info(
+        &self,
+        table_id: TableId,
+        attribute_base_index: i32,
+    ) -> CostModelResult<Attr> {
+        let attr = self
+            .backend_manager
+            .get_attribute(table_id.into(), attribute_base_index)
+            .await?;
+        attr.ok_or_else(|| {
+            CostModelError::SemanticError(SemanticError::AttributeNotFound(
+                table_id,
+                attribute_base_index,
+            ))
+        })
+    }
+
+    /// TODO: documentation
+    /// TODO: if we have memory cache,
+    /// we should add the reference. (&AttributeCombValueStats)
+    pub fn get_attributes_comb_statistics(
+        &self,
+        table_id: TableId,
+        attr_comb: &[usize],
+    ) -> CostModelResult<Option<AttributeCombValueStats>> {
+        todo!()
+    }
 }

optd-persistent/src/cost_model/interface.rs

@@ -1,6 +1,5 @@
 #![allow(dead_code, unused_imports)]
 
-use crate::entities::attribute::Model as AttributeEntity;
 use crate::entities::cascades_group;
 use crate::entities::logical_expression;
 use crate::entities::physical_expression;
@@ -83,6 +82,16 @@ pub struct Cost {
     pub estimated_statistic: i32,
 }
 
+#[derive(Clone, Debug)]
+pub struct Attr {
+    pub table_id: i32,
+    pub name: String,
+    pub compression_method: String,
+    pub attr_type: i32,
+    pub base_index: i32,
+    pub nullable: bool,
+}
+
 /// TODO: documentation
 #[trait_variant::make(Send)]
 pub trait CostModelStorageLayer {
@@ -133,4 +142,10 @@ pub trait CostModelStorageLayer {
     ) -> StorageResult<Option<Cost>>;
 
     async fn get_cost(&self, expr_id: ExprId) -> StorageResult<Option<Cost>>;
+
+    async fn get_attribute(
+        &self,
+        table_id: TableId,
+        attribute_base_index: i32,
+    ) -> StorageResult<Option<Attr>>;
 }

optd-persistent/src/cost_model/orm.rs

@@ -13,7 +13,9 @@ use sea_orm::{
 use serde_json::json;
 
 use super::catalog::mock_catalog::{self, MockCatalog};
-use super::interface::{AttrId, CatalogSource, EpochId, EpochOption, ExprId, Stat, StatId};
+use super::interface::{
+    Attr, AttrId, CatalogSource, EpochId, EpochOption, ExprId, Stat, StatId, TableId,
+};
 
 impl BackendManager {
     fn get_description_from_attr_ids(&self, attr_ids: Vec<AttrId>) -> String {
@@ -505,6 +507,26 @@ impl CostModelStorageLayer for BackendManager {
         let _ = PlanCost::insert(new_cost).exec(&self.db).await?;
         Ok(())
     }
+
+    async fn get_attribute(
+        &self,
+        table_id: TableId,
+        attribute_base_index: i32,
+    ) -> StorageResult<Option<Attr>> {
+        Ok(Attribute::find()
+            .filter(attribute::Column::TableId.eq(table_id))
+            .filter(attribute::Column::BaseAttributeNumber.eq(attribute_base_index))
+            .one(&self.db)
+            .await?
+            .map(|attr| Attr {
+                table_id,
+                name: attr.name,
+                compression_method: attr.compression_method,
+                attr_type: attr.variant_tag,
+                base_index: attribute_base_index,
+                nullable: !attr.is_not_null,
+            }))
+    }
 }
 
 #[cfg(test)]