iridium_core 0.1.4

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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use super::super::virtual_table_def;
use super::super::VirtualTable;
use crate::catalog::Catalog;
use crate::executor::context::ExecutionContext;
use crate::storage::StoredRow;
use crate::types::{DataType, Value};

pub(crate) struct SysIndexes;

impl VirtualTable for SysIndexes {
    fn definition(&self) -> crate::catalog::TableDef {
        virtual_table_def(
            "indexes",
            vec![
                ("object_id", DataType::Int, false),
                ("index_id", DataType::Int, false),
                ("name", DataType::VarChar { max_len: 128 }, false),
                ("type", DataType::TinyInt, false),
                ("type_desc", DataType::VarChar { max_len: 60 }, false),
                ("is_unique", DataType::Bit, false),
                ("data_space_id", DataType::Int, false),
                ("is_primary_key", DataType::Bit, false),
                ("is_unique_constraint", DataType::Bit, false),
                ("is_hypothetical", DataType::Bit, false),
                ("is_disabled", DataType::Bit, false),
                ("has_filter", DataType::Bit, false),
            ],
        )
    }

    fn rows(&self, catalog: &dyn Catalog, _ctx: &ExecutionContext) -> Vec<StoredRow> {
        let tables = catalog.get_tables();
        catalog
            .get_indexes()
            .iter()
            .map(|idx| {
                let table = tables.iter().find(|t| t.id == idx.table_id);
                let is_primary_key = table
                    .map(|t| {
                        idx.column_ids.iter().all(|&col_id| {
                            t.columns
                                .iter()
                                .find(|c| c.id == col_id)
                                .map(|c| c.primary_key)
                                .unwrap_or(false)
                        }) && t.columns.iter().filter(|c| c.primary_key).count()
                            == idx.column_ids.len()
                    })
                    .unwrap_or(false);

                let is_unique_constraint = table
                    .map(|t| {
                        idx.column_ids.iter().all(|&col_id| {
                            t.columns
                                .iter()
                                .find(|c| c.id == col_id)
                                .map(|c| c.unique)
                                .unwrap_or(false)
                        }) && t.columns.iter().filter(|c| c.unique).count() == idx.column_ids.len()
                    })
                    .unwrap_or(false);

                StoredRow {
                    values: vec![
                        Value::Int(idx.table_id as i32),
                        Value::Int(idx.id as i32),
                        Value::VarChar(idx.name.clone()),
                        Value::TinyInt(if idx.is_clustered { 1 } else { 2 }),
                        Value::VarChar(if idx.is_clustered {
                            "CLUSTERED".to_string()
                        } else {
                            "NONCLUSTERED".to_string()
                        }),
                        Value::Bit(idx.is_unique),
                        Value::Int(1),
                        Value::Bit(is_primary_key),
                        Value::Bit(is_unique_constraint),
                        Value::Bit(false),
                        Value::Bit(false),
                        Value::Bit(false),
                    ],
                    deleted: false,
                }
            })
            .collect()
    }
}