zlog: Check crate name against filters if scope empty (#31892)
Fixes https://github.com/zed-industries/zed/discussions/29541#discussioncomment-13243073 Release Notes: - N/A *or* Added/Fixed/Improved ...
This commit is contained in:
parent
b24f614ca3
commit
f90333f92e
1 changed files with 62 additions and 14 deletions
|
@ -341,26 +341,42 @@ impl ScopeMap {
|
||||||
where
|
where
|
||||||
S: AsRef<str>,
|
S: AsRef<str>,
|
||||||
{
|
{
|
||||||
let mut enabled = None;
|
fn search<S>(map: &ScopeMap, scope: &[S; SCOPE_DEPTH_MAX]) -> Option<log::LevelFilter>
|
||||||
let mut cur_range = &self.entries[0..self.root_count];
|
where
|
||||||
let mut depth = 0;
|
S: AsRef<str>,
|
||||||
|
|
||||||
'search: while !cur_range.is_empty()
|
|
||||||
&& depth < SCOPE_DEPTH_MAX
|
|
||||||
&& scope[depth].as_ref() != ""
|
|
||||||
{
|
{
|
||||||
for entry in cur_range {
|
let mut enabled = None;
|
||||||
if entry.scope == scope[depth].as_ref() {
|
let mut cur_range = &map.entries[0..map.root_count];
|
||||||
enabled = entry.enabled.or(enabled);
|
let mut depth = 0;
|
||||||
cur_range = &self.entries[entry.descendants.clone()];
|
'search: while !cur_range.is_empty()
|
||||||
depth += 1;
|
&& depth < SCOPE_DEPTH_MAX
|
||||||
continue 'search;
|
&& scope[depth].as_ref() != ""
|
||||||
|
{
|
||||||
|
for entry in cur_range {
|
||||||
|
if entry.scope == scope[depth].as_ref() {
|
||||||
|
enabled = entry.enabled.or(enabled);
|
||||||
|
cur_range = &map.entries[entry.descendants.clone()];
|
||||||
|
depth += 1;
|
||||||
|
continue 'search;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
break 'search;
|
||||||
}
|
}
|
||||||
break 'search;
|
return enabled;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let mut enabled = search(self, scope);
|
||||||
|
|
||||||
if let Some(module_path) = module_path {
|
if let Some(module_path) = module_path {
|
||||||
|
let scope_is_empty = scope[0].as_ref().is_empty();
|
||||||
|
|
||||||
|
if enabled.is_none() && scope_is_empty {
|
||||||
|
let crate_name = private::extract_crate_name_from_module_path(module_path);
|
||||||
|
let mut crate_name_scope = [""; SCOPE_DEPTH_MAX];
|
||||||
|
crate_name_scope[0] = crate_name;
|
||||||
|
enabled = search(self, &crate_name_scope);
|
||||||
|
}
|
||||||
|
|
||||||
if !self.modules.is_empty() {
|
if !self.modules.is_empty() {
|
||||||
let crate_name = private::extract_crate_name_from_module_path(module_path);
|
let crate_name = private::extract_crate_name_from_module_path(module_path);
|
||||||
let is_scope_just_crate_name =
|
let is_scope_just_crate_name =
|
||||||
|
@ -388,6 +404,8 @@ impl ScopeMap {
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
use log::LevelFilter;
|
||||||
|
|
||||||
use crate::private::scope_new;
|
use crate::private::scope_new;
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
@ -663,6 +681,7 @@ mod tests {
|
||||||
("p.q.r", log::LevelFilter::Info), // Should be overridden by kv
|
("p.q.r", log::LevelFilter::Info), // Should be overridden by kv
|
||||||
("x.y.z", log::LevelFilter::Warn), // Not overridden
|
("x.y.z", log::LevelFilter::Warn), // Not overridden
|
||||||
("crate::module::default", log::LevelFilter::Error), // Module in default
|
("crate::module::default", log::LevelFilter::Error), // Module in default
|
||||||
|
("crate::module::user", log::LevelFilter::Off), // Module disabled in default
|
||||||
];
|
];
|
||||||
|
|
||||||
// Environment filters - these should override default but be overridden by kv
|
// Environment filters - these should override default but be overridden by kv
|
||||||
|
@ -759,6 +778,22 @@ mod tests {
|
||||||
"Default filters correctly limit log level for modules"
|
"Default filters correctly limit log level for modules"
|
||||||
);
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
map.is_enabled(&scope_new(&[""]), Some("crate::module::user"), Level::Error),
|
||||||
|
EnabledStatus::Disabled,
|
||||||
|
"Module turned off in default filters is not enabled"
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
map.is_enabled(
|
||||||
|
&scope_new(&["crate"]),
|
||||||
|
Some("crate::module::user"),
|
||||||
|
Level::Error
|
||||||
|
),
|
||||||
|
EnabledStatus::Disabled,
|
||||||
|
"Module turned off in default filters is not enabled, even with crate name as scope"
|
||||||
|
);
|
||||||
|
|
||||||
// Test non-conflicting but similar paths
|
// Test non-conflicting but similar paths
|
||||||
|
|
||||||
// Test that "a.b" and "a.b.c" don't conflict (different depth)
|
// Test that "a.b" and "a.b.c" don't conflict (different depth)
|
||||||
|
@ -789,4 +824,17 @@ mod tests {
|
||||||
"Module crate::module::default::sub should not be affected by crate::module::default filter"
|
"Module crate::module::default::sub should not be affected by crate::module::default filter"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn default_filter_crate() {
|
||||||
|
let default_filters = &[("crate", LevelFilter::Off)];
|
||||||
|
let map = scope_map_from_all(&[], &env_config::parse("").unwrap(), default_filters);
|
||||||
|
|
||||||
|
use log::Level;
|
||||||
|
assert_eq!(
|
||||||
|
map.is_enabled(&scope_new(&[""]), Some("crate::submodule"), Level::Error),
|
||||||
|
EnabledStatus::Disabled,
|
||||||
|
"crate::submodule should be disabled by disabling `crate` filter"
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue