borrowck nested items in dead code

This commit is contained in:
lcnr
2025-05-02 16:24:28 +00:00
parent f97b3c6044
commit ffa7d1ee5d
11 changed files with 135 additions and 64 deletions

View File

@@ -29,10 +29,10 @@ mod implied_bounds;
mod instance;
mod layout;
mod needs_drop;
mod nested_bodies;
mod opaque_types;
mod representability;
pub mod sig_types;
mod stalled_generators;
mod structural_match;
mod ty;
@@ -51,5 +51,5 @@ pub fn provide(providers: &mut Providers) {
ty::provide(providers);
instance::provide(providers);
structural_match::provide(providers);
stalled_generators::provide(providers);
nested_bodies::provide(providers);
}

View File

@@ -0,0 +1,34 @@
use rustc_hir as hir;
use rustc_hir::def_id::{DefId, LocalDefId};
use rustc_hir::intravisit::Visitor;
use rustc_middle::query::Providers;
use rustc_middle::ty::{self, TyCtxt};
fn nested_bodies_within<'tcx>(tcx: TyCtxt<'tcx>, item: LocalDefId) -> &'tcx ty::List<LocalDefId> {
let body = tcx.hir_body_owned_by(item);
let mut collector =
NestedBodiesVisitor { tcx, root_def_id: item.to_def_id(), nested_bodies: vec![] };
collector.visit_body(body);
tcx.mk_local_def_ids(&collector.nested_bodies)
}
struct NestedBodiesVisitor<'tcx> {
tcx: TyCtxt<'tcx>,
root_def_id: DefId,
nested_bodies: Vec<LocalDefId>,
}
impl<'tcx> Visitor<'tcx> for NestedBodiesVisitor<'tcx> {
fn visit_nested_body(&mut self, id: hir::BodyId) {
let body_def_id = self.tcx.hir_body_owner_def_id(id);
if self.tcx.typeck_root_def_id(body_def_id.to_def_id()) == self.root_def_id {
self.nested_bodies.push(body_def_id);
let body = self.tcx.hir_body(id);
self.visit_body(body);
}
}
}
pub(super) fn provide(providers: &mut Providers) {
*providers = Providers { nested_bodies_within, ..*providers };
}

View File

@@ -1,54 +0,0 @@
use rustc_hir as hir;
use rustc_hir::def_id::{DefId, LocalDefId};
use rustc_hir::intravisit;
use rustc_hir::intravisit::Visitor;
use rustc_middle::query::Providers;
use rustc_middle::ty::{self, TyCtxt};
fn stalled_generators_within<'tcx>(
tcx: TyCtxt<'tcx>,
item: LocalDefId,
) -> &'tcx ty::List<LocalDefId> {
if !tcx.next_trait_solver_globally() {
return ty::List::empty();
}
let body = tcx.hir_body_owned_by(item);
let mut collector =
StalledGeneratorVisitor { tcx, root_def_id: item.to_def_id(), stalled_coroutines: vec![] };
collector.visit_body(body);
tcx.mk_local_def_ids(&collector.stalled_coroutines)
}
struct StalledGeneratorVisitor<'tcx> {
tcx: TyCtxt<'tcx>,
root_def_id: DefId,
stalled_coroutines: Vec<LocalDefId>,
}
impl<'tcx> Visitor<'tcx> for StalledGeneratorVisitor<'tcx> {
fn visit_nested_body(&mut self, id: hir::BodyId) {
if self.tcx.typeck_root_def_id(self.tcx.hir_body_owner_def_id(id).to_def_id())
== self.root_def_id
{
let body = self.tcx.hir_body(id);
self.visit_body(body);
}
}
fn visit_expr(&mut self, ex: &'tcx hir::Expr<'tcx>) {
if let hir::ExprKind::Closure(&hir::Closure {
def_id,
kind: hir::ClosureKind::Coroutine(_),
..
}) = ex.kind
{
self.stalled_coroutines.push(def_id);
}
intravisit::walk_expr(self, ex);
}
}
pub(super) fn provide(providers: &mut Providers) {
*providers = Providers { stalled_generators_within, ..*providers };
}