spg-engine 7.37.16

Execution engine for SPG: glues spg-sql parsing to spg-storage. Foreign keys, joins, vectors, cold tier.
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
106
107

//! v7.37.17 (17.6 siblings) — MySQL json_array_append /
//! json_array_insert / json_contains.

use spg_engine::{Engine, QueryResult};

fn first(e: &mut Engine, sql: &str) -> spg_storage::Value<'static> {
    let r = e
        .execute(sql)
        .unwrap_or_else(|err| panic!("{sql}: {err:?}"));
    let QueryResult::Rows { rows, .. } = r else {
        panic!("expected Rows");
    };
    rows[0].values[0].clone()
}

fn text(v: &spg_storage::Value<'_>) -> String {
    match v {
        spg_storage::Value::Text(s) => s.to_string(),
        spg_storage::Value::Json(s) => s.to_string(),
        other => panic!("expected Text/Json, got {other:?}"),
    }
}

#[test]
fn array_append_vectors() {
    let mut e = Engine::new();
    // MySQL doc vectors on '["a", ["b", "c"], "d"]'.
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_array_append('[\"a\", [\"b\", \"c\"], \"d\"]', '$[1]', 1)"
        )),
        "[\"a\", [\"b\", \"c\", 1], \"d\"]"
    );
    // Non-array target wraps as [old, new].
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_array_append('[\"a\", [\"b\", \"c\"], \"d\"]', '$[0]', 2)"
        )),
        "[[\"a\", 2], [\"b\", \"c\"], \"d\"]"
    );
}

#[test]
fn array_insert_vectors() {
    let mut e = Engine::new();
    // MySQL doc vector: JSON_ARRAY_INSERT('["a", {"b": [1, 2]},
    // [3, 4]]', '$[1]', 'x') → ["a", "x", {"b": [1, 2]}, [3, 4]].
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_array_insert('[\"a\", {\"b\": [1, 2]}, [3, 4]]', '$[1]', 'x')"
        )),
        "[\"a\", \"x\", {\"b\": [1, 2]}, [3, 4]]"
    );
    // Past-the-end appends.
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_array_insert('[1, 2]', '$[100]', 3)"
        )),
        "[1, 2, 3]"
    );
    // Path not ending in [N] errors.
    let err = e
        .execute("SELECT json_array_insert('{\"a\": []}', '$.a', 1)")
        .unwrap_err();
    let msg = format!("{err:?}");
    assert!(msg.contains("array index"), "unexpected error: {msg}");
}

#[test]
fn json_contains_vectors() {
    let mut e = Engine::new();
    // MySQL doc vectors on '{"a": 1, "b": 2, "c": {"d": 4}}'.
    let doc = "'{\"a\": 1, \"b\": 2, \"c\": {\"d\": 4}}'";
    assert!(matches!(
        first(&mut e, &format!("SELECT json_contains({doc}, '1', '$.a')")),
        spg_storage::Value::Bool(true)
    ));
    assert!(matches!(
        first(&mut e, &format!("SELECT json_contains({doc}, '1', '$.b')")),
        spg_storage::Value::Bool(false)
    ));
    assert!(matches!(
        first(
            &mut e,
            &format!("SELECT json_contains({doc}, '{{\"d\": 4}}', '$.c')")
        ),
        spg_storage::Value::Bool(true)
    ));
    // Array containment: every candidate element must be contained.
    assert!(matches!(
        first(&mut e, "SELECT json_contains('[1, 2, 3]', '[1, 3]')"),
        spg_storage::Value::Bool(true)
    ));
    assert!(matches!(
        first(&mut e, "SELECT json_contains('[1, 2, 3]', '[1, 9]')"),
        spg_storage::Value::Bool(false)
    ));
    // Scalar in array.
    assert!(matches!(
        first(&mut e, "SELECT json_contains('[1, 2]', '2')"),
        spg_storage::Value::Bool(true)
    ));
}