language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-10-03 15:15:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

internal: cleanup hprof

Browse source
This commit is contained in:
Aleksey Kladov 2021-03-29 20:54:15 +03:00
parent 5dd6b93138
commit 2381a54c2f
1 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
crates/profile/src/hprof.rs
View file

@ -1,5 +1,4 @@
//! Simple hierarchical profiler //! Simple hierarchical profiler
use once_cell::sync::Lazy;
use std::{ use std::{
cell::RefCell, cell::RefCell,
collections::{BTreeMap, HashSet}, collections::{BTreeMap, HashSet},
@ -12,6 +11,8 @@ use std::{
time::{Duration, Instant}, time::{Duration, Instant},
}; };
use once_cell::sync::Lazy;
use crate::tree::{Idx, Tree}; use crate::tree::{Idx, Tree};
/// Filtering syntax /// Filtering syntax
@ -56,12 +57,12 @@ type Label = &'static str;
/// 0ms - profile /// 0ms - profile
/// 0ms - profile2 /// 0ms - profile2
/// ``` /// ```
#[inline]
pub fn span(label: Label) -> ProfileSpan { pub fn span(label: Label) -> ProfileSpan {
assert!(!label.is_empty()); debug_assert!(!label.is_empty());
if PROFILING_ENABLED.load(Ordering::Relaxed) let enabled = PROFILING_ENABLED.load(Ordering::Relaxed);
&& PROFILE_STACK.with(|stack| stack.borrow_mut().push(label)) if enabled && with_profile_stack(|stack| stack.push(label)) {
{
ProfileSpan(Some(ProfilerImpl { label, detail: None })) ProfileSpan(Some(ProfilerImpl { label, detail: None }))
} else { } else {
ProfileSpan(None) ProfileSpan(None)
@ -85,14 +86,19 @@ impl ProfileSpan {
} }
impl Drop for ProfilerImpl { impl Drop for ProfilerImpl {
#[inline]
fn drop(&mut self) { fn drop(&mut self) {
PROFILE_STACK.with(|it| it.borrow_mut().pop(self.label, self.detail.take())); with_profile_stack(|it| it.pop(self.label, self.detail.take()));
} }
} }
static PROFILING_ENABLED: AtomicBool = AtomicBool::new(false); static PROFILING_ENABLED: AtomicBool = AtomicBool::new(false);
static FILTER: Lazy<RwLock<Filter>> = Lazy::new(Default::default); static FILTER: Lazy<RwLock<Filter>> = Lazy::new(Default::default);
thread_local!(static PROFILE_STACK: RefCell<ProfileStack> = RefCell::new(ProfileStack::new()));
fn with_profile_stack<T>(f: impl FnOnce(&mut ProfileStack) -> T) -> T {
thread_local!(static STACK: RefCell<ProfileStack> = RefCell::new(ProfileStack::new()));
STACK.with(|it| f(&mut *it.borrow_mut()))
}
#[derive(Default, Clone, Debug)] #[derive(Default, Clone, Debug)]
struct Filter { struct Filter {