uqa-sql 0.5.2

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

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

//! Scalar SQL RETURN bodies over the existing typed expression deparser.

use super::{expression, invalid, list, ExpressionNames, Field, SQLError};

/// Print a built-in routine's single scalar SELECT RETURN query without discarding query clauses.
pub fn return_body(value: &Field, names: &dyn ExpressionNames) -> Result<String, SQLError> {
    let Field::Node(query) = value else {
        return Err(invalid("expected a SQL RETURN query"));
    };
    if query.kind != "QUERY"
        || query.integer("commandType")? != 1
        || !query.boolean("isReturn")?
        || query.integer("resultRelation")? != 0
    {
        return Err(invalid("expected a scalar SELECT RETURN query"));
    }
    for field in [
        "cteList",
        "rtable",
        "groupClause",
        "groupingSets",
        "windowClause",
        "distinctClause",
        "sortClause",
        "rowMarks",
        "returningList",
    ] {
        if !list(query, field)?.is_empty() {
            return Err(invalid(format!(
                "unsupported {field} in scalar RETURN query"
            )));
        }
    }
    for field in [
        "utilityStmt",
        "onConflict",
        "havingQual",
        "limitOffset",
        "limitCount",
        "setOperations",
    ] {
        if query.field(field)? != &Field::Null {
            return Err(invalid(format!(
                "unsupported {field} in scalar RETURN query"
            )));
        }
    }
    let Field::Node(from) = query.field("jointree")? else {
        return Err(invalid("expected an empty FROMEXPR in scalar RETURN query"));
    };
    if from.kind != "FROMEXPR"
        || !list(from, "fromlist")?.is_empty()
        || from.field("quals")? != &Field::Null
    {
        return Err(invalid("expected an empty FROMEXPR in scalar RETURN query"));
    }
    let [Field::Node(target)] = list(query, "targetList")? else {
        return Err(invalid("expected one target in scalar RETURN query"));
    };
    if target.kind != "TARGETENTRY" || target.boolean("resjunk")? || target.integer("resno")? != 1 {
        return Err(invalid("expected one non-junk scalar RETURN target"));
    }
    let target_expression @ Field::Node(_) = target.field("expr")? else {
        return Err(invalid("expected a scalar RETURN expression node"));
    };
    expression(target_expression, names, false).map(|expression| format!("RETURN {expression}"))
}