akar-common 0.1.21

Shared types and utilities for the Akar embedded graph database
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

//! Memory management utilities for the buffer manager.
//!
//! Tracks allocated memory and provides backpressure hints.

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

/// A simple memory tracker for the database instance.
#[derive(Debug)]
pub struct MemoryManager {
    total_allocated: AtomicU64,
    max_memory: u64,
}

impl MemoryManager {
    pub fn new(max_memory: u64) -> Self {
        Self {
            total_allocated: AtomicU64::new(0),
            max_memory,
        }
    }

    pub fn max_memory(&self) -> u64 {
        self.max_memory
    }

    pub fn total_allocated(&self) -> u64 {
        self.total_allocated.load(Ordering::Relaxed)
    }

    pub fn allocate(&self, amount: u64) {
        self.total_allocated.fetch_add(amount, Ordering::Relaxed);
    }

    pub fn deallocate(&self, amount: u64) {
        self.total_allocated.fetch_sub(amount, Ordering::Relaxed);
    }

    pub fn is_under_limit(&self) -> bool {
        self.total_allocated() <= self.max_memory
    }
}

impl Default for MemoryManager {
    fn default() -> Self {
        // Default to 80% of available system memory (approximate).
        Self::new(u64::MAX)
    }
}