Implement to_column for alias expr

systemxlabs · Feb 6, 2024 · 85cc5b8 · 85cc5b8
1 parent 3be9b63
commit 85cc5b8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/bustubx/src/expression/alias.rs b/bustubx/src/expression/alias.rs
@@ -4,7 +4,6 @@ use crate::common::ScalarValue;
 use crate::error::BustubxResult;
 use crate::expression::{Expr, ExprTrait};
 use crate::storage::Tuple;
-use crate::BustubxError;
 
 #[derive(Clone, PartialEq, Eq, Debug)]
 pub struct Alias {
@@ -26,10 +25,11 @@ impl ExprTrait for Alias {
     }
 
     fn to_column(&self, input_schema: &Schema) -> BustubxResult<Column> {
-        Err(BustubxError::Plan(format!(
-            "expr {:?} as column not supported",
-            self
-        )))
+        Ok(Column::new(
+            self.name.clone(),
+            self.data_type(input_schema)?,
+            self.nullable(input_schema)?,
+        ))
     }
 }