mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-23 04:39:04 +00:00
Merge branch 'main' into overtrue/activate-data-usage-e2e
This commit is contained in:
Generated
+22
-27
@@ -1858,9 +1858,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "cc"
|
name = "cc"
|
||||||
version = "1.4.3"
|
version = "1.4.4"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "509591b7bcd67f4ef775afad7662703b4935daaa6ec0e5605cfb1090b32a2b6d"
|
checksum = "0ad534f4357a5264cce5019c989cf66a4f0dc4e0d1b1d15f8aacec0ff7360273"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"find-msvc-tools",
|
"find-msvc-tools",
|
||||||
"jobserver",
|
"jobserver",
|
||||||
@@ -2522,12 +2522,6 @@ dependencies = [
|
|||||||
"subtle",
|
"subtle",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "cty"
|
|
||||||
version = "0.2.2"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "b365fabc795046672053e29c954733ec3b05e4be654ab130fe8f1f94d7051f35"
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "curve25519-dalek"
|
name = "curve25519-dalek"
|
||||||
version = "4.1.3"
|
version = "4.1.3"
|
||||||
@@ -5988,15 +5982,6 @@ version = "0.2.16"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b6d2cec3eae94f9f509c767b45932f1ada8350c4bdb85af2fcab4a3c14807981"
|
checksum = "b6d2cec3eae94f9f509c767b45932f1ada8350c4bdb85af2fcab4a3c14807981"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "libmimalloc-sys"
|
|
||||||
version = "0.1.49"
|
|
||||||
source = "git+https://github.com/xonatius/mimalloc_rust.git?rev=6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11#6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11"
|
|
||||||
dependencies = [
|
|
||||||
"cc",
|
|
||||||
"cty",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "libredox"
|
name = "libredox"
|
||||||
version = "0.1.20"
|
version = "0.1.20"
|
||||||
@@ -6397,14 +6382,6 @@ dependencies = [
|
|||||||
"synstructure 0.13.2",
|
"synstructure 0.13.2",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "mimalloc"
|
|
||||||
version = "0.1.52"
|
|
||||||
source = "git+https://github.com/xonatius/mimalloc_rust.git?rev=6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11#6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11"
|
|
||||||
dependencies = [
|
|
||||||
"libmimalloc-sys",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "mime"
|
name = "mime"
|
||||||
version = "0.3.17"
|
version = "0.3.17"
|
||||||
@@ -9162,13 +9139,11 @@ dependencies = [
|
|||||||
"insta",
|
"insta",
|
||||||
"jiff",
|
"jiff",
|
||||||
"libc",
|
"libc",
|
||||||
"libmimalloc-sys",
|
|
||||||
"libsystemd",
|
"libsystemd",
|
||||||
"matchit 0.9.2",
|
"matchit 0.9.2",
|
||||||
"md-5 0.11.0",
|
"md-5 0.11.0",
|
||||||
"metrics",
|
"metrics",
|
||||||
"metrics-util",
|
"metrics-util",
|
||||||
"mimalloc",
|
|
||||||
"mime_guess",
|
"mime_guess",
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
"opentelemetry_sdk",
|
"opentelemetry_sdk",
|
||||||
@@ -9204,6 +9179,8 @@ dependencies = [
|
|||||||
"rustfs-lock",
|
"rustfs-lock",
|
||||||
"rustfs-log-analyzer",
|
"rustfs-log-analyzer",
|
||||||
"rustfs-madmin",
|
"rustfs-madmin",
|
||||||
|
"rustfs-mimalloc",
|
||||||
|
"rustfs-mimalloc-sys",
|
||||||
"rustfs-notify",
|
"rustfs-notify",
|
||||||
"rustfs-object-capacity",
|
"rustfs-object-capacity",
|
||||||
"rustfs-object-data-cache",
|
"rustfs-object-data-cache",
|
||||||
@@ -9875,6 +9852,24 @@ dependencies = [
|
|||||||
"tokio",
|
"tokio",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rustfs-mimalloc"
|
||||||
|
version = "0.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a406f4aa07084301d485beec873af6dccc8e3f8762da244743df92038b1db1a6"
|
||||||
|
dependencies = [
|
||||||
|
"rustfs-mimalloc-sys",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rustfs-mimalloc-sys"
|
||||||
|
version = "0.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c3051b819175f58445d4c369a72f0ab88149f3885ba8bea2aff3be01f53fe7cd"
|
||||||
|
dependencies = [
|
||||||
|
"cc",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "rustfs-notify"
|
name = "rustfs-notify"
|
||||||
version = "1.0.0-rc.3"
|
version = "1.0.0-rc.3"
|
||||||
|
|||||||
+2
-2
@@ -350,8 +350,8 @@ russh-sftp = "2.4.0"
|
|||||||
dav-server = "0.11.0"
|
dav-server = "0.11.0"
|
||||||
|
|
||||||
# Performance Analysis and Memory Profiling
|
# Performance Analysis and Memory Profiling
|
||||||
mimalloc = { version = "0.1.52", git = "https://github.com/xonatius/mimalloc_rust.git", rev = "6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11" }
|
rustfs-mimalloc = { version = "0.5.0" }
|
||||||
libmimalloc-sys = { version = "0.1.49", git = "https://github.com/xonatius/mimalloc_rust.git", rev = "6d4c41bb10c6d9da1d1b6f07b38c4cc051667f11", features = ["extended"] }
|
rustfs-mimalloc-sys = { version = "0.5.0" }
|
||||||
hotpath = { version = "0.23.3", default-features = false }
|
hotpath = { version = "0.23.3", default-features = false }
|
||||||
# Snapshot testing for output format regression detection
|
# Snapshot testing for output format regression detection
|
||||||
insta = { version = "1.48" }
|
insta = { version = "1.48" }
|
||||||
|
|||||||
@@ -2124,26 +2124,13 @@ impl SetDisks {
|
|||||||
|
|
||||||
let put_object_size = known_put_object_storage_size(data.size());
|
let put_object_size = known_put_object_storage_size(data.size());
|
||||||
let shard_file_size_raw = erasure.shard_file_size(put_object_size);
|
let shard_file_size_raw = erasure.shard_file_size(put_object_size);
|
||||||
let is_inline_buffer =
|
let is_inline_buffer = storage_class_config.should_inline(shard_file_size_raw, erasure.data_shards, opts.versioned);
|
||||||
storage_class_config.should_inline(shard_file_size_raw, erasure.data_shards, opts.versioned);
|
|
||||||
|
|
||||||
let collect_stage_timing = rustfs_io_metrics::put_stage_metrics_enabled() || issue3031_diag_enabled();
|
let collect_stage_timing = rustfs_io_metrics::put_stage_metrics_enabled() || issue3031_diag_enabled();
|
||||||
let shard_file_size = shard_file_size_raw;
|
let shard_file_size = shard_file_size_raw;
|
||||||
let shard_size = erasure.shard_size();
|
let shard_size = erasure.shard_size();
|
||||||
let write_path = classify_put_write_path(is_inline_buffer, put_object_size, fi.erasure.block_size);
|
let write_path = classify_put_write_path(is_inline_buffer, put_object_size, fi.erasure.block_size);
|
||||||
let direct_inline_commit = matches!(write_path, SmallWritePath::Inline);
|
let direct_inline_commit = matches!(write_path, SmallWritePath::Inline);
|
||||||
{
|
|
||||||
use std::io::Write;
|
|
||||||
let msg = format!(
|
|
||||||
"INLINE_DEBUG: bucket={} obj={} size={} shard_fs={} ds={} bs={} inline={} direct={} path={} iblock={} ver={}\n",
|
|
||||||
bucket, object, put_object_size, shard_file_size_raw, erasure.data_shards, fi.erasure.block_size,
|
|
||||||
is_inline_buffer, direct_inline_commit, write_path.metric_label(), storage_class_config.inline_block(), opts.versioned
|
|
||||||
);
|
|
||||||
if let Ok(mut f) = std::fs::OpenOptions::new().create(true).append(true).open("/tmp/rustfs_inline_debug.log") {
|
|
||||||
let _ = f.write_all(msg.as_bytes());
|
|
||||||
}
|
|
||||||
let _ = std::io::stderr().write_all(msg.as_bytes());
|
|
||||||
}
|
|
||||||
rustfs_io_metrics::record_put_object_path(write_path.metric_label());
|
rustfs_io_metrics::record_put_object_path(write_path.metric_label());
|
||||||
let writer_setup_stage_start = collect_stage_timing.then(Instant::now);
|
let writer_setup_stage_start = collect_stage_timing.then(Instant::now);
|
||||||
let (mut writers, errors) = if direct_inline_commit {
|
let (mut writers, errors) = if direct_inline_commit {
|
||||||
|
|||||||
@@ -3194,7 +3194,7 @@ impl ECStore {
|
|||||||
|
|
||||||
// Default return value
|
// Default return value
|
||||||
let mut del_objects = vec![DeletedObject::default(); objects.len()];
|
let mut del_objects = vec![DeletedObject::default(); objects.len()];
|
||||||
let mut accounting = vec![None; objects.len()];
|
let accounting = vec![None; objects.len()];
|
||||||
|
|
||||||
let mut del_errs = Vec::with_capacity(objects.len());
|
let mut del_errs = Vec::with_capacity(objects.len());
|
||||||
for _ in 0..objects.len() {
|
for _ in 0..objects.len() {
|
||||||
|
|||||||
@@ -271,7 +271,7 @@ pub(super) fn resolve_latest_object_info_candidates(
|
|||||||
.filter(|candidate| latest_candidate_mod_time(candidate) == Some(latest_mod_time))
|
.filter(|candidate| latest_candidate_mod_time(candidate) == Some(latest_mod_time))
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
latest_candidates.sort_by(|left, right| right.idx.cmp(&left.idx));
|
latest_candidates.sort_by_key(|candidate| std::cmp::Reverse(candidate.idx));
|
||||||
|
|
||||||
let Some(winner) = latest_candidates.first() else {
|
let Some(winner) = latest_candidates.first() else {
|
||||||
return Err(Error::ErasureReadQuorum);
|
return Err(Error::ErasureReadQuorum);
|
||||||
|
|||||||
@@ -43,9 +43,6 @@ allow-git = [
|
|||||||
# RustFS fork carrying presigned expiry and constant-time authentication fixes.
|
# RustFS fork carrying presigned expiry and constant-time authentication fixes.
|
||||||
# owner: rustfs-maintainers review: 2026-10
|
# owner: rustfs-maintainers review: 2026-10
|
||||||
"https://github.com/rustfs/s3s.git",
|
"https://github.com/rustfs/s3s.git",
|
||||||
# MiMalloc fork pinned for hotpath allocation counting support.
|
|
||||||
# owner: houseme review: 2026-10
|
|
||||||
"https://github.com/xonatius/mimalloc_rust.git",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[bans]
|
[bans]
|
||||||
|
|||||||
+2
-2
@@ -336,13 +336,13 @@ opentelemetry = { workspace = true }
|
|||||||
tracing-opentelemetry = { workspace = true }
|
tracing-opentelemetry = { workspace = true }
|
||||||
# Data structures
|
# Data structures
|
||||||
hashbrown = { workspace = true, features = ["serde", "rayon"] }
|
hashbrown = { workspace = true, features = ["serde", "rayon"] }
|
||||||
mimalloc = { workspace = true }
|
rustfs-mimalloc = { workspace = true }
|
||||||
|
|
||||||
[target.'cfg(target_os = "linux")'.dependencies]
|
[target.'cfg(target_os = "linux")'.dependencies]
|
||||||
libsystemd.workspace = true
|
libsystemd.workspace = true
|
||||||
|
|
||||||
[target.'cfg(not(target_os = "windows"))'.dependencies]
|
[target.'cfg(not(target_os = "windows"))'.dependencies]
|
||||||
libmimalloc-sys.workspace = true
|
rustfs-mimalloc-sys.workspace = true
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
uuid = { workspace = true, features = ["v4", "v5", "fast-rng", "macro-diagnostics"] }
|
uuid = { workspace = true, features = ["v4", "v5", "fast-rng", "macro-diagnostics"] }
|
||||||
|
|||||||
@@ -369,14 +369,8 @@ pub fn allocator_reclaim_controller_snapshot(ctx: &CancellationToken) -> Allocat
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(not(target_os = "windows"))]
|
#[cfg(not(target_os = "windows"))]
|
||||||
#[allow(unsafe_code)]
|
|
||||||
fn collect_allocator_memory(force: bool) -> Result<(), String> {
|
fn collect_allocator_memory(force: bool) -> Result<(), String> {
|
||||||
// SAFETY: `mi_collect` is provided by the active global allocator backend
|
rustfs_mimalloc::MiMalloc::collect(force);
|
||||||
// on this target family. It is explicitly intended to reclaim retained
|
|
||||||
// pages/segments and does not require additional invariants from the caller.
|
|
||||||
unsafe {
|
|
||||||
libmimalloc_sys::mi_collect(force);
|
|
||||||
}
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+10
-8
@@ -26,22 +26,22 @@ struct MiMallocAllocator;
|
|||||||
unsafe impl GlobalAlloc for MiMallocAllocator {
|
unsafe impl GlobalAlloc for MiMallocAllocator {
|
||||||
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
||||||
// SAFETY: the caller upholds GlobalAlloc's contract for layout.
|
// SAFETY: the caller upholds GlobalAlloc's contract for layout.
|
||||||
unsafe { mimalloc::MiMalloc.alloc(layout) }
|
unsafe { rustfs_mimalloc::MiMalloc.alloc(layout) }
|
||||||
}
|
}
|
||||||
|
|
||||||
unsafe fn alloc_zeroed(&self, layout: Layout) -> *mut u8 {
|
unsafe fn alloc_zeroed(&self, layout: Layout) -> *mut u8 {
|
||||||
// SAFETY: the caller upholds GlobalAlloc's contract for layout.
|
// SAFETY: the caller upholds GlobalAlloc's contract for layout.
|
||||||
unsafe { mimalloc::MiMalloc.alloc_zeroed(layout) }
|
unsafe { rustfs_mimalloc::MiMalloc.alloc_zeroed(layout) }
|
||||||
}
|
}
|
||||||
|
|
||||||
unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
|
unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
|
||||||
// SAFETY: ptr and layout came from this allocator and are forwarded unchanged.
|
// SAFETY: ptr and layout came from this allocator and are forwarded unchanged.
|
||||||
unsafe { mimalloc::MiMalloc.dealloc(ptr, layout) }
|
unsafe { rustfs_mimalloc::MiMalloc.dealloc(ptr, layout) }
|
||||||
}
|
}
|
||||||
|
|
||||||
unsafe fn realloc(&self, ptr: *mut u8, layout: Layout, new_size: usize) -> *mut u8 {
|
unsafe fn realloc(&self, ptr: *mut u8, layout: Layout, new_size: usize) -> *mut u8 {
|
||||||
// SAFETY: ptr and layout came from this allocator and are forwarded unchanged.
|
// SAFETY: ptr and layout came from this allocator and are forwarded unchanged.
|
||||||
unsafe { mimalloc::MiMalloc.realloc(ptr, layout, new_size) }
|
unsafe { rustfs_mimalloc::MiMalloc.realloc(ptr, layout, new_size) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -51,7 +51,7 @@ static GLOBAL: hotpath::CountingAllocator<MiMallocAllocator> = hotpath::Counting
|
|||||||
|
|
||||||
#[cfg(not(all(feature = "hotpath", feature = "hotpath-alloc")))]
|
#[cfg(not(all(feature = "hotpath", feature = "hotpath-alloc")))]
|
||||||
#[global_allocator]
|
#[global_allocator]
|
||||||
static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
|
static GLOBAL: rustfs_mimalloc::MiMalloc = rustfs_mimalloc::MiMalloc;
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let _hotpath_guard = hotpath::HotpathGuardBuilder::new("main").build();
|
let _hotpath_guard = hotpath::HotpathGuardBuilder::new("main").build();
|
||||||
@@ -71,8 +71,9 @@ mod tests {
|
|||||||
allocation.extend_from_slice(&[7_u8; 64]);
|
allocation.extend_from_slice(&[7_u8; 64]);
|
||||||
|
|
||||||
assert_eq!(allocation.len(), 64);
|
assert_eq!(allocation.len(), 64);
|
||||||
|
let heap = rustfs_mimalloc::heap::Heap::main();
|
||||||
// SAFETY: the live Vec pointer is valid to inspect for heap ownership.
|
// SAFETY: the live Vec pointer is valid to inspect for heap ownership.
|
||||||
assert!(unsafe { libmimalloc_sys::mi_is_in_heap_region(allocation.as_ptr().cast()) });
|
assert!(unsafe { heap.contains(allocation.as_ptr()) });
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -85,12 +86,13 @@ mod tests {
|
|||||||
let layout = Layout::from_size_align(32, 8).expect("valid test allocation layout");
|
let layout = Layout::from_size_align(32, 8).expect("valid test allocation layout");
|
||||||
let grown_layout = Layout::from_size_align(64, 8).expect("valid grown test allocation layout");
|
let grown_layout = Layout::from_size_align(64, 8).expect("valid grown test allocation layout");
|
||||||
let allocator = super::MiMallocAllocator;
|
let allocator = super::MiMallocAllocator;
|
||||||
|
let heap = rustfs_mimalloc::heap::Heap::main();
|
||||||
|
|
||||||
// SAFETY: The pointer is checked for null before use and later released
|
// SAFETY: The pointer is checked for null before use and later released
|
||||||
// through the same allocator with the corresponding layout.
|
// through the same allocator with the corresponding layout.
|
||||||
let ptr = unsafe { allocator.alloc_zeroed(layout) };
|
let ptr = unsafe { allocator.alloc_zeroed(layout) };
|
||||||
assert!(!ptr.is_null());
|
assert!(!ptr.is_null());
|
||||||
assert!(unsafe { libmimalloc_sys::mi_is_in_heap_region(ptr.cast()) });
|
assert!(unsafe { heap.contains(ptr) });
|
||||||
assert!(unsafe { std::slice::from_raw_parts(ptr, 32).iter().all(|byte| *byte == 0) });
|
assert!(unsafe { std::slice::from_raw_parts(ptr, 32).iter().all(|byte| *byte == 0) });
|
||||||
|
|
||||||
// SAFETY: `ptr` was allocated by `allocator` with `layout`; on failure
|
// SAFETY: `ptr` was allocated by `allocator` with `layout`; on failure
|
||||||
@@ -102,7 +104,7 @@ mod tests {
|
|||||||
panic!("mimalloc realloc failed in allocator smoke test");
|
panic!("mimalloc realloc failed in allocator smoke test");
|
||||||
}
|
}
|
||||||
|
|
||||||
assert!(unsafe { libmimalloc_sys::mi_is_in_heap_region(grown_ptr.cast()) });
|
assert!(unsafe { heap.contains(grown_ptr) });
|
||||||
// SAFETY: `grown_ptr` was reallocated by `allocator` and is released
|
// SAFETY: `grown_ptr` was reallocated by `allocator` and is released
|
||||||
// with the matching grown layout.
|
// with the matching grown layout.
|
||||||
unsafe { allocator.dealloc(grown_ptr, grown_layout) };
|
unsafe { allocator.dealloc(grown_ptr, grown_layout) };
|
||||||
|
|||||||
@@ -17,10 +17,7 @@ use rustfs_io_metrics::{
|
|||||||
record_cpu_usage, record_memory_usage, record_process_memory_split,
|
record_cpu_usage, record_memory_usage, record_process_memory_split,
|
||||||
};
|
};
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
#[cfg(not(target_os = "windows"))]
|
|
||||||
use std::ffi::CStr;
|
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::sync::{Arc, Mutex, OnceLock};
|
use std::sync::{Arc, Mutex, OnceLock};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@@ -231,7 +228,18 @@ fn read_cgroup_memory_snapshot() -> Option<CgroupMemorySnapshot> {
|
|||||||
read_cgroup_v2().or_else(read_cgroup_v1)
|
read_cgroup_v2().or_else(read_cgroup_v1)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
fn read_allocator_memory_snapshot() -> Option<AllocatorMemorySnapshot> {
|
||||||
|
let json = rustfs_mimalloc::MiMalloc::stats_json();
|
||||||
|
if json.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let observation = parse_mimalloc_stats_json(&json)?;
|
||||||
|
Some(AllocatorMemorySnapshot {
|
||||||
|
backend: crate::allocator_reclaim::allocator_backend(),
|
||||||
|
observation,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
fn numeric_json_value(value: &Value) -> Option<u64> {
|
fn numeric_json_value(value: &Value) -> Option<u64> {
|
||||||
match value {
|
match value {
|
||||||
Value::Number(number) => number
|
Value::Number(number) => number
|
||||||
@@ -242,7 +250,6 @@ fn numeric_json_value(value: &Value) -> Option<u64> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
fn numeric_json_field(value: &Value, field: &str) -> Option<u64> {
|
fn numeric_json_field(value: &Value, field: &str) -> Option<u64> {
|
||||||
match value {
|
match value {
|
||||||
Value::Object(fields) => fields
|
Value::Object(fields) => fields
|
||||||
@@ -254,7 +261,6 @@ fn numeric_json_field(value: &Value, field: &str) -> Option<u64> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
fn mimalloc_stat_field(value: &Value, metric: &str, field: &str) -> Option<u64> {
|
fn mimalloc_stat_field(value: &Value, metric: &str, field: &str) -> Option<u64> {
|
||||||
match value {
|
match value {
|
||||||
Value::Object(fields) => {
|
Value::Object(fields) => {
|
||||||
@@ -271,12 +277,10 @@ fn mimalloc_stat_field(value: &Value, metric: &str, field: &str) -> Option<u64>
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
fn mimalloc_stat_current(value: &Value, metric: &str) -> Option<u64> {
|
fn mimalloc_stat_current(value: &Value, metric: &str) -> Option<u64> {
|
||||||
mimalloc_stat_field(value, metric, "current")
|
mimalloc_stat_field(value, metric, "current")
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
fn mimalloc_stat_sum(value: &Value, metrics: &[&str], field: &str) -> Option<u64> {
|
fn mimalloc_stat_sum(value: &Value, metrics: &[&str], field: &str) -> Option<u64> {
|
||||||
metrics
|
metrics
|
||||||
.iter()
|
.iter()
|
||||||
@@ -285,7 +289,6 @@ fn mimalloc_stat_sum(value: &Value, metrics: &[&str], field: &str) -> Option<u64
|
|||||||
.filter(|value| *value > 0)
|
.filter(|value| *value > 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(any(test, not(target_os = "windows")))]
|
|
||||||
fn parse_mimalloc_stats_json(stats_json: &str) -> Option<AllocatorMemoryObservation> {
|
fn parse_mimalloc_stats_json(stats_json: &str) -> Option<AllocatorMemoryObservation> {
|
||||||
let value = serde_json::from_str::<Value>(stats_json).ok()?;
|
let value = serde_json::from_str::<Value>(stats_json).ok()?;
|
||||||
let malloc_metrics = ["malloc_normal", "malloc_huge"];
|
let malloc_metrics = ["malloc_normal", "malloc_huge"];
|
||||||
@@ -312,33 +315,6 @@ fn parse_mimalloc_stats_json(stats_json: &str) -> Option<AllocatorMemoryObservat
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(not(target_os = "windows"))]
|
|
||||||
#[allow(unsafe_code)]
|
|
||||||
fn read_allocator_memory_snapshot() -> Option<AllocatorMemorySnapshot> {
|
|
||||||
// SAFETY: `mi_stats_get_json` returns a null-terminated JSON buffer owned by
|
|
||||||
// mimalloc when called with a null input buffer. The mimalloc API requires
|
|
||||||
// freeing that buffer with `mi_free`; parsing finishes before the buffer is freed.
|
|
||||||
let observation = unsafe {
|
|
||||||
let stats_ptr = libmimalloc_sys::mi_stats_get_json(0, std::ptr::null_mut());
|
|
||||||
if stats_ptr.is_null() {
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let observation = CStr::from_ptr(stats_ptr).to_str().ok().and_then(parse_mimalloc_stats_json);
|
|
||||||
libmimalloc_sys::mi_free(stats_ptr.cast());
|
|
||||||
observation?
|
|
||||||
};
|
|
||||||
Some(AllocatorMemorySnapshot {
|
|
||||||
backend: crate::allocator_reclaim::allocator_backend(),
|
|
||||||
observation,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(target_os = "windows")]
|
|
||||||
fn read_allocator_memory_snapshot() -> Option<AllocatorMemorySnapshot> {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
|
|
||||||
fn configured_memory_observability_interval_secs() -> u64 {
|
fn configured_memory_observability_interval_secs() -> u64 {
|
||||||
rustfs_utils::get_env_u64(ENV_MEMORY_OBSERVABILITY_INTERVAL_SECS, DEFAULT_MEMORY_OBSERVABILITY_INTERVAL_SECS).max(1)
|
rustfs_utils::get_env_u64(ENV_MEMORY_OBSERVABILITY_INTERVAL_SECS, DEFAULT_MEMORY_OBSERVABILITY_INTERVAL_SECS).max(1)
|
||||||
}
|
}
|
||||||
@@ -566,6 +542,13 @@ mod tests {
|
|||||||
assert_eq!(parse_mimalloc_stats_json(r#"{ "allocator": "unknown" }"#), None);
|
assert_eq!(parse_mimalloc_stats_json(r#"{ "allocator": "unknown" }"#), None);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn read_allocator_memory_snapshot_uses_mimalloc_stats_json() {
|
||||||
|
let snapshot = super::read_allocator_memory_snapshot();
|
||||||
|
#[cfg(not(target_os = "windows"))]
|
||||||
|
assert!(snapshot.is_some(), "allocator snapshot should be available on non-Windows");
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn memory_observability_snapshot_reports_disabled_when_metrics_are_disabled() {
|
fn memory_observability_snapshot_reports_disabled_when_metrics_are_disabled() {
|
||||||
let snapshot = build_memory_observability_status_snapshot(false, 15, false);
|
let snapshot = build_memory_observability_status_snapshot(false, 15, false);
|
||||||
|
|||||||
Reference in New Issue
Block a user