refactor: enumerate the childrens attribute of LogicalPlan

Author: KKould

Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name          = "fnck_sql"
-version       = "0.0.6"
+version       = "0.0.7"
 edition       = "2021"
 authors       = ["Kould <[email protected]>", "Xwg <[email protected]>"]
 description   = "SQL as a Function for Rust"

benchmarks/query_benchmark.rs

@@ -25,17 +25,20 @@ fn query_cases() -> Vec<(&'static str, &'static str)> {
 }
 
 fn init_fncksql_query_bench() -> Result<(), DatabaseError> {
-    let database = DataBaseBuilder::path(QUERY_BENCH_FNCK_SQL_PATH)
-        .build()?;
-    database.run("create table t1 (c1 int primary key, c2 int)")?.done()?;
+    let database = DataBaseBuilder::path(QUERY_BENCH_FNCK_SQL_PATH).build()?;
+    database
+        .run("create table t1 (c1 int primary key, c2 int)")?
+        .done()?;
     let pb = ProgressBar::new(TABLE_ROW_NUM);
     pb.set_style(
         ProgressStyle::default_bar()
             .template("[{elapsed_precise}] {bar:40.cyan/white} {pos}/{len} {msg}")
             .unwrap(),
     );
     for i in 0..TABLE_ROW_NUM {
-        database.run(format!("insert into t1 values({}, {})", i, i + 1).as_str())?.done()?;
+        database
+            .run(format!("insert into t1 values({}, {})", i, i + 1).as_str())?
+            .done()?;
         pb.set_position(i + 1);
     }
     pb.finish_with_message("Insert completed!");
@@ -106,9 +109,7 @@ fn query_on_execute(c: &mut Criterion) {
         let connection = sqlite::open(QUERY_BENCH_SQLITE_PATH.to_owned()).unwrap();
         c.bench_function(format!("SQLite: {} by '{}'", name, case).as_str(), |b| {
             b.iter(|| {
-                for row in connection
-                    .prepare(case)
-                    .unwrap() {
+                for row in connection.prepare(case).unwrap() {
                     let _ = row.unwrap();
                 }
             })

src/bin/server.rs

@@ -173,10 +173,12 @@ impl SimpleQueryHandler for SessionBackend {
                 let mut guard = self.tx.lock();
 
                 let iter = if let Some(transaction) = guard.as_mut() {
-                    unsafe { transaction.as_mut().run(query) }.map(Box::new) as Result<Box<dyn ResultIter>, _>
+                    unsafe { transaction.as_mut().run(query) }.map(Box::new)
+                        as Result<Box<dyn ResultIter>, _>
                 } else {
                     self.inner.run(query).map(Box::new)
-                }.map_err(|e| PgWireError::ApiError(Box::new(e)))?;
+                }
+                .map_err(|e| PgWireError::ApiError(Box::new(e)))?;
 
                 let mut tuples = Vec::new();
                 for tuple in iter {

src/binder/alter_table.rs

@@ -9,7 +9,7 @@ use crate::planner::operator::alter_table::add_column::AddColumnOperator;
 use crate::planner::operator::alter_table::drop_column::DropColumnOperator;
 use crate::planner::operator::table_scan::TableScanOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 
 impl<T: Transaction> Binder<'_, '_, T> {
@@ -43,7 +43,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                         if_not_exists: *if_not_exists,
                         column,
                     }),
-                    vec![plan],
+                    Childrens::Only(plan),
                 )
             }
             AlterTableOperation::DropColumn {
@@ -60,7 +60,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                         if_exists: *if_exists,
                         column_name,
                     }),
-                    vec![plan],
+                    Childrens::Only(plan),
                 )
             }
             AlterTableOperation::DropPrimaryKey => todo!(),

src/binder/analyze.rs

@@ -3,7 +3,7 @@ use crate::errors::DatabaseError;
 use crate::planner::operator::analyze::AnalyzeOperator;
 use crate::planner::operator::table_scan::TableScanOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 use sqlparser::ast::ObjectName;
 use std::sync::Arc;
@@ -31,7 +31,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                 table_name,
                 index_metas,
             }),
-            vec![scan_op],
+            Childrens::Only(scan_op),
         ))
     }
 }

src/binder/copy.rs

@@ -7,6 +7,7 @@ use crate::errors::DatabaseError;
 use crate::planner::operator::copy_from_file::CopyFromFileOperator;
 use crate::planner::operator::copy_to_file::CopyToFileOperator;
 use crate::planner::operator::Operator;
+use crate::planner::Childrens;
 use fnck_sql_serde_macros::ReferenceSerialization;
 use serde::{Deserialize, Serialize};
 use sqlparser::ast::{CopyOption, CopySource, CopyTarget};
@@ -98,7 +99,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                         target: ext_source,
                         schema_ref,
                     }),
-                    vec![],
+                    Childrens::None,
                 ))
             } else {
                 // COPY <dest_table> FROM <source_file>
@@ -108,7 +109,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                         schema_ref,
                         table: table_name.to_string(),
                     }),
-                    vec![],
+                    Childrens::None,
                 ))
             }
         } else {

src/binder/create_index.rs

@@ -4,7 +4,7 @@ use crate::expression::ScalarExpression;
 use crate::planner::operator::create_index::CreateIndexOperator;
 use crate::planner::operator::table_scan::TableScanOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 use crate::types::index::IndexType;
 use sqlparser::ast::{ObjectName, OrderByExpr};
@@ -60,7 +60,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                 if_not_exists,
                 ty,
             }),
-            vec![plan],
+            Childrens::Only(plan),
         ))
     }
 }

src/binder/create_table.rs

@@ -10,7 +10,7 @@ use crate::errors::DatabaseError;
 use crate::expression::ScalarExpression;
 use crate::planner::operator::create_table::CreateTableOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 use crate::types::LogicalType;
 
@@ -90,7 +90,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                 columns,
                 if_not_exists,
             }),
-            vec![],
+            Childrens::None,
         ))
     }
 

src/binder/create_view.rs

@@ -5,7 +5,7 @@ use crate::errors::DatabaseError;
 use crate::expression::{AliasType, ScalarExpression};
 use crate::planner::operator::create_view::CreateViewOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 use itertools::Itertools;
 use sqlparser::ast::{Ident, ObjectName, Query};
@@ -56,7 +56,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                 },
                 or_replace: *or_replace,
             }),
-            vec![],
+            Childrens::None,
         ))
     }
 }

src/binder/delete.rs

@@ -3,7 +3,7 @@ use crate::errors::DatabaseError;
 use crate::planner::operator::delete::DeleteOperator;
 use crate::planner::operator::table_scan::TableScanOperator;
 use crate::planner::operator::Operator;
-use crate::planner::LogicalPlan;
+use crate::planner::{Childrens, LogicalPlan};
 use crate::storage::Transaction;
 use itertools::Itertools;
 use sqlparser::ast::{Expr, TableAlias, TableFactor, TableWithJoins};
@@ -52,7 +52,7 @@ impl<T: Transaction> Binder<'_, '_, T> {
                     table_name,
                     primary_keys,
                 }),
-                vec![plan],
+                Childrens::Only(plan),
             ))
         } else {
             unreachable!("only table")