uqa_sql/assignment/
routines.rs

1//
2// Unified Query Algebra
3//
4// Copyright (c) 2023-2026 Cognica, Inc.
5//
6
7use crate::{
8    ast::ColumnType, expr::cast_value, type_resolution::canonical_routine_type_name, SQLError,
9};
10use uqa_core::Value;
11pub trait RoutineValueContext: super::AssignmentContext {
12    fn catalog_column_type(&self, name: &str) -> Option<ColumnType>;
13}
14pub fn coerce_routine_value(
15    context: &dyn RoutineValueContext,
16    value: &Value,
17    type_name: &str,
18) -> Result<Value, SQLError> {
19    coerce_routine_value_from(context, value, type_name, None)
20}
21
22pub fn coerce_routine_value_from(
23    context: &dyn RoutineValueContext,
24    value: &Value,
25    type_name: &str,
26    source: Option<&ColumnType>,
27) -> Result<Value, SQLError> {
28    match canonical_routine_type_name(type_name).as_str() {
29        "record" => match value {
30            Value::Record(_) | Value::Null => Ok(value.clone()),
31            Value::Row(values) => Ok(Value::Record(
32                values
33                    .iter()
34                    .cloned()
35                    .enumerate()
36                    .map(|(index, value)| (format!("f{}", index + 1), value))
37                    .collect(),
38            )),
39            _ => Err(SQLError::Routine {
40                sqlstate: "42804".into(),
41                message: "cannot cast non-composite value to type record".into(),
42            }),
43        },
44        "trigger" => match value {
45            Value::Record(_) | Value::Row(_) | Value::Null => Ok(value.clone()),
46            _ => Err(SQLError::Routine {
47                sqlstate: "42804".into(),
48                message: "trigger function must return a row or NULL".into(),
49            }),
50        },
51        "anyarray" => match value {
52            Value::Array(_) | Value::List(_) | Value::Null => Ok(value.clone()),
53            _ => Err(SQLError::Routine {
54                sqlstate: "42804".into(),
55                message: "cannot cast non-array value to type anyarray".into(),
56            }),
57        },
58        "refcursor" => match value {
59            Value::Str(_) | Value::Null => Ok(value.clone()),
60            _ => Err(SQLError::Routine {
61                sqlstate: "42804".into(),
62                message: "cannot cast value to type refcursor".into(),
63            }),
64        },
65        "void" if matches!(value, Value::Null) => Ok(Value::Null),
66        "void" => Err(SQLError::Routine {
67            sqlstate: "42804".into(),
68            message: "cannot cast non-null value to type void".into(),
69        }),
70        _ => {
71            if let Some(target) = context.catalog_column_type(type_name) {
72                return crate::assignment::conversion::coerce_assignment_value(
73                    context,
74                    value.clone(),
75                    &target,
76                    source,
77                );
78            }
79            cast_value(value, type_name)
80        }
81    }
82}
uqa_sql/assignment/routines.rs

uqa_sql/assignment/
routines.rs
