uqa-execution 0.4.5

Volcano physical operators with row-batch pipelines
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

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

use super::*;

#[test]
fn selected_column_payloads_are_admitted_once_and_survive_nested_readers() {
    let mut columns = columns("CREATE TABLE t (body TEXT)");
    columns[0].missing_value = Some(Value::Str("selected".repeat(4096)));
    let index = MemoryInvertedIndex::new(uqa_analysis::whitespace_analyzer());
    let schema = schema(&columns, &index);
    let definitions = Arc::downgrade(&schema.columns);
    let source: Arc<dyn DocumentStore> = Arc::new(MemoryDocumentStore::new());
    let rejected = StorageReadControl::with_limit(4096);
    let error = retain(
        Arc::clone(&source),
        &columns,
        &schema,
        DocumentChanges::default(),
        &rejected,
    )
    .err()
    .unwrap();
    assert_eq!(error.sqlstate(), Some("53200"));
    assert_eq!(rejected.memory().used(), 0);

    let control = StorageReadControl::with_limit(128 << 10);
    let admitted =
        RetainedColumns::capture(&schema.columns, control.memory(), control.cancellation())
            .unwrap();
    let catalog_bytes = admitted.reserved_bytes();
    drop(admitted);
    let view = retain(
        source,
        &columns,
        &schema,
        DocumentChanges::default(),
        &control,
    )
    .unwrap();
    let retained = control.memory().used();
    assert!(retained >= catalog_bytes);
    assert!(retained < 2 * catalog_bytes);
    let full = control
        .memory()
        .reserve(control.memory().limit() - retained)
        .unwrap();
    let nested = view.documents.snapshot().unwrap().snapshot().unwrap();
    assert_eq!(control.memory().used(), control.memory().limit());
    drop((full, view, schema));
    drop(columns);
    assert!(definitions.upgrade().is_some());
    assert!(control.memory().used() >= catalog_bytes);
    drop(nested);
    assert!(definitions.upgrade().is_none());
    assert_eq!(control.memory().used(), 0);
}