Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2025-01-14 22:24:14 +00:00
Code Issues Projects Releases Packages Wiki Activity

fast path for disabled profiler

Browse source
This commit is contained in:
Aleksey Kladov 2019-04-14 22:54:57 +03:00
parent 30a4099ea7
commit e6f32c6d3a
1 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
crates/ra_prof/src/lib.rs
View file

@ -6,7 +6,7 @@ use std::iter::repeat;
use std::collections::{HashSet}; use std::collections::{HashSet};
use std::default::Default; use std::default::Default;
use std::iter::FromIterator; use std::iter::FromIterator;
use std::sync::RwLock; use std::sync::{RwLock, atomic::{AtomicBool, Ordering}};
use lazy_static::lazy_static; use lazy_static::lazy_static;
/// Set profiling filter. It specifies descriptions allowed to profile. /// Set profiling filter. It specifies descriptions allowed to profile.
@ -24,6 +24,7 @@ use lazy_static::lazy_static;
/// ``` /// ```
/// ///
pub fn set_filter(f: Filter) { pub fn set_filter(f: Filter) {
PROFILING_ENABLED.store(f.depth > 0, Ordering::SeqCst);
let set = HashSet::from_iter(f.allowed.iter().cloned()); let set = HashSet::from_iter(f.allowed.iter().cloned());
let mut old = FILTER.write().unwrap(); let mut old = FILTER.write().unwrap();
let filter_data = FilterData { depth: f.depth, allowed: set, version: old.version + 1 }; let filter_data = FilterData { depth: f.depth, allowed: set, version: old.version + 1 };
@ -62,6 +63,11 @@ pub fn set_filter(f: Filter) {
/// ``` /// ```
/// ///
pub fn profile(desc: &str) -> Profiler { pub fn profile(desc: &str) -> Profiler {
assert!(!desc.is_empty());
if !PROFILING_ENABLED.load(Ordering::Relaxed) {
return Profiler { desc: None };
}
PROFILE_STACK.with(|stack| { PROFILE_STACK.with(|stack| {
let mut stack = stack.borrow_mut(); let mut stack = stack.borrow_mut();
if stack.starts.len() == 0 { if stack.starts.len() == 0 {
@ -74,14 +80,14 @@ pub fn profile(desc: &str) -> Profiler {
Err(_) => (), Err(_) => (),
}; };
} }
let desc_str = desc.to_string();
if desc_str.is_empty() { if stack.starts.len() > stack.filter_data.depth {
Profiler { desc: None } return Profiler { desc: None };
} else if stack.starts.len() < stack.filter_data.depth }
&& stack.filter_data.allowed.contains(&desc_str)
{ if stack.filter_data.allowed.is_empty() || stack.filter_data.allowed.contains(desc) {
stack.starts.push(Instant::now()); stack.starts.push(Instant::now());
Profiler { desc: Some(desc_str) } Profiler { desc: Some(desc.to_string()) }
} else { } else {
Profiler { desc: None } Profiler { desc: None }
} }
@ -128,6 +134,8 @@ struct FilterData {
allowed: HashSet<String>, allowed: HashSet<String>,
} }
static PROFILING_ENABLED: AtomicBool = AtomicBool::new(false);
lazy_static! { lazy_static! {
static ref FILTER: RwLock<FilterData> = RwLock::new(Default::default()); static ref FILTER: RwLock<FilterData> = RwLock::new(Default::default());
} }