uqa_sql/catalog/index/
output.rs1use 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}