nodedb-sql 0.1.0

SQL parser, planner, and optimizer for NodeDB
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

// SPDX-License-Identifier: Apache-2.0

//! Lowering helpers for PG full-text search surface operators.
//!
//! These functions are called from `resolver/expr.rs` to lower PG FTS
//! operators/casts to internal `SqlExpr::Function` nodes before the generic
//! expression lowering path runs.

use crate::types_expr::SqlExpr;

/// Lower `col @@ query_expr` to `pg_fts_match(col, query_expr)`.
pub fn lower_pg_fts_match(col: SqlExpr, query_expr: SqlExpr) -> SqlExpr {
    SqlExpr::Function {
        name: "pg_fts_match".into(),
        args: vec![col, query_expr],
        distinct: false,
    }
}

/// Lower `to_tsquery('...')` / `to_tsquery(lang, '...')` args.
///
/// The caller (resolver) has already converted the function args; this just
/// wraps them in a `pg_to_tsquery` function call.
pub fn lower_pg_to_tsquery(args: Vec<SqlExpr>) -> SqlExpr {
    SqlExpr::Function {
        name: "pg_to_tsquery".into(),
        args,
        distinct: false,
    }
}

/// Lower `plainto_tsquery('...')` / `plainto_tsquery(lang, '...')` args.
pub fn lower_pg_plainto_tsquery(args: Vec<SqlExpr>) -> SqlExpr {
    SqlExpr::Function {
        name: "pg_plainto_tsquery".into(),
        args,
        distinct: false,
    }
}

/// Lower `phraseto_tsquery('...')` args.
///
/// The resulting `pg_phraseto_tsquery` call is evaluated by the executor
/// to `FtsQuery::Phrase`, which is then rejected with `Unsupported`.
pub fn lower_phraseto_tsquery(args: Vec<SqlExpr>) -> SqlExpr {
    SqlExpr::Function {
        name: "pg_phraseto_tsquery".into(),
        args,
        distinct: false,
    }
}

/// Lower `websearch_to_tsquery('...')` / `websearch_to_tsquery(lang, '...')` args.
pub fn lower_pg_websearch_to_tsquery(args: Vec<SqlExpr>) -> SqlExpr {
    SqlExpr::Function {
        name: "pg_websearch_to_tsquery".into(),
        args,
        distinct: false,
    }
}