reifydb-allocator 0.10.0

Facade crate selecting a process-wide global allocator with shared-heap support for dlopened operator modules
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
86
87
88
89

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2026 ReifyDB

#![cfg_attr(not(debug_assertions), deny(clippy::disallowed_methods))]
#![cfg_attr(debug_assertions, warn(clippy::disallowed_methods))]
#![cfg_attr(not(debug_assertions), deny(warnings))]
#![allow(clippy::tabs_in_doc_comments)]

#[cfg(all(feature = "alloc-jemalloc", not(target_env = "msvc")))]
use reifydb_allocator_jemalloc::{stats, stats_dump};

#[cfg(all(feature = "alloc-jemalloc", not(target_env = "msvc")))]
pub mod backend {
	use reifydb_allocator_jemalloc::{ALLOCATOR as JEMALLOC_ALLOCATOR, verify as jemalloc_verify};

	pub type Allocator = reifydb_allocator_jemalloc::Allocator;
	pub const ALLOCATOR: Allocator = JEMALLOC_ALLOCATOR;

	pub fn verify() {
		jemalloc_verify();
	}
}

#[cfg(all(feature = "alloc-jemalloc", target_env = "msvc"))]
compile_error!(
	"reifydb-allocator: jemalloc is not supported on MSVC. \
	 Use the (Phase 2) `alloc-mimalloc` feature instead. \
	 See reifydb/plan-alloc.md."
);

#[macro_export]
#[cfg(feature = "alloc-jemalloc")]
macro_rules! set_global_allocator {
	() => {
		#[global_allocator]
		static REIFYDB_GLOBAL_ALLOCATOR: $crate::backend::Allocator = $crate::backend::ALLOCATOR;
	};
}

#[macro_export]
#[cfg(not(feature = "alloc-jemalloc"))]
macro_rules! set_global_allocator {
	() => {};
}

#[cfg(feature = "alloc-jemalloc")]
pub fn verify() {
	backend::verify();
}

#[cfg(not(feature = "alloc-jemalloc"))]
pub fn verify() {}

pub struct JemallocStats {
	pub allocated: u64,
	pub active: u64,
	pub resident: u64,
	pub mapped: u64,
	pub retained: u64,
	pub metadata: u64,
}

#[cfg(all(feature = "alloc-jemalloc", not(target_env = "msvc")))]
pub fn jemalloc_stats() -> Option<JemallocStats> {
	let (allocated, active, resident, mapped, retained, metadata) = stats();
	Some(JemallocStats {
		allocated,
		active,
		resident,
		mapped,
		retained,
		metadata,
	})
}

#[cfg(not(all(feature = "alloc-jemalloc", not(target_env = "msvc"))))]
pub fn jemalloc_stats() -> Option<JemallocStats> {
	None
}

#[cfg(all(feature = "alloc-jemalloc", not(target_env = "msvc")))]
pub fn jemalloc_stats_dump() -> Option<String> {
	stats_dump()
}

#[cfg(not(all(feature = "alloc-jemalloc", not(target_env = "msvc"))))]
pub fn jemalloc_stats_dump() -> Option<String> {
	None
}