spg-engine 7.37.23

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

//! v7.37.17 (17.6 siblings) — PG 14+ date_bin(stride, ts, origin).

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 ts(v: &spg_storage::Value<'_>) -> i64 {
    match v {
        spg_storage::Value::Timestamp(t) => *t,
        other => panic!("expected Timestamp, got {other:?}"),
    }
}

#[test]
fn date_bin_15_minute_stride() {
    let mut e = Engine::new();
    let expected = ts(&first(&mut e, "SELECT '2020-01-01 00:30:00'::timestamp"));
    let got = ts(&first(
        &mut e,
        "SELECT date_bin(INTERVAL '15 minutes', \
         '2020-01-01 00:37:00'::timestamp, \
         '2020-01-01 00:00:00'::timestamp)",
    ));
    assert_eq!(got, expected);
}

#[test]
fn date_bin_1_hour_stride() {
    let mut e = Engine::new();
    let expected = ts(&first(&mut e, "SELECT '2020-01-01 03:00:00'::timestamp"));
    let got = ts(&first(
        &mut e,
        "SELECT date_bin(INTERVAL '1 hour', \
         '2020-01-01 03:45:00'::timestamp, \
         '2020-01-01 00:00:00'::timestamp)",
    ));
    assert_eq!(got, expected);
}

#[test]
fn date_bin_1_day_stride() {
    let mut e = Engine::new();
    let expected = ts(&first(&mut e, "SELECT '2020-01-15 00:00:00'::timestamp"));
    let got = ts(&first(
        &mut e,
        "SELECT date_bin(INTERVAL '1 day', \
         '2020-01-15 14:30:00'::timestamp, \
         '2020-01-01 00:00:00'::timestamp)",
    ));
    assert_eq!(got, expected);
}

#[test]
fn date_bin_negative_stride_errors() {
    let mut e = Engine::new();
    assert!(
        e.execute(
            "SELECT date_bin(INTERVAL '-1 hour', \
             '2020-01-01'::timestamp, '2020-01-01'::timestamp)"
        )
        .is_err()
    );
}

#[test]
fn date_bin_null_passthrough() {
    let mut e = Engine::new();
    assert!(matches!(
        first(
            &mut e,
            "SELECT date_bin(NULL::interval, \
             '2020-01-01'::timestamp, '2020-01-01'::timestamp)"
        ),
        spg_storage::Value::Null
    ));
}

#[test]
fn date_bin_accepts_text_stride() {
    // PG coerces an unadorned interval string to the stride type, so
    // date_bin('15 minutes', …) works like INTERVAL '15 minutes'.
    // Verified vs live PG18.4.
    let mut e = Engine::new();
    let q = |e: &mut Engine, sql: &str| -> String {
        match e.execute(sql) {
            Ok(QueryResult::Rows { rows, .. }) => match &rows[0].values[0] {
                spg_storage::Value::Text(s) => s.to_string(),
                o => format!("{o:?}"),
            },
            o => format!("{o:?}"),
        }
    };
    assert_eq!(
        q(
            &mut e,
            "SELECT date_bin('15 minutes', timestamp '2020-01-01 00:17:00', timestamp '2020-01-01 00:00:00')::text"
        ),
        "2020-01-01 00:15:00"
    );
    assert_eq!(
        q(
            &mut e,
            "SELECT date_bin('2 days', timestamp '2020-01-06 12:00:00', timestamp '2020-01-01 00:00:00')::text"
        ),
        "2020-01-05 00:00:00"
    );
    // A month-bearing stride still errors (PG: "cannot be binned into
    // intervals containing months or years").
    assert!(
        e.execute("SELECT date_bin('1 month', timestamp '2020-03-15', timestamp '2020-01-01')")
            .is_err()
    );
}