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::LegacyVector(_) | Value::List(_) | Value::Null => {
53                Ok(value.clone())
54            }
55            _ => Err(SQLError::Routine {
56                sqlstate: "42804".into(),
57                message: "cannot cast non-array value to type anyarray".into(),
58            }),
59        },
60        "refcursor" => match value {
61            Value::Str(_) | Value::Null => Ok(value.clone()),
62            _ => Err(SQLError::Routine {
63                sqlstate: "42804".into(),
64                message: "cannot cast value to type refcursor".into(),
65            }),
66        },
67        "void" if matches!(value, Value::Null) => Ok(Value::Null),
68        "void" => Err(SQLError::Routine {
69            sqlstate: "42804".into(),
70            message: "cannot cast non-null value to type void".into(),
71        }),
72        _ => {
73            if let Some(target) = context.catalog_column_type(type_name) {
74                return crate::assignment::conversion::coerce_assignment_value(
75                    context,
76                    value.clone(),
77                    &target,
78                    source,
79                );
80            }
81            cast_value(value, type_name)
82        }
83    }
84}
uqa_sql/assignment/routines.rs

uqa_sql/assignment/
routines.rs
