hawdb-storage 0.14.0

An embedded Rust graph database for local data-plane workloads
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

use super::*;
use crate::projection::ProjectedRelationshipPredicate;

fn entry() -> WalEntry {
    WalEntry {
        lsn: 17,
        op: WalOp::ProjectGraph {
            name: "wide-predicate".into(),
            node_labels: vec!["Memory".into()],
            rel_types: vec!["LINKS".into()],
            relationship_predicates: BTreeMap::from([(
                "LINKS".into(),
                ProjectedRelationshipPredicate::Eq {
                    property: "body".into(),
                    value: Value::Binary((0..257 * 1024 + 3).map(|i| (i % 256) as u8).collect()),
                },
            )]),
        },
    }
}

#[test]
fn checkpoint_units_predicate_admission_counts_before_any_payload_sized_allocation() {
    let entry = entry();
    let governor = governor(1);
    let permit = governor
        .try_admit(RuntimeWorkRequest::background_maintenance(1))
        .unwrap();
    let work = CheckpointWorkContext::new(permit.bind_task_context(RuntimeTaskContext::default()));
    let observation = crate::test_allocator::AllocationObservation::start();
    let result = work.classify(|work| encode_binary_wal_record_with_work_context(&entry, 23, work));
    let large_allocations = observation.finish();
    assert!(matches!(
        result,
        Err(crate::background::CheckpointOperationError::Work(
            CheckpointWorkError::Memory(RuntimeMemoryError::ReservationExceeded { .. })
        ))
    ));
    assert_eq!(
        large_allocations, 0,
        "denied payload admission must precede any allocation over the 64 KiB work bound"
    );
    drop(work);
    drop(permit);
    assert_eq!(governor.snapshot().admitted_memory_bytes, 0);
}

#[test]
fn checkpoint_units_predicate_admission_charges_emission_scratch_with_retained_output() {
    let entry = entry();
    let expected = encode_binary_wal_record(&entry, 23).unwrap();
    let requested = expected.len() as u64 + 24;
    let ceiling = requested + 64 * 1024 - 1;
    let governor = governor(ceiling);
    let permit = governor
        .try_admit(RuntimeWorkRequest::background_maintenance(ceiling))
        .unwrap();
    let work = CheckpointWorkContext::new(permit.bind_task_context(RuntimeTaskContext::default()));
    let result = work.classify(|work| encode_binary_wal_record_with_work_context(&entry, 23, work));
    assert!(matches!(
        result,
        Err(crate::background::CheckpointOperationError::Work(
            CheckpointWorkError::Memory(RuntimeMemoryError::ReservationExceeded {
                requested_bytes: 65560,
                available_bytes: 65535
            })
        ))
    ));
    drop(work);
    drop(permit);
    assert_eq!(governor.snapshot().admitted_memory_bytes, 0);
}