Skip to content

Commit c070bfb

Browse files
petrochenkovpetrochenkov
committed
resolve: Split module_map into two maps for local and extern modules
1 parent e86700d commit c070bfb

File tree

3 files changed

+65
-40
lines changed

3 files changed

+65
-40
lines changed

‎compiler/rustc_resolve/src/build_reduced_graph.rs‎

Lines changed: 30 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -102,33 +102,36 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
102102
/// or trait), then this function returns that module's resolver representation, otherwise it
103103
/// returns `None`.
104104
pub(crate) fn get_module(&mut self, def_id: DefId) -> Option<Module<'ra>> {
105-
if let module @ Some(..) = self.module_map.get(&def_id) {
106-
return module.copied();
107-
}
105+
match def_id.as_local() {
106+
Some(local_def_id) => self.local_module_map.get(&local_def_id).copied(),
107+
None => {
108+
if let module @ Some(..) = self.extern_module_map.borrow().get(&def_id) {
109+
return module.copied();
110+
}
108111

109-
if !def_id.is_local() {
110-
// Query `def_kind` is not used because query system overhead is too expensive here.
111-
let def_kind = self.cstore().def_kind_untracked(def_id);
112-
if def_kind.is_module_like() {
113-
let parent = self
114-
.tcx
115-
.opt_parent(def_id)
116-
.map(|parent_id| self.get_nearest_non_block_module(parent_id));
117-
// Query `expn_that_defined` is not used because
118-
// hashing spans in its result is expensive.
119-
let expn_id = self.cstore().expn_that_defined_untracked(def_id, self.tcx.sess);
120-
return Some(self.new_module(
121-
parent,
122-
ModuleKind::Def(def_kind, def_id, Some(self.tcx.item_name(def_id))),
123-
expn_id,
124-
self.def_span(def_id),
125-
// FIXME: Account for `#[no_implicit_prelude]` attributes.
126-
parent.is_some_and(|module| module.no_implicit_prelude),
127-
));
112+
// Query `def_kind` is not used because query system overhead is too expensive here.
113+
let def_kind = self.cstore().def_kind_untracked(def_id);
114+
if def_kind.is_module_like() {
115+
let parent = self
116+
.tcx
117+
.opt_parent(def_id)
118+
.map(|parent_id| self.get_nearest_non_block_module(parent_id));
119+
// Query `expn_that_defined` is not used because
120+
// hashing spans in its result is expensive.
121+
let expn_id = self.cstore().expn_that_defined_untracked(def_id, self.tcx.sess);
122+
return Some(self.new_extern_module(
123+
parent,
124+
ModuleKind::Def(def_kind, def_id, Some(self.tcx.item_name(def_id))),
125+
expn_id,
126+
self.def_span(def_id),
127+
// FIXME: Account for `#[no_implicit_prelude]` attributes.
128+
parent.is_some_and(|module| module.no_implicit_prelude),
129+
));
130+
}
131+
132+
None
128133
}
129134
}
130-
131-
None
132135
}
133136

134137
pub(crate) fn expn_def_scope(&mut self, expn_id: ExpnId) -> Module<'ra> {
@@ -758,7 +761,7 @@ impl<'a, 'ra, 'tcx> BuildReducedGraphVisitor<'a, 'ra, 'tcx> {
758761
if let ast::ModKind::Loaded(_, _, _, Err(_)) = mod_kind {
759762
self.r.mods_with_parse_errors.insert(def_id);
760763
}
761-
self.parent_scope.module = self.r.new_module(
764+
self.parent_scope.module = self.r.new_local_module(
762765
Some(parent),
763766
ModuleKind::Def(def_kind, def_id, Some(ident.name)),
764767
expansion.to_expn_id(),
@@ -790,7 +793,7 @@ impl<'a, 'ra, 'tcx> BuildReducedGraphVisitor<'a, 'ra, 'tcx> {
790793
ItemKind::Enum(ident, _, _) | ItemKind::Trait(box ast::Trait { ident, .. }) => {
791794
self.r.define(parent, ident, TypeNS, res, vis, sp, expansion);
792795

793-
self.parent_scope.module = self.r.new_module(
796+
self.parent_scope.module = self.r.new_local_module(
794797
Some(parent),
795798
ModuleKind::Def(def_kind, def_id, Some(ident.name)),
796799
expansion.to_expn_id(),
@@ -986,7 +989,7 @@ impl<'a, 'ra, 'tcx> BuildReducedGraphVisitor<'a, 'ra, 'tcx> {
986989
let parent = self.parent_scope.module;
987990
let expansion = self.parent_scope.expansion;
988991
if self.block_needs_anonymous_module(block) {
989-
let module = self.r.new_module(
992+
let module = self.r.new_local_module(
990993
Some(parent),
991994
ModuleKind::Block,
992995
expansion.to_expn_id(),

‎compiler/rustc_resolve/src/diagnostics.rs‎

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2159,7 +2159,16 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
21592159
.keys()
21602160
.map(|ident| ident.name)
21612161
.chain(
2162-
self.module_map
2162+
self.local_module_map
2163+
.iter()
2164+
.filter(|(_, module)| {
2165+
current_module.is_ancestor_of(**module) && current_module != **module
2166+
})
2167+
.flat_map(|(_, module)| module.kind.name()),
2168+
)
2169+
.chain(
2170+
self.extern_module_map
2171+
.borrow()
21632172
.iter()
21642173
.filter(|(_, module)| {
21652174
current_module.is_ancestor_of(**module) && current_module != **module

‎compiler/rustc_resolve/src/lib.rs‎

Lines changed: 25 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1081,7 +1081,8 @@ pub struct Resolver<'ra, 'tcx> {
10811081
/// some AST passes can generate identifiers that only resolve to local or
10821082
/// lang items.
10831083
empty_module: Module<'ra>,
1084-
module_map: FxIndexMap<DefId, Module<'ra>>,
1084+
local_module_map: FxIndexMap<LocalDefId, Module<'ra>>,
1085+
extern_module_map: RefCell<FxIndexMap<DefId, Module<'ra>>>,
10851086
binding_parent_modules: FxHashMap<NameBinding<'ra>, Module<'ra>>,
10861087

10871088
underscore_disambiguator: u32,
@@ -1265,7 +1266,6 @@ impl<'ra> ResolverArenas<'ra> {
12651266
expn_id: ExpnId,
12661267
span: Span,
12671268
no_implicit_prelude: bool,
1268-
module_map: &mut FxIndexMap<DefId, Module<'ra>>,
12691269
) -> Module<'ra> {
12701270
let (def_id, self_binding) = match kind {
12711271
ModuleKind::Def(def_kind, def_id, _) => (
@@ -1285,9 +1285,6 @@ impl<'ra> ResolverArenas<'ra> {
12851285
if def_id.is_none_or(|def_id| def_id.is_local()) {
12861286
self.local_modules.borrow_mut().push(module);
12871287
}
1288-
if let Some(def_id) = def_id {
1289-
module_map.insert(def_id, module);
1290-
}
12911288
module
12921289
}
12931290
fn local_modules(&'ra self) -> std::cell::Ref<'ra, Vec<Module<'ra>>> {
@@ -1428,22 +1425,21 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
14281425
arenas: &'ra ResolverArenas<'ra>,
14291426
) -> Resolver<'ra, 'tcx> {
14301427
let root_def_id = CRATE_DEF_ID.to_def_id();
1431-
let mut module_map = FxIndexMap::default();
1428+
let mut local_module_map = FxIndexMap::default();
14321429
let graph_root = arenas.new_module(
14331430
None,
14341431
ModuleKind::Def(DefKind::Mod, root_def_id, None),
14351432
ExpnId::root(),
14361433
crate_span,
14371434
attr::contains_name(attrs, sym::no_implicit_prelude),
1438-
&mut module_map,
14391435
);
1436+
local_module_map.insert(CRATE_DEF_ID, graph_root);
14401437
let empty_module = arenas.new_module(
14411438
None,
14421439
ModuleKind::Def(DefKind::Mod, root_def_id, None),
14431440
ExpnId::root(),
14441441
DUMMY_SP,
14451442
true,
1446-
&mut Default::default(),
14471443
);
14481444

14491445
let mut node_id_to_def_id = NodeMap::default();
@@ -1504,7 +1500,8 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
15041500
trait_map: NodeMap::default(),
15051501
underscore_disambiguator: 0,
15061502
empty_module,
1507-
module_map,
1503+
local_module_map,
1504+
extern_module_map: Default::default(),
15081505
block_map: Default::default(),
15091506
binding_parent_modules: FxHashMap::default(),
15101507
ast_transform_scopes: FxHashMap::default(),
@@ -1608,16 +1605,32 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
16081605
resolver
16091606
}
16101607

1611-
fn new_module(
1608+
fn new_local_module(
1609+
&mut self,
1610+
parent: Option<Module<'ra>>,
1611+
kind: ModuleKind,
1612+
expn_id: ExpnId,
1613+
span: Span,
1614+
no_implicit_prelude: bool,
1615+
) -> Module<'ra> {
1616+
let module = self.arenas.new_module(parent, kind, expn_id, span, no_implicit_prelude);
1617+
if let Some(def_id) = module.opt_def_id() {
1618+
self.local_module_map.insert(def_id.expect_local(), module);
1619+
}
1620+
module
1621+
}
1622+
1623+
fn new_extern_module(
16121624
&mut self,
16131625
parent: Option<Module<'ra>>,
16141626
kind: ModuleKind,
16151627
expn_id: ExpnId,
16161628
span: Span,
16171629
no_implicit_prelude: bool,
16181630
) -> Module<'ra> {
1619-
let module_map = &mut self.module_map;
1620-
self.arenas.new_module(parent, kind, expn_id, span, no_implicit_prelude, module_map)
1631+
let module = self.arenas.new_module(parent, kind, expn_id, span, no_implicit_prelude);
1632+
self.extern_module_map.borrow_mut().insert(module.def_id(), module);
1633+
module
16211634
}
16221635

16231636
fn new_local_macro(&mut self, def_id: LocalDefId, macro_data: MacroData) -> &'ra MacroData {

0 commit comments

Comments
 (0)