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:
Ben Kunkle 2025-06-02 08:22:32 -05:00 committed by GitHub
parent b24f614ca3
commit f90333f92e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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"
);
}
} }