Skip to main content

uqa_sql/catalog/index/
output.rs

1//
2// Unified Query Algebra
3//
4// Copyright (c) 2023-2026 Cognica, Inc.
5//
6
7//! Unique-key diagnostics use the default B-tree operator class input type, which can differ from the declared column type.
8
9use crate::{expr::EngineHook, result::format_postgres_text, ColumnType, SQLError};
10use uqa_core::Value;
11
12pub fn format_key_value(
13    value: &Value,
14    ty: &ColumnType,
15    resolver: Option<&dyn EngineHook>,
16) -> Result<String, SQLError> {
17    match ty {
18        ColumnType::Domain { base, .. } => format_key_value(value, base, resolver),
19        ColumnType::Regproc
20        | ColumnType::Regprocedure
21        | ColumnType::Regclass
22        | ColumnType::Regnamespace
23        | ColumnType::Regrole
24        | ColumnType::Regtype => format_postgres_text(value, &ColumnType::Oid, resolver),
25        ColumnType::Int2Vector => format_postgres_text(
26            value,
27            &ColumnType::Array(Box::new(ColumnType::SmallInteger)),
28            resolver,
29        ),
30        _ => format_postgres_text(value, ty, resolver),
31    }
32}
33
34#[cfg(test)]
35mod tests {
36    use super::*;
37    use uqa_core::{LegacyVectorKind, LegacyVectorValue};
38
39    #[test]
40    fn default_btree_input_types_preserve_alias_and_vector_output() {
41        assert_eq!(
42            format_key_value(&Value::Int(42), &ColumnType::Regclass, None).unwrap(),
43            "42"
44        );
45        for (kind, ty, expected) in [
46            (
47                LegacyVectorKind::SmallInteger,
48                ColumnType::Int2Vector,
49                "[0:1]={1,2}",
50            ),
51            (LegacyVectorKind::Oid, ColumnType::OidVector, "1 2"),
52        ] {
53            let value = Value::LegacyVector(
54                LegacyVectorValue::try_new(kind, vec![Value::Int(1), Value::Int(2)]).unwrap(),
55            );
56            assert_eq!(format_key_value(&value, &ty, None).unwrap(), expected);
57        }
58    }
59}