From b228947b6863f5864b48bb3a7f3dcca921f58d0b Mon Sep 17 00:00:00 2001 From: Aleksey Kladov Date: Sun, 14 Apr 2019 23:04:08 +0300 Subject: cleanup syntax --- crates/ra_lsp_server/src/main.rs | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) (limited to 'crates/ra_lsp_server/src') diff --git a/crates/ra_lsp_server/src/main.rs b/crates/ra_lsp_server/src/main.rs index 4646742ca..28f9985b6 100644 --- a/crates/ra_lsp_server/src/main.rs +++ b/crates/ra_lsp_server/src/main.rs @@ -12,15 +12,27 @@ fn main() -> Result<()> { Ok(ref v) if v == "1" => logger.log_to_file().directory("log").start()?, _ => logger.start()?, }; - let prof_depth = match std::env::var("RA_PROFILE_DEPTH") { - Ok(ref d) => d.parse()?, - _ => 0, - }; - let profile_allowed = match std::env::var("RA_PROFILE") { - Ok(ref p) => p.split(";").map(String::from).collect(), - _ => Vec::new(), + // Filtering syntax + // env RA_PROFILE=* // dump everything + // env RA_PROFILE=foo|bar|baz // enabled only selected entries + // env RA_PROFILE=*@3 // dump everything, up to depth 3 + let filter = match std::env::var("RA_PROFILE") { + Ok(p) => { + let mut p = p.as_str(); + let depth = if let Some(idx) = p.rfind("@") { + let depth: usize = p[idx + 1..].parse().expect("invalid profile depth"); + p = &p[..idx]; + depth + } else { + 999 + }; + let allowed = + if p == "*" { Vec::new() } else { p.split(";").map(String::from).collect() }; + ra_prof::Filter::new(depth, allowed) + } + Err(_) => ra_prof::Filter::disabled(), }; - ra_prof::set_filter(ra_prof::Filter::new(prof_depth, profile_allowed)); + ra_prof::set_filter(filter); log::info!("lifecycle: server started"); match ::std::panic::catch_unwind(main_inner) { Ok(res) => { -- cgit v1.2.3