uqa-sql 0.3.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
106
107
108
109
110
111
112
113
114
115
116
117
118

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

//! Procedure argument validation, static overload selection, and declared result schemas.

use crate::{
    assignment::conversion::column_type_name,
    ast::ColumnType,
    ir::{analyze_expression_call_arguments, ScalarCallArgument},
    plan::ExpressionPlan,
    routines::{
        declaration::RoutineTypeCatalog,
        resolution::{RoutineCallKind, RoutineOverloadContext},
    },
    RowSchema, SQLError, ScalarExpr,
};
use uqa_core::Value;

/// Reject forbidden subqueries before the caller captures its statement catalog scope.
pub fn validate_call_arguments(arguments: &[ExpressionPlan]) -> Result<(), SQLError> {
    if arguments
        .iter()
        .any(|argument| !argument.subqueries.is_empty())
    {
        return Err(SQLError::Unsupported(
            "cannot use subquery in CALL argument".into(),
        ));
    }
    Ok(())
}

/// Unknown string and NULL literals participate in procedure overload resolution without a concrete type.
pub fn infer_call_argument_types(
    arguments: &[ExpressionPlan],
    decoded: &[ScalarCallArgument<'_>],
    infer: &mut dyn FnMut(&ExpressionPlan) -> Result<Option<ColumnType>, SQLError>,
) -> Result<Vec<Option<ColumnType>>, SQLError> {
    arguments
        .iter()
        .zip(decoded)
        .map(|(argument, call_argument)| {
            if matches!(
                call_argument.value,
                ScalarExpr::Literal(Value::Str(_) | Value::Null)
            ) {
                Ok(None)
            } else {
                infer(argument)
            }
        })
        .collect()
}

/// The syntax metadata needed before the caller captures the scope used for result description.
pub struct ProcedureCallAnalysis<'a> {
    arguments: &'a [ExpressionPlan],
    decoded: Vec<ScalarCallArgument<'a>>,
    names: Vec<Option<String>>,
    explicit_variadic: bool,
}

impl<'a> ProcedureCallAnalysis<'a> {
    pub fn new(arguments: &'a [ExpressionPlan]) -> Result<Self, SQLError> {
        validate_call_arguments(arguments)?;
        let (decoded, explicit_variadic) = analyze_expression_call_arguments(arguments)?;
        let names = decoded
            .iter()
            .map(|argument| argument.name.map(str::to_string))
            .collect();
        Ok(Self {
            arguments,
            decoded,
            names,
            explicit_variadic,
        })
    }

    pub fn result_schema(
        &self,
        name: &str,
        overloads: &RoutineOverloadContext<'_>,
        types: &dyn RoutineTypeCatalog,
        infer: &mut dyn FnMut(&ExpressionPlan) -> Result<Option<ColumnType>, SQLError>,
    ) -> Result<Option<RowSchema>, SQLError> {
        let argument_types = infer_call_argument_types(self.arguments, &self.decoded, infer)?;
        let Some(resolved) = overloads.resolve_static_sql_routine_match(
            name,
            None,
            &self.names,
            &argument_types,
            self.explicit_variadic,
            RoutineCallKind::Procedure,
        )?
        else {
            let signature = argument_types
                .iter()
                .map(|argument| {
                    argument
                        .as_ref()
                        .map_or_else(|| "unknown", column_type_name)
                })
                .collect::<Vec<_>>()
                .join(", ");
            return Err(SQLError::Routine {
                sqlstate: "42883".into(),
                message: format!("procedure {name}({signature}) does not exist"),
            });
        };
        super::invocation::call_output_schema(
            types,
            &resolved.function.def,
            &resolved.invocation.parameter_types,
        )
    }
}