Ensure doc link maps have one entry per module.

Author: cjgillot

compiler/rustc_metadata/src/rmeta/encoder.rs

@@ -1581,6 +1581,10 @@ impl<'a, 'tcx> EncodeContext<'a, 'tcx> {
                 let table = tcx.associated_types_for_impl_traits_in_associated_fn(def_id);
                 record_defaulted_array!(self.tables.associated_types_for_impl_traits_in_associated_fn[def_id] <- table);
             }
+            if let DefKind::Mod = tcx.def_kind(def_id) {
+                record!(self.tables.doc_link_resolutions[def_id] <- tcx.doc_link_resolutions(def_id));
+                record_array!(self.tables.doc_link_traits_in_scope[def_id] <- tcx.doc_link_traits_in_scope(def_id));
+            }
         }
 
         for (def_id, impls) in &tcx.crate_inherent_impls(()).0.inherent_impls {
@@ -1589,14 +1593,6 @@ impl<'a, 'tcx> EncodeContext<'a, 'tcx> {
                 def_id.index
             }));
         }
-
-        for (def_id, res_map) in &tcx.resolutions(()).doc_link_resolutions {
-            record!(self.tables.doc_link_resolutions[def_id.to_def_id()] <- res_map);
-        }
-
-        for (def_id, traits) in &tcx.resolutions(()).doc_link_traits_in_scope {
-            record_array!(self.tables.doc_link_traits_in_scope[def_id.to_def_id()] <- traits);
-        }
     }
 
     #[instrument(level = "trace", skip(self))]

compiler/rustc_resolve/src/late.rs

@@ -1559,12 +1559,15 @@ impl<'a, 'ast, 'ra, 'tcx> LateResolutionVisitor<'a, 'ast, 'ra, 'tcx> {
     }
 
     fn with_mod_rib<T>(&mut self, id: NodeId, f: impl FnOnce(&mut Self) -> T) -> T {
-        let module = self.r.expect_module(self.r.local_def_id(id).to_def_id());
+        let def_id = self.r.local_def_id(id);
+        let module = self.r.expect_module(def_id.to_def_id());
         // Move down in the graph.
         let orig_module = replace(&mut self.parent_scope.module, module);
         self.with_rib(ValueNS, RibKind::Module(module), |this| {
             this.with_rib(TypeNS, RibKind::Module(module), |this| {
                 let ret = f(this);
+                this.r.doc_link_resolutions.entry(def_id).or_default();
+                this.r.doc_link_traits_in_scope.entry(def_id).or_default();
                 this.parent_scope.module = orig_module;
                 ret
             })
@@ -5247,6 +5250,8 @@ impl<'ra, 'tcx> Resolver<'ra, 'tcx> {
                 BuiltinLintDiag::UnusedLabel,
             );
         }
+        self.doc_link_resolutions.entry(CRATE_DEF_ID).or_default();
+        self.doc_link_traits_in_scope.entry(CRATE_DEF_ID).or_default();
     }
 }