zeppelin-embed 0.8.0

In-process vector, lexical, and hybrid search for macOS and Apple silicon
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

//! Pre-allocation resident and temporary byte ceilings.

use super::StoreError;

#[derive(Clone, Copy)]
pub(crate) struct Budgets {
    resident: u64,
    temporary: u64,
}

impl Budgets {
    pub(crate) const fn new(resident: u64, temporary: u64) -> Self {
        Self {
            resident,
            temporary,
        }
    }

    pub(crate) fn check(
        self,
        resident_now: u64,
        temporary_now: u64,
        bytes: u64,
        component: &'static str,
        is_temporary: bool,
    ) -> Result<(u64, u64), StoreError> {
        let needed_resident =
            resident_now
                .checked_add(bytes)
                .ok_or(StoreError::BudgetExceeded {
                    needed: u64::MAX,
                    budget: self.resident,
                    component,
                })?;
        let needed_temporary = if is_temporary {
            temporary_now
                .checked_add(bytes)
                .ok_or(StoreError::BudgetExceeded {
                    needed: u64::MAX,
                    budget: self.temporary,
                    component,
                })?
        } else {
            temporary_now
        };
        if is_temporary && needed_temporary > self.temporary {
            return Err(StoreError::BudgetExceeded {
                needed: needed_temporary,
                budget: self.temporary,
                component,
            });
        }
        if needed_resident > self.resident {
            return Err(StoreError::BudgetExceeded {
                needed: needed_resident,
                budget: self.resident,
                component,
            });
        }
        Ok((needed_resident, needed_temporary))
    }
}