selene-db-gql 1.3.0

ISO/IEC 39075:2024 GQL parser, planner, optimizer, and executor for selene-db.
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

//! TxContext adaptive re-optimization hook tests.

mod exec_common;

use std::sync::{
    Arc,
    atomic::{AtomicU64, Ordering},
};

use selene_core::GraphId;
use selene_gql::{
    AdaptiveOptimizer, Binding, BindingTable, BindingTableSchema, EmptyProcedureRegistry,
    ImplDefinedCaps, PipelineOpId, TxContext, execute_pipeline,
};
use selene_graph::SeleneGraph;

struct CountingHook {
    observed: AtomicU64,
}

impl AdaptiveOptimizer for CountingHook {
    fn observe_cardinality(&self, _op: PipelineOpId, rows: u64) {
        self.observed.fetch_add(rows, Ordering::Relaxed);
    }
}

#[test]
fn tx_context_default_reopt_hook_is_none() {
    let caps = ImplDefinedCaps::default();
    let ctx = exec_common::empty_graph_context(&caps);

    assert!(ctx.reopt_hook().is_none());
}

#[test]
fn tx_context_with_reopt_hook_carries_through_dispatch() {
    let caps = ImplDefinedCaps::default();
    let hook = CountingHook {
        observed: AtomicU64::new(0),
    };
    let mut ctx = TxContext::read_only_with_reopt(
        Arc::new(SeleneGraph::new(GraphId::new(1_001))),
        &caps,
        &EmptyProcedureRegistry,
        &[],
        &hook,
    );
    let plan = exec_common::planned("RETURN 1 AS n");
    let input = BindingTable::new(
        BindingTableSchema { columns: vec![] },
        vec![Binding::empty()],
    );

    let output = execute_pipeline(&plan.pipeline, input, &mut ctx).expect("pipeline executes");

    assert!(ctx.reopt_hook().is_some());
    assert_eq!(output.row_count(), 1);
    assert_eq!(hook.observed.load(Ordering::Relaxed), 0);
}