uqa-sql 0.5.0

PostgreSQL-compatible SQL compiler built on libpg_query
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

use super::*;
use crate::routines::{
    routine_signature_types, CompiledFunctionBody, RoutineBody, SQLUserFunction,
};
use std::sync::Arc;

fn routine(sql: &str, identity: u8) -> Arc<SQLUserFunction> {
    let crate::Statement::CreateFunction(mut definition) = crate::compile(sql).unwrap().remove(0)
    else {
        unreachable!("CREATE FUNCTION fixture")
    };
    definition.object_id = Some([identity; 16]);
    Arc::new(SQLUserFunction::new(
        *definition,
        RoutineBody::Bound(Arc::new(CompiledFunctionBody::SQL(Vec::new()))),
    ))
}

fn register(registry: &mut RoutineRegistry, routine: &Arc<SQLUserFunction>) {
    registry
        .entry(routine.def.name.clone())
        .or_default()
        .push(Arc::clone(routine));
}

fn target(routine: &SQLUserFunction) -> RoutineDropTarget {
    RoutineDropTarget {
        object_id: routine.def.object_id,
        name: routine.def.name.clone(),
        argument_types: routine_signature_types(&routine.def),
        is_procedure: routine.def.is_procedure,
    }
}

fn identities(registry: &RoutineRegistry) -> Vec<Option<[u8; 16]>> {
    registry
        .values()
        .flatten()
        .map(|routine| routine.def.object_id)
        .collect()
}

#[test]
fn an_overload_registered_after_the_targets_were_resolved_is_retained() {
    let resolved = routine(
        "CREATE FUNCTION public.f() RETURNS INTEGER LANGUAGE SQL AS 'SELECT 1'",
        1,
    );
    let mut registry = RoutineRegistry::new();
    register(&mut registry, &resolved);
    let targets = [target(&resolved)];
    register(
        &mut registry,
        &routine(
            "CREATE FUNCTION public.f(value INTEGER) RETURNS INTEGER LANGUAGE SQL AS 'SELECT $1'",
            2,
        ),
    );
    remove_routine_registry_targets(&mut registry, &targets).unwrap();
    assert_eq!(identities(&registry), [Some([2; 16])]);
}

#[test]
fn a_removal_whose_target_disappeared_removes_nothing() {
    let first = routine(
        "CREATE FUNCTION public.first() RETURNS INTEGER LANGUAGE SQL AS 'SELECT 1'",
        1,
    );
    let second = routine(
        "CREATE FUNCTION public.second() RETURNS INTEGER LANGUAGE SQL AS 'SELECT 2'",
        2,
    );
    let mut registry = RoutineRegistry::new();
    register(&mut registry, &first);
    let targets = [target(&first), target(&second)];
    let error = remove_routine_registry_targets(&mut registry, &targets).unwrap_err();
    assert!(matches!(error, SQLError::Internal(_)), "{error}");
    assert_eq!(identities(&registry), [Some([1; 16])]);
}

#[test]
fn a_routine_recreated_under_another_identity_is_not_removed() {
    let resolved = routine(
        "CREATE FUNCTION public.f() RETURNS INTEGER LANGUAGE SQL AS 'SELECT 1'",
        1,
    );
    let targets = [target(&resolved)];
    let mut registry = RoutineRegistry::new();
    register(
        &mut registry,
        &routine(
            "CREATE FUNCTION public.f() RETURNS INTEGER LANGUAGE SQL AS 'SELECT 1'",
            3,
        ),
    );
    let error = remove_routine_registry_targets(&mut registry, &targets).unwrap_err();
    assert!(matches!(error, SQLError::Internal(_)), "{error}");
    assert_eq!(identities(&registry), [Some([3; 16])]);
}