feat: impl Drop Index

Author: KKould

docs/features.md

@@ -105,7 +105,8 @@ let kite_sql = DataBaseBuilder::path("./data")
     - [x] View
 - Drop
     - [x] Table
-    - [ ] Index
+    - [x] Index
+      - Tips: `Drop Index table_name.index_name`
     - [x] View
 - Alert
     - [x] Add Column

src/binder/drop_index.rs

@@ -0,0 +1,32 @@
+use crate::binder::{lower_ident, Binder};
+use crate::errors::DatabaseError;
+use crate::planner::operator::drop_index::DropIndexOperator;
+use crate::planner::operator::Operator;
+use crate::planner::{Childrens, LogicalPlan};
+use crate::storage::Transaction;
+use crate::types::value::DataValue;
+use sqlparser::ast::ObjectName;
+use std::sync::Arc;
+
+impl<T: Transaction, A: AsRef<[(&'static str, DataValue)]>> Binder<'_, '_, T, A> {
+    pub(crate) fn bind_drop_index(
+        &mut self,
+        name: &ObjectName,
+        if_exists: &bool,
+    ) -> Result<LogicalPlan, DatabaseError> {
+        let table_name = name.0.first().ok_or(DatabaseError::InvalidIndex)?;
+        let index_name = name.0.get(1).ok_or(DatabaseError::InvalidIndex)?;
+
+        let table_name = Arc::new(lower_ident(table_name));
+        let index_name = lower_ident(index_name);
+
+        Ok(LogicalPlan::new(
+            Operator::DropIndex(DropIndexOperator {
+                table_name,
+                index_name,
+                if_exists: *if_exists,
+            }),
+            Childrens::None,
+        ))
+    }
+}

src/binder/mod.rs

@@ -8,6 +8,7 @@ mod create_view;
 mod delete;
 mod describe;
 mod distinct;
+mod drop_index;
 mod drop_table;
 mod drop_view;
 mod explain;
@@ -375,6 +376,7 @@ impl<'a, 'b, T: Transaction, A: AsRef<[(&'static str, DataValue)]>> Binder<'a, '
                 match object_type {
                     ObjectType::Table => self.bind_drop_table(&names[0], if_exists)?,
                     ObjectType::View => self.bind_drop_view(&names[0], if_exists)?,
+                    ObjectType::Index => self.bind_drop_index(&names[0], if_exists)?,
                     _ => {
                         return Err(DatabaseError::UnsupportedStmt(
                             "only `Table` and `View` are allowed to be Dropped".to_string(),

src/execution/ddl/drop_index.rs

@@ -0,0 +1,43 @@
+use crate::execution::{Executor, WriteExecutor};
+use crate::planner::operator::drop_index::DropIndexOperator;
+use crate::storage::{StatisticsMetaCache, TableCache, Transaction, ViewCache};
+use crate::throw;
+use crate::types::tuple_builder::TupleBuilder;
+
+pub struct DropIndex {
+    op: DropIndexOperator,
+}
+
+impl From<DropIndexOperator> for DropIndex {
+    fn from(op: DropIndexOperator) -> Self {
+        Self { op }
+    }
+}
+
+impl<'a, T: Transaction + 'a> WriteExecutor<'a, T> for DropIndex {
+    fn execute_mut(
+        self,
+        (table_cache, _, _): (&'a TableCache, &'a ViewCache, &'a StatisticsMetaCache),
+        transaction: *mut T,
+    ) -> Executor<'a> {
+        Box::new(
+            #[coroutine]
+            move || {
+                let DropIndexOperator {
+                    table_name,
+                    index_name,
+                    if_exists,
+                } = self.op;
+
+                throw!(unsafe { &mut (*transaction) }.drop_index(
+                    table_cache,
+                    table_name,
+                    &index_name,
+                    if_exists
+                ));
+
+                yield Ok(TupleBuilder::build_result(index_name.to_string()));
+            },
+        )
+    }
+}

src/execution/ddl/mod.rs

@@ -3,6 +3,7 @@ pub(crate) mod create_index;
 pub(crate) mod create_table;
 pub(crate) mod create_view;
 pub(crate) mod drop_column;
+pub(crate) mod drop_index;
 pub(crate) mod drop_table;
 pub(crate) mod drop_view;
 pub(crate) mod truncate;

src/execution/dml/analyze.rs

@@ -19,6 +19,7 @@ use std::fmt::Formatter;
 use std::fs::DirEntry;
 use std::ops::Coroutine;
 use std::ops::CoroutineState;
+use std::path::PathBuf;
 use std::pin::Pin;
 use std::sync::Arc;
 use std::{fmt, fs};
@@ -98,10 +99,7 @@ impl<'a, T: Transaction + 'a> WriteExecutor<'a, T> for Analyze {
                 }
                 drop(coroutine);
                 let mut values = Vec::with_capacity(builders.len());
-                let dir_path = dirs::home_dir()
-                    .expect("Your system does not have a Config directory!")
-                    .join(DEFAULT_STATISTICS_META_PATH)
-                    .join(table_name.as_str());
+                let dir_path = Self::build_statistics_meta_path(&table_name);
                 // For DEBUG
                 // println!("Statistics Path: {:#?}", dir_path);
                 throw!(fs::create_dir_all(&dir_path).map_err(DatabaseError::IO));
@@ -149,6 +147,15 @@ impl<'a, T: Transaction + 'a> WriteExecutor<'a, T> for Analyze {
     }
 }
 
+impl Analyze {
+    pub fn build_statistics_meta_path(table_name: &TableName) -> PathBuf {
+        dirs::home_dir()
+            .expect("Your system does not have a Config directory!")
+            .join(DEFAULT_STATISTICS_META_PATH)
+            .join(table_name.as_str())
+    }
+}
+
 impl fmt::Display for AnalyzeOperator {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
         let indexes = self.index_metas.iter().map(|index| &index.name).join(", ");

src/execution/mod.rs

@@ -10,6 +10,7 @@ use crate::execution::ddl::create_index::CreateIndex;
 use crate::execution::ddl::create_table::CreateTable;
 use crate::execution::ddl::create_view::CreateView;
 use crate::execution::ddl::drop_column::DropColumn;
+use crate::execution::ddl::drop_index::DropIndex;
 use crate::execution::ddl::drop_table::DropTable;
 use crate::execution::ddl::drop_view::DropView;
 use crate::execution::ddl::truncate::Truncate;
@@ -194,6 +195,7 @@ pub fn build_write<'a, T: Transaction + 'a>(
         Operator::CreateView(op) => CreateView::from(op).execute_mut(cache, transaction),
         Operator::DropTable(op) => DropTable::from(op).execute_mut(cache, transaction),
         Operator::DropView(op) => DropView::from(op).execute_mut(cache, transaction),
+        Operator::DropIndex(op) => DropIndex::from(op).execute_mut(cache, transaction),
         Operator::Truncate(op) => Truncate::from(op).execute_mut(cache, transaction),
         Operator::CopyFromFile(op) => CopyFromFile::from(op).execute_mut(cache, transaction),
         Operator::CopyToFile(op) => {

src/optimizer/rule/normalization/column_pruning.rs

@@ -152,6 +152,7 @@ impl ColumnPruning {
             | Operator::CreateView(_)
             | Operator::DropTable(_)
             | Operator::DropView(_)
+            | Operator::DropIndex(_)
             | Operator::Truncate(_)
             | Operator::ShowTable
             | Operator::ShowView

src/optimizer/rule/normalization/compilation_in_advance.rs

@@ -104,6 +104,7 @@ impl ExpressionRemapper {
             | Operator::CreateView(_)
             | Operator::DropTable(_)
             | Operator::DropView(_)
+            | Operator::DropIndex(_)
             | Operator::Truncate(_)
             | Operator::CopyFromFile(_)
             | Operator::CopyToFile(_)
@@ -212,6 +213,7 @@ impl EvaluatorBind {
             | Operator::CreateView(_)
             | Operator::DropTable(_)
             | Operator::DropView(_)
+            | Operator::DropIndex(_)
             | Operator::Truncate(_)
             | Operator::CopyFromFile(_)
             | Operator::CopyToFile(_)

src/planner/mod.rs

@@ -221,6 +221,9 @@ impl LogicalPlan {
             Operator::DropView(_) => SchemaOutput::Schema(vec![ColumnRef::from(
                 ColumnCatalog::new_dummy("DROP VIEW SUCCESS".to_string()),
             )]),
+            Operator::DropIndex(_) => SchemaOutput::Schema(vec![ColumnRef::from(
+                ColumnCatalog::new_dummy("DROP INDEX SUCCESS".to_string()),
+            )]),
             Operator::Truncate(_) => SchemaOutput::Schema(vec![ColumnRef::from(
                 ColumnCatalog::new_dummy("TRUNCATE TABLE SUCCESS".to_string()),
             )]),