uqa-sql 0.5.4

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

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

//! `PostgreSQL` identities shared by JSON path overload selection and catalog projection.

use crate::{ast::FunctionDispatch, ColumnType};

#[derive(Debug, Clone, Copy)]
pub struct JsonPathFunction {
    pub name: &'static str,
    pub oid: i64,
    pub argument_oids: &'static [i64],
    pub result_oid: i64,
    pub jsonb: bool,
    pub as_text: bool,
}

pub const JSON_PATH_FUNCTIONS: [JsonPathFunction; 4] = [
    JsonPathFunction {
        name: "json_extract_path",
        oid: 3951,
        argument_oids: &[114, 1009],
        result_oid: 114,
        jsonb: false,
        as_text: false,
    },
    JsonPathFunction {
        name: "json_extract_path_text",
        oid: 3953,
        argument_oids: &[114, 1009],
        result_oid: 25,
        jsonb: false,
        as_text: true,
    },
    JsonPathFunction {
        name: "jsonb_extract_path",
        oid: 3217,
        argument_oids: &[3802, 1009],
        result_oid: 3802,
        jsonb: true,
        as_text: false,
    },
    JsonPathFunction {
        name: "jsonb_extract_path_text",
        oid: 3940,
        argument_oids: &[3802, 1009],
        result_oid: 25,
        jsonb: true,
        as_text: true,
    },
];

impl JsonPathFunction {
    #[must_use]
    pub const fn by_oid(oid: i64) -> Option<Self> {
        let mut index = 0;
        while index < JSON_PATH_FUNCTIONS.len() {
            if JSON_PATH_FUNCTIONS[index].oid == oid {
                return Some(JSON_PATH_FUNCTIONS[index]);
            }
            index += 1;
        }
        None
    }

    #[must_use]
    pub fn lookup(name: &str) -> Option<Self> {
        JSON_PATH_FUNCTIONS.into_iter().find(|entry| {
            name.eq_ignore_ascii_case(entry.name)
                || name
                    .get(..11)
                    .is_some_and(|prefix| prefix.eq_ignore_ascii_case("pg_catalog."))
                    && name[11..].eq_ignore_ascii_case(entry.name)
        })
    }

    #[must_use]
    pub const fn input_type(self) -> ColumnType {
        if self.jsonb {
            ColumnType::JsonB
        } else {
            ColumnType::Json
        }
    }

    #[must_use]
    pub const fn result_type(self) -> ColumnType {
        if self.as_text {
            ColumnType::Text
        } else {
            self.input_type()
        }
    }

    #[must_use]
    pub const fn dispatch(self) -> FunctionDispatch {
        FunctionDispatch::JsonPath {
            jsonb: self.jsonb,
            as_text: self.as_text,
        }
    }
}