diff options
Diffstat (limited to 'crates/ra_prof/src')
-rw-r--r-- | crates/ra_prof/src/memory_usage.rs | 25 |
1 files changed, 15 insertions, 10 deletions
diff --git a/crates/ra_prof/src/memory_usage.rs b/crates/ra_prof/src/memory_usage.rs index 9768f656c..b1858b06f 100644 --- a/crates/ra_prof/src/memory_usage.rs +++ b/crates/ra_prof/src/memory_usage.rs | |||
@@ -1,5 +1,6 @@ | |||
1 | //! FIXME: write short doc here | 1 | //! FIXME: write short doc here |
2 | 2 | ||
3 | use cfg_if::cfg_if; | ||
3 | use std::fmt; | 4 | use std::fmt; |
4 | 5 | ||
5 | pub struct MemoryUsage { | 6 | pub struct MemoryUsage { |
@@ -8,19 +9,23 @@ pub struct MemoryUsage { | |||
8 | } | 9 | } |
9 | 10 | ||
10 | impl MemoryUsage { | 11 | impl MemoryUsage { |
11 | #[cfg(all(feature = "jemalloc", not(target_env = "msvc")))] | ||
12 | pub fn current() -> MemoryUsage { | 12 | pub fn current() -> MemoryUsage { |
13 | jemalloc_ctl::epoch::advance().unwrap(); | 13 | cfg_if! { |
14 | MemoryUsage { | 14 | if #[cfg(all(feature = "jemalloc", not(target_env = "msvc")))] { |
15 | allocated: Bytes(jemalloc_ctl::stats::allocated::read().unwrap()), | 15 | jemalloc_ctl::epoch::advance().unwrap(); |
16 | resident: Bytes(jemalloc_ctl::stats::resident::read().unwrap()), | 16 | MemoryUsage { |
17 | allocated: Bytes(jemalloc_ctl::stats::allocated::read().unwrap()), | ||
18 | resident: Bytes(jemalloc_ctl::stats::resident::read().unwrap()), | ||
19 | } | ||
20 | } else if #[cfg(target_os = "linux")] { | ||
21 | // Note: This is incredibly slow. | ||
22 | let alloc = unsafe { libc::mallinfo() }.uordblks as u32 as usize; | ||
23 | MemoryUsage { allocated: Bytes(alloc), resident: Bytes(0) } | ||
24 | } else { | ||
25 | MemoryUsage { allocated: Bytes(0), resident: Bytes(0) } | ||
26 | } | ||
17 | } | 27 | } |
18 | } | 28 | } |
19 | |||
20 | #[cfg(any(not(feature = "jemalloc"), target_env = "msvc"))] | ||
21 | pub fn current() -> MemoryUsage { | ||
22 | MemoryUsage { allocated: Bytes(0), resident: Bytes(0) } | ||
23 | } | ||
24 | } | 29 | } |
25 | 30 | ||
26 | impl fmt::Display for MemoryUsage { | 31 | impl fmt::Display for MemoryUsage { |