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
88
89
90

use super::super::{numeric_precision, numeric_scale_val, virtual_table_def, VirtualTable};
use crate::ast::RoutineParamType;
use crate::catalog::Catalog;
use crate::executor::context::ExecutionContext;
use crate::executor::type_mapping::data_type_spec_to_runtime;
use crate::storage::StoredRow;
use crate::types::{DataType, Value};

pub(crate) struct SysParameters;

impl VirtualTable for SysParameters {
    fn definition(&self) -> crate::catalog::TableDef {
        virtual_table_def(
            "parameters",
            vec![
                ("object_id", DataType::Int, false),
                ("name", DataType::VarChar { max_len: 128 }, true),
                ("parameter_id", DataType::Int, false),
                ("system_type_id", DataType::TinyInt, false),
                ("user_type_id", DataType::Int, false),
                ("max_length", DataType::SmallInt, false),
                ("precision", DataType::TinyInt, false),
                ("scale", DataType::TinyInt, false),
                ("is_output", DataType::Bit, false),
                ("is_readonly", DataType::Bit, false),
                ("is_xml_document", DataType::Bit, false),
                ("xml_collection_id", DataType::Int, false),
                ("default_value", DataType::VarChar { max_len: 128 }, true),
                ("has_default_value", DataType::Bit, false),
            ],
        )
    }

    fn rows(&self, catalog: &dyn Catalog, _ctx: &ExecutionContext) -> Vec<StoredRow> {
        let mut rows = Vec::new();
        for r in catalog.get_routines() {
            for (i, p) in r.params.iter().enumerate() {
                let (dt, is_user_defined) = match &p.param_type {
                    RoutineParamType::Scalar(dt_spec) => {
                        (data_type_spec_to_runtime(dt_spec), false)
                    }
                    RoutineParamType::TableType(_) => {
                        // For table types, we use a placeholder or look up its base type if applicable.
                        // SQL Server sys.parameters for TVP shows system_type_id = 243 (table type).
                        (DataType::SqlVariant, true) // Simplified
                    }
                };

                let type_id = if is_user_defined {
                    match &p.param_type {
                        RoutineParamType::TableType(obj) => catalog
                            .find_table_type(obj.schema_or_dbo(), &obj.name)
                            .map(|tt| tt.object_id)
                            .unwrap_or(0),
                        _ => 0,
                    }
                } else {
                    super::super::system_type_id(&dt)
                };

                rows.push(StoredRow {
                    values: vec![
                        Value::Int(r.object_id),
                        Value::VarChar(p.name.clone()),
                        Value::Int((i + 1) as i32),
                        Value::TinyInt(if is_user_defined { 243 } else { type_id as u8 }),
                        Value::Int(type_id),
                        Value::SmallInt(super::super::type_max_length(&dt)),
                        match numeric_precision(&dt) {
                            Value::TinyInt(v) => Value::TinyInt(v),
                            _ => Value::TinyInt(0),
                        },
                        match numeric_scale_val(&dt) {
                            Value::Int(v) => Value::TinyInt(v as u8),
                            _ => Value::TinyInt(0),
                        },
                        Value::Bit(p.is_output),
                        Value::Bit(p.is_readonly),
                        Value::Bit(false),
                        Value::Int(0),
                        Value::Null,
                        Value::Bit(p.default.is_some()),
                    ],
                    deleted: false,
                });
            }
        }
        rows
    }
}