fix: collect naming infos for fields and variants for which item used (…

…#258)
cloudwego · Jun 24, 2024 · 89ea743 · 89ea743
1 parent d4be29a
commit 89ea743
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pilota-build/src/middle/context.rs b/pilota-build/src/middle/context.rs
@@ -353,7 +353,11 @@ impl ContextBuilder {
             let mut map: FxHashMap<String, Vec<DefId>> = FxHashMap::default();
             cx.nodes().iter().for_each(|(def_id, node)| {
                 if let Mode::Workspace(_) = &*cx.mode {
-                    if !cx.location_map.contains_key(def_id) {
+                    let mut item_def_id = *def_id;
+                    while !matches!(cx.node(item_def_id).unwrap().kind, NodeKind::Item(_)) {
+                        item_def_id = cx.node(item_def_id).unwrap().parent.unwrap()
+                    }
+                    if !cx.location_map.contains_key(&item_def_id) {
                         return;
                     }
                 }