icydb-core 0.257.9

IcyDB — A schema-first typed query engine and persistence runtime for Internet Computer canisters
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

use super::ConstructionBudget;
use crate::db::{
    QueryError, RequestExecutionRoot,
    executor::budget::{HardExecutionBudget, HardExecutionFailureHeadroom},
    query::preparation::PreparationWork,
};
use icydb_diagnostic_code::{
    DiagnosticExecutionBudgetResource as Resource, DiagnosticExecutionLane as Lane,
    DiagnosticFactTag,
};

fn request(bytes: u64) -> RequestExecutionRoot {
    RequestExecutionRoot::new_for_tests(
        HardExecutionBudget::uniform_for_tests(
            16_000_000,
            HardExecutionFailureHeadroom::new(500_000_000, 64 * 1024),
        )
        .with_limit_for_tests(Resource::TemporaryBytes, bytes),
    )
}

#[test]
fn vector_reservation_charges_replacement_capacity_before_mutation() {
    let exact = (4 + 8) * size_of::<u64>() as u64;
    for limit in [exact - 1, exact] {
        let root = request(limit);
        PreparationWork::run(&root.scope(), Lane::PublicRead, |work| {
            let mut values = Vec::new();
            work.reserve_vec(&mut values, 4)?;
            values.extend_from_slice(&[3_u64, 1, 2, 1]);
            let pointer = values.as_ptr();
            let capacity = values.capacity();
            let result = (work as &dyn ConstructionBudget).reserve_vec(&mut values, 1);
            if limit == exact {
                result.unwrap();
                assert!(values.capacity() >= 5);
                // A reserve that fits existing backing is free.
                work.reserve_vec(&mut values, 1)?;
            } else {
                let error = QueryError::execute(result.unwrap_err());
                assert!(error.diagnostic_facts().contains(&(
                    DiagnosticFactTag::BudgetResource,
                    Resource::TemporaryBytes.raw()
                )));
                assert_eq!(values.capacity(), capacity);
                assert_eq!(values.as_ptr(), pointer);
            }
            assert_eq!(values, [3, 1, 2, 1]);
            Ok(())
        })
        .unwrap();
        assert_eq!(root.observed(Resource::TemporaryBytes), exact);
    }
}

#[test]
fn text_reservation_and_append_share_the_same_growth_charge() {
    for limit in [11, 12] {
        let root = request(limit);
        PreparationWork::run(&root.scope(), Lane::Diagnostic, |work| {
            let budget: &dyn ConstructionBudget = work;
            let mut text = String::new();
            work.reserve_string(&mut text, 4)?;
            budget
                .push_text(&mut text, "éé")
                .map_err(QueryError::execute)?;
            let pointer = text.as_ptr();
            let capacity = text.capacity();
            let result = budget.push_text(&mut text, "!");
            if limit == 12 {
                result.unwrap();
                assert_eq!(text, "éé!");
            } else {
                assert!(result.is_err());
                assert_eq!(text, "éé");
                assert_eq!(text.capacity(), capacity);
                assert_eq!(text.as_ptr(), pointer);
            }
            Ok(())
        })
        .unwrap();
        assert_eq!(root.observed(Resource::TemporaryBytes), 12);
        assert_eq!(root.observed(Resource::PredicateExpressionSteps), 5);
    }
}