polars_utils/
sys.rs

1use std::sync::{LazyLock, Mutex};
2
3use sysinfo::System;
4
5/// Startup system is expensive, so we do it once
6pub struct MemInfo {
7    sys: Mutex<System>,
8}
9
10impl MemInfo {
11    /// This call is quite expensive, cache the results.
12    pub fn free(&self) -> u64 {
13        let mut sys = self.sys.lock().unwrap();
14        sys.refresh_memory();
15        match sys.cgroup_limits() {
16            Some(limits) => limits.free_memory,
17            None => sys.available_memory(),
18        }
19    }
20}
21
22pub static MEMINFO: LazyLock<MemInfo> = LazyLock::new(|| MemInfo {
23    sys: Mutex::new(System::new()),
24});