rrdb 0.0.2

ready
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

use std::io::ErrorKind;

use crate::ast::ddl::drop_database::DropDatabaseQuery;
use crate::errors::predule::ExecuteError;
use crate::errors::RRDBError;
use crate::executor::predule::{ExecuteResult, Executor};
use crate::executor::result::{ExecuteColumn, ExecuteColumnType, ExecuteField, ExecuteRow};

impl Executor {
    pub async fn drop_database(
        &self,
        query: DropDatabaseQuery,
    ) -> Result<ExecuteResult, RRDBError> {
        let base_path = self.get_data_directory();
        let mut database_path = base_path.clone();

        let database_name = query
            .database_name
            .clone()
            .ok_or_else(|| ExecuteError::wrap("no database name"))?;

        database_path.push(&database_name);

        if let Err(error) = tokio::fs::remove_dir_all(database_path.clone()).await {
            match error.kind() {
                ErrorKind::NotFound => return Err(ExecuteError::wrap("database not found")),
                _ => {
                    return Err(ExecuteError::wrap("database drop failed"));
                }
            }
        }

        Ok(ExecuteResult {
            columns: (vec![ExecuteColumn {
                name: "desc".into(),
                data_type: ExecuteColumnType::String,
            }]),
            rows: (vec![ExecuteRow {
                fields: vec![ExecuteField::String(format!(
                    "database dropped: {}",
                    database_name
                ))],
            }]),
        })
    }
}