spg-engine 7.37.21

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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

//! v7.37.17 (17.6 siblings) — MySQL JSON path machinery:
//! json_extract + json_contains_path.

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 json_extract_single_path() {
    let mut e = Engine::new();
    // MySQL doc vector: JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]') → 20.
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_extract('[10, 20, [30, 40]]', '$[1]')"
        )),
        "20"
    );
    // Nested object + index.
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_extract('{\"a\": {\"b\": [1, 2]}}', '$.a.b[1]')"
        )),
        "2"
    );
    // Quoted key.
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_extract('{\"a b\": 5}', '$.\"a b\"')"
        )),
        "5"
    );
    // Miss → NULL.
    assert!(matches!(
        first(&mut e, "SELECT json_extract('{\"a\": 1}', '$.zzz')"),
        spg_storage::Value::Null
    ));
}

#[test]
fn json_extract_multiple_paths() {
    let mut e = Engine::new();
    // MySQL doc vector: JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]',
    // '$[0]') → [20, 10].
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_extract('[10, 20, [30, 40]]', '$[1]', '$[0]')"
        )),
        "[20, 10]"
    );
}

#[test]
fn scalar_autowrap_at_index_zero() {
    let mut e = Engine::new();
    // MySQL: a scalar behaves as a one-element array for [0].
    assert_eq!(
        text(&first(
            &mut e,
            "SELECT json_extract('{\"a\": 5}', '$.a[0]')"
        )),
        "5"
    );
}

#[test]
fn json_contains_path_one_and_all() {
    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_path({doc}, 'one', '$.a', '$.e')")
        ),
        spg_storage::Value::Bool(true)
    ));
    assert!(matches!(
        first(
            &mut e,
            &format!("SELECT json_contains_path({doc}, 'all', '$.a', '$.e')")
        ),
        spg_storage::Value::Bool(false)
    ));
    assert!(matches!(
        first(
            &mut e,
            &format!("SELECT json_contains_path({doc}, 'one', '$.c.d')")
        ),
        spg_storage::Value::Bool(true)
    ));
}

#[test]
fn wildcard_errors_honestly() {
    let mut e = Engine::new();
    let err = e
        .execute("SELECT json_extract('[1, 2]', '$[*]')")
        .unwrap_err();
    let msg = format!("{err:?}");
    assert!(
        msg.contains("wildcards are not supported"),
        "unexpected error: {msg}"
    );
}