iridium_core 0.1.9

SQL Server-compatible Rust engine core for Iridium SQL
Documentation 
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 super::*;
use crate::error::DbError;
use crate::executor::string_norm::normalize_identifier;

impl SchemaRegistry for CatalogImpl {
    fn get_schemas(&self) -> &[SchemaDef] {
        &self.schemas
    }

    fn get_schema_id(&self, name: &str) -> Option<u32> {
        let idx = self.schema_map.get(&normalize_identifier(name))?;
        Some(self.schemas[*idx].id)
    }

    fn create_schema(&mut self, name: &str) -> Result<(), DbError> {
        if self.get_schema_id(name).is_some() {
            return Err(DbError::duplicate_schema(name));
        }
        let id = self.alloc_schema_id();
        let idx = self.schemas.len();
        self.schemas.push(SchemaDef {
            id,
            name: name.to_string(),
        });
        self.schema_map.insert(normalize_identifier(name), idx);
        Ok(())
    }

    fn drop_schema(&mut self, name: &str) -> Result<(), DbError> {
        let schema_id = self
            .get_schema_id(name)
            .ok_or_else(|| DbError::schema_not_found(name))?;

        let has_tables = self.tables.iter().any(|t| t.schema_id == schema_id);
        if has_tables {
            return Err(DbError::Execution(format!(
                "schema '{}' cannot be dropped because it contains tables",
                name
            )));
        }

        self.schemas.retain(|s| s.id != schema_id);
        self.rebuild_maps();
        Ok(())
    }
}