axion-db 0.0.4

Database functionality for the axion framework
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

// axion-db/src/types/postgres.rs
use crate::metadata::AxionDataType;
use crate::types::TypeMapper;

#[derive(Debug, Default, Clone, Copy)]
pub struct PostgresTypeMapper;

// A declarative macro to simplify the large match statement.
macro_rules! map_sql_to_axion {
    (
        $lower_sql_type:expr,
        { $( $sql:pat => $axion:expr ),* $(,)? }
    ) => {
        match $lower_sql_type {
            $( $sql => $axion, )*
            // Fallback for unmapped built-ins
            _ => AxionDataType::Unsupported($lower_sql_type.to_string()),
        }
    };
}

impl TypeMapper for PostgresTypeMapper {
    fn sql_to_axion(&self, sql_type: &str, udt_name: Option<&str>) -> AxionDataType {
        // Handle Array types first, as they are a special case of `data_type`
        if sql_type == "ARRAY" {
            // For arrays, the element type is in the UDT name (e.g., "_int4", "_varchar").
            // We strip the leading underscore to get the base type.
            if let Some(udt) = udt_name.and_then(|u| u.strip_prefix('_')) {
                return AxionDataType::Array(Box::new(self.sql_to_axion(udt, Some(udt))));
            }
        }

        // Handle User-Defined types next (this is for custom enums)
        if sql_type == "USER-DEFINED" {
            if let Some(udt) = udt_name {
                return AxionDataType::Enum(udt.to_string());
            }
        }

        // Handle all other standard types
        let lower_sql_type = sql_type.to_lowercase();

        map_sql_to_axion!(lower_sql_type.as_str(), {
            "uuid" => AxionDataType::Uuid,
            "integer" | "int" | "int4" => AxionDataType::Integer(32),
            "bigint" | "int8" => AxionDataType::Integer(64),
            "smallint" | "int2" => AxionDataType::Integer(16),
            "character varying" | "varchar" | "text" | "name" | "citext" | "char" | "bpchar" => AxionDataType::Text,
            "boolean" | "bool" => AxionDataType::Boolean,
            "date" => AxionDataType::Date,
            "time without time zone" | "time" => AxionDataType::Time,
            "timestamp without time zone" | "timestamp" => AxionDataType::Timestamp,
            "timestamp with time zone" | "timestamptz" => AxionDataType::TimestampTz,
            "numeric" | "decimal" => AxionDataType::Numeric,
            "real" | "float4" => AxionDataType::Float(32),
            "double precision" | "float8" => AxionDataType::Float(64),
            "bytea" => AxionDataType::Bytes,
            "json" => AxionDataType::Json,
            "jsonb" => AxionDataType::JsonB,
            "inet" | "cidr" => AxionDataType::Inet,
        })
    }
}