bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Remove {Early,Late}LintPassObjects.

Browse source 
`EarlyContextAndPass` wraps a single early lint pass. We aggregate
multiple passes into that single pass by using `EarlyLintPassObjects`.

This commit removes `EarlyLintPassObjects` by changing
`EarlyContextAndPass` into `EarlyContextAndPasses`. I.e. it just removes
a level of indirection. This makes the code simpler and slightly faster.

The commit does likewise for late lints.
This commit is contained in:
Nicholas Nethercote 2022-12-05 19:31:36 +11:00
parent e72ea1dc37
commit 8980d9a76d
2 changed files with 51 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

107
compiler/rustc_lint/src/early.rs
View file

@ -20,21 +20,23 @@ use rustc_ast::ptr::P;
use rustc_ast::visit::{self as ast_visit, Visitor}; use rustc_ast::visit::{self as ast_visit, Visitor};
use rustc_ast::{self as ast, walk_list, HasAttrs}; use rustc_ast::{self as ast, walk_list, HasAttrs};
use rustc_middle::ty::RegisteredTools; use rustc_middle::ty::RegisteredTools;
use rustc_session::lint::{BufferedEarlyLint, LintBuffer, LintPass}; use rustc_session::lint::{BufferedEarlyLint, LintBuffer};
use rustc_session::Session; use rustc_session::Session;
use rustc_span::symbol::Ident; use rustc_span::symbol::Ident;
use rustc_span::Span; use rustc_span::Span;
macro_rules! run_early_pass { ($cx:expr, $f:ident, $($args:expr),*) => ({ macro_rules! run_early_passes { ($cx:expr, $f:ident, $($args:expr),*) => ({
$cx.pass.$f(&$cx.context, $($args),*); for pass in $cx.passes.iter_mut() {
pass.$f(&$cx.context, $($args),*);
}
}) } }) }
pub struct EarlyContextAndPass<'a, T: EarlyLintPass> { pub struct EarlyContextAndPasses<'a> {
context: EarlyContext<'a>, context: EarlyContext<'a>,
pass: T, passes: Vec<EarlyLintPassObject>,
} }
impl<'a, T: EarlyLintPass> EarlyContextAndPass<'a, T> { impl<'a> EarlyContextAndPasses<'a> {
fn check_id(&mut self, id: ast::NodeId) { fn check_id(&mut self, id: ast::NodeId) {
for early_lint in self.context.buffered.take(id) { for early_lint in self.context.buffered.take(id) {
let BufferedEarlyLint { span, msg, node_id: _, lint_id, diagnostic } = early_lint; let BufferedEarlyLint { span, msg, node_id: _, lint_id, diagnostic } = early_lint;
@ -61,27 +63,27 @@ impl<'a, T: EarlyLintPass> EarlyContextAndPass<'a, T> {
self.check_id(id); self.check_id(id);
debug!("early context: enter_attrs({:?})", attrs); debug!("early context: enter_attrs({:?})", attrs);
run_early_pass!(self, enter_lint_attrs, attrs); run_early_passes!(self, enter_lint_attrs, attrs);
f(self); f(self);
debug!("early context: exit_attrs({:?})", attrs); debug!("early context: exit_attrs({:?})", attrs);
run_early_pass!(self, exit_lint_attrs, attrs); run_early_passes!(self, exit_lint_attrs, attrs);
self.context.builder.pop(push); self.context.builder.pop(push);
} }
} }
impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T> { impl<'a> ast_visit::Visitor<'a> for EarlyContextAndPasses<'a> {
fn visit_param(&mut self, param: &'a ast::Param) { fn visit_param(&mut self, param: &'a ast::Param) {
self.with_lint_attrs(param.id, &param.attrs, |cx| { self.with_lint_attrs(param.id, &param.attrs, |cx| {
run_early_pass!(cx, check_param, param); run_early_passes!(cx, check_param, param);
ast_visit::walk_param(cx, param); ast_visit::walk_param(cx, param);
}); });
} }
fn visit_item(&mut self, it: &'a ast::Item) { fn visit_item(&mut self, it: &'a ast::Item) {
self.with_lint_attrs(it.id, &it.attrs, |cx| { self.with_lint_attrs(it.id, &it.attrs, |cx| {
run_early_pass!(cx, check_item, it); run_early_passes!(cx, check_item, it);
ast_visit::walk_item(cx, it); ast_visit::walk_item(cx, it);
run_early_pass!(cx, check_item_post, it); run_early_passes!(cx, check_item_post, it);
}) })
} }
@ -92,10 +94,10 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
} }
fn visit_pat(&mut self, p: &'a ast::Pat) { fn visit_pat(&mut self, p: &'a ast::Pat) {
run_early_pass!(self, check_pat, p); run_early_passes!(self, check_pat, p);
self.check_id(p.id); self.check_id(p.id);
ast_visit::walk_pat(self, p); ast_visit::walk_pat(self, p);
run_early_pass!(self, check_pat_post, p); run_early_passes!(self, check_pat_post, p);
} }
fn visit_pat_field(&mut self, field: &'a ast::PatField) { fn visit_pat_field(&mut self, field: &'a ast::PatField) {
@ -111,7 +113,7 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
fn visit_expr(&mut self, e: &'a ast::Expr) { fn visit_expr(&mut self, e: &'a ast::Expr) {
self.with_lint_attrs(e.id, &e.attrs, |cx| { self.with_lint_attrs(e.id, &e.attrs, |cx| {
run_early_pass!(cx, check_expr, e); run_early_passes!(cx, check_expr, e);
ast_visit::walk_expr(cx, e); ast_visit::walk_expr(cx, e);
}) })
} }
@ -132,7 +134,7 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
// Note that statements get their attributes from // Note that statements get their attributes from
// the AST struct that they wrap (e.g. an item) // the AST struct that they wrap (e.g. an item)
self.with_lint_attrs(s.id, s.attrs(), |cx| { self.with_lint_attrs(s.id, s.attrs(), |cx| {
run_early_pass!(cx, check_stmt, s); run_early_passes!(cx, check_stmt, s);
cx.check_id(s.id); cx.check_id(s.id);
}); });
// The visitor for the AST struct wrapped // The visitor for the AST struct wrapped
@ -143,7 +145,7 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
} }
fn visit_fn(&mut self, fk: ast_visit::FnKind<'a>, span: Span, id: ast::NodeId) { fn visit_fn(&mut self, fk: ast_visit::FnKind<'a>, span: Span, id: ast::NodeId) {
run_early_pass!(self, check_fn, fk, span, id); run_early_passes!(self, check_fn, fk, span, id);
self.check_id(id); self.check_id(id);
ast_visit::walk_fn(self, fk); ast_visit::walk_fn(self, fk);
@ -171,37 +173,37 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
fn visit_variant(&mut self, v: &'a ast::Variant) { fn visit_variant(&mut self, v: &'a ast::Variant) {
self.with_lint_attrs(v.id, &v.attrs, |cx| { self.with_lint_attrs(v.id, &v.attrs, |cx| {
run_early_pass!(cx, check_variant, v); run_early_passes!(cx, check_variant, v);
ast_visit::walk_variant(cx, v); ast_visit::walk_variant(cx, v);
}) })
} }
fn visit_ty(&mut self, t: &'a ast::Ty) { fn visit_ty(&mut self, t: &'a ast::Ty) {
run_early_pass!(self, check_ty, t); run_early_passes!(self, check_ty, t);
self.check_id(t.id); self.check_id(t.id);
ast_visit::walk_ty(self, t); ast_visit::walk_ty(self, t);
} }
fn visit_ident(&mut self, ident: Ident) { fn visit_ident(&mut self, ident: Ident) {
run_early_pass!(self, check_ident, ident); run_early_passes!(self, check_ident, ident);
} }
fn visit_local(&mut self, l: &'a ast::Local) { fn visit_local(&mut self, l: &'a ast::Local) {
self.with_lint_attrs(l.id, &l.attrs, |cx| { self.with_lint_attrs(l.id, &l.attrs, |cx| {
run_early_pass!(cx, check_local, l); run_early_passes!(cx, check_local, l);
ast_visit::walk_local(cx, l); ast_visit::walk_local(cx, l);
}) })
} }
fn visit_block(&mut self, b: &'a ast::Block) { fn visit_block(&mut self, b: &'a ast::Block) {
run_early_pass!(self, check_block, b); run_early_passes!(self, check_block, b);
self.check_id(b.id); self.check_id(b.id);
ast_visit::walk_block(self, b); ast_visit::walk_block(self, b);
} }
fn visit_arm(&mut self, a: &'a ast::Arm) { fn visit_arm(&mut self, a: &'a ast::Arm) {
self.with_lint_attrs(a.id, &a.attrs, |cx| { self.with_lint_attrs(a.id, &a.attrs, |cx| {
run_early_pass!(cx, check_arm, a); run_early_passes!(cx, check_arm, a);
ast_visit::walk_arm(cx, a); ast_visit::walk_arm(cx, a);
}) })
} }
@ -220,19 +222,19 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
} }
fn visit_generic_arg(&mut self, arg: &'a ast::GenericArg) { fn visit_generic_arg(&mut self, arg: &'a ast::GenericArg) {
run_early_pass!(self, check_generic_arg, arg); run_early_passes!(self, check_generic_arg, arg);
ast_visit::walk_generic_arg(self, arg); ast_visit::walk_generic_arg(self, arg);
} }
fn visit_generic_param(&mut self, param: &'a ast::GenericParam) { fn visit_generic_param(&mut self, param: &'a ast::GenericParam) {
self.with_lint_attrs(param.id, &param.attrs, |cx| { self.with_lint_attrs(param.id, &param.attrs, |cx| {
run_early_pass!(cx, check_generic_param, param); run_early_passes!(cx, check_generic_param, param);
ast_visit::walk_generic_param(cx, param); ast_visit::walk_generic_param(cx, param);
}); });
} }
fn visit_generics(&mut self, g: &'a ast::Generics) { fn visit_generics(&mut self, g: &'a ast::Generics) {
run_early_pass!(self, check_generics, g); run_early_passes!(self, check_generics, g);
ast_visit::walk_generics(self, g); ast_visit::walk_generics(self, g);
} }
@ -241,18 +243,18 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
} }
fn visit_poly_trait_ref(&mut self, t: &'a ast::PolyTraitRef) { fn visit_poly_trait_ref(&mut self, t: &'a ast::PolyTraitRef) {
run_early_pass!(self, check_poly_trait_ref, t); run_early_passes!(self, check_poly_trait_ref, t);
ast_visit::walk_poly_trait_ref(self, t); ast_visit::walk_poly_trait_ref(self, t);
} }
fn visit_assoc_item(&mut self, item: &'a ast::AssocItem, ctxt: ast_visit::AssocCtxt) { fn visit_assoc_item(&mut self, item: &'a ast::AssocItem, ctxt: ast_visit::AssocCtxt) {
self.with_lint_attrs(item.id, &item.attrs, |cx| match ctxt { self.with_lint_attrs(item.id, &item.attrs, |cx| match ctxt {
ast_visit::AssocCtxt::Trait => { ast_visit::AssocCtxt::Trait => {
run_early_pass!(cx, check_trait_item, item); run_early_passes!(cx, check_trait_item, item);
ast_visit::walk_assoc_item(cx, item, ctxt); ast_visit::walk_assoc_item(cx, item, ctxt);
} }
ast_visit::AssocCtxt::Impl => { ast_visit::AssocCtxt::Impl => {
run_early_pass!(cx, check_impl_item, item); run_early_passes!(cx, check_impl_item, item);
ast_visit::walk_assoc_item(cx, item, ctxt); ast_visit::walk_assoc_item(cx, item, ctxt);
} }
}); });
@ -273,45 +275,20 @@ impl<'a, T: EarlyLintPass> ast_visit::Visitor<'a> for EarlyContextAndPass<'a, T>
} }
fn visit_attribute(&mut self, attr: &'a ast::Attribute) { fn visit_attribute(&mut self, attr: &'a ast::Attribute) {
run_early_pass!(self, check_attribute, attr); run_early_passes!(self, check_attribute, attr);
} }
fn visit_mac_def(&mut self, mac: &'a ast::MacroDef, id: ast::NodeId) { fn visit_mac_def(&mut self, mac: &'a ast::MacroDef, id: ast::NodeId) {
run_early_pass!(self, check_mac_def, mac); run_early_passes!(self, check_mac_def, mac);
self.check_id(id); self.check_id(id);
} }
fn visit_mac_call(&mut self, mac: &'a ast::MacCall) { fn visit_mac_call(&mut self, mac: &'a ast::MacCall) {
run_early_pass!(self, check_mac, mac); run_early_passes!(self, check_mac, mac);
ast_visit::walk_mac(self, mac); ast_visit::walk_mac(self, mac);
} }
} }
struct EarlyLintPassObjects<'a> {
lints: &'a mut [EarlyLintPassObject],
}
#[allow(rustc::lint_pass_impl_without_macro)]
impl LintPass for EarlyLintPassObjects<'_> {
fn name(&self) -> &'static str {
panic!()
}
}
macro_rules! early_lint_pass_impl {
([], [$($(#[$attr:meta])* fn $name:ident($($param:ident: $arg:ty),*);)*]) => (
impl EarlyLintPass for EarlyLintPassObjects<'_> {
$(fn $name(&mut self, context: &EarlyContext<'_>, $($param: $arg),*) {
for obj in self.lints.iter_mut() {
obj.$name(context, $($param),*);
}
})*
}
)
}
crate::early_lint_methods!(early_lint_pass_impl, []);
/// Early lints work on different nodes - either on the crate root, or on freshly loaded modules. /// Early lints work on different nodes - either on the crate root, or on freshly loaded modules.
/// This trait generalizes over those nodes. /// This trait generalizes over those nodes.
pub trait EarlyCheckNode<'a>: Copy { pub trait EarlyCheckNode<'a>: Copy {
@ -319,7 +296,7 @@ pub trait EarlyCheckNode<'a>: Copy {
fn attrs<'b>(self) -> &'b [ast::Attribute] fn attrs<'b>(self) -> &'b [ast::Attribute]
where where
'a: 'b; 'a: 'b;
fn check<'b>(self, cx: &mut EarlyContextAndPass<'b, impl EarlyLintPass>) fn check<'b>(self, cx: &mut EarlyContextAndPasses<'b>)
where where
'a: 'b; 'a: 'b;
} }
@ -334,13 +311,13 @@ impl<'a> EarlyCheckNode<'a> for &'a ast::Crate {
{ {
&self.attrs &self.attrs
} }
fn check<'b>(self, cx: &mut EarlyContextAndPass<'b, impl EarlyLintPass>) fn check<'b>(self, cx: &mut EarlyContextAndPasses<'b>)
where where
'a: 'b, 'a: 'b,
{ {
run_early_pass!(cx, check_crate, self); run_early_passes!(cx, check_crate, self);
ast_visit::walk_crate(cx, self); ast_visit::walk_crate(cx, self);
run_early_pass!(cx, check_crate_post, self); run_early_passes!(cx, check_crate_post, self);
} }
} }
@ -354,7 +331,7 @@ impl<'a> EarlyCheckNode<'a> for (ast::NodeId, &'a [ast::Attribute], &'a [P<ast::
{ {
self.1 self.1
} }
fn check<'b>(self, cx: &mut EarlyContextAndPass<'b, impl EarlyLintPass>) fn check<'b>(self, cx: &mut EarlyContextAndPasses<'b>)
where where
'a: 'b, 'a: 'b,
{ {
@ -374,10 +351,10 @@ pub fn check_ast_node<'a>(
) { ) {
let passes = let passes =
if pre_expansion { &lint_store.pre_expansion_passes } else { &lint_store.early_passes }; if pre_expansion { &lint_store.pre_expansion_passes } else { &lint_store.early_passes };
let mut passes: Vec<_> = passes.iter().map(|p| (p)()).collect(); let mut passes: Vec<EarlyLintPassObject> = passes.iter().map(|p| (p)()).collect();
passes.push(Box::new(builtin_lints)); passes.push(Box::new(builtin_lints));
let mut cx = EarlyContextAndPass { let mut cx = EarlyContextAndPasses {
context: EarlyContext::new( context: EarlyContext::new(
sess, sess,
!pre_expansion, !pre_expansion,
@ -385,7 +362,7 @@ pub fn check_ast_node<'a>(
registered_tools, registered_tools,
lint_buffer.unwrap_or_default(), lint_buffer.unwrap_or_default(),
), ),
pass: EarlyLintPassObjects { lints: &mut passes[..] }, passes,
}; };
cx.with_lint_attrs(check_node.id(), check_node.attrs(), |cx| check_node.check(cx)); cx.with_lint_attrs(check_node.id(), check_node.attrs(), |cx| check_node.check(cx));

44
compiler/rustc_lint/src/late.rs
View file

@ -23,7 +23,6 @@ use rustc_hir::intravisit as hir_visit;
use rustc_hir::intravisit::Visitor; use rustc_hir::intravisit::Visitor;
use rustc_middle::hir::nested_filter; use rustc_middle::hir::nested_filter;
use rustc_middle::ty::{self, TyCtxt}; use rustc_middle::ty::{self, TyCtxt};
use rustc_session::lint::LintPass;
use rustc_span::Span; use rustc_span::Span;
use std::any::Any; use std::any::Any;
@ -37,15 +36,17 @@ pub fn unerased_lint_store(tcx: TyCtxt<'_>) -> &LintStore {
} }
macro_rules! lint_callback { ($cx:expr, $f:ident, $($args:expr),*) => ({ macro_rules! lint_callback { ($cx:expr, $f:ident, $($args:expr),*) => ({
$cx.pass.$f(&$cx.context, $($args),*); for pass in $cx.passes.iter_mut() {
pass.$f(&$cx.context, $($args),*);
}
}) } }) }
struct LateContextAndPass<'tcx, T: LateLintPass<'tcx>> { struct LateContextAndPasses<'tcx> {
context: LateContext<'tcx>, context: LateContext<'tcx>,
pass: T, passes: Vec<LateLintPassObject<'tcx>>,
} }
impl<'tcx, T: LateLintPass<'tcx>> LateContextAndPass<'tcx, T> { impl<'tcx> LateContextAndPasses<'tcx> {
/// Merge the lints specified by any lint attributes into the /// Merge the lints specified by any lint attributes into the
/// current lint context, call the provided function, then reset the /// current lint context, call the provided function, then reset the
/// lints in effect to their previous state. /// lints in effect to their previous state.
@ -81,7 +82,7 @@ impl<'tcx, T: LateLintPass<'tcx>> LateContextAndPass<'tcx, T> {
} }
} }
impl<'tcx, T: LateLintPass<'tcx>> hir_visit::Visitor<'tcx> for LateContextAndPass<'tcx, T> { impl<'tcx> hir_visit::Visitor<'tcx> for LateContextAndPasses<'tcx> {
type NestedFilter = nested_filter::All; type NestedFilter = nested_filter::All;
/// Because lints are scoped lexically, we want to walk nested /// Because lints are scoped lexically, we want to walk nested
@ -301,31 +302,6 @@ impl<'tcx, T: LateLintPass<'tcx>> hir_visit::Visitor<'tcx> for LateContextAndPas
} }
} }
struct LateLintPassObjects<'a, 'tcx> {
lints: &'a mut [LateLintPassObject<'tcx>],
}
#[allow(rustc::lint_pass_impl_without_macro)]
impl LintPass for LateLintPassObjects<'_, '_> {
fn name(&self) -> &'static str {
panic!()
}
}
macro_rules! late_lint_pass_impl {
([], [$hir:tt], [$($(#[$attr:meta])* fn $name:ident($($param:ident: $arg:ty),*);)*]) => {
impl<$hir> LateLintPass<$hir> for LateLintPassObjects<'_, $hir> {
$(fn $name(&mut self, context: &LateContext<$hir>, $($param: $arg),*) {
for obj in self.lints.iter_mut() {
obj.$name(context, $($param),*);
}
})*
}
};
}
crate::late_lint_methods!(late_lint_pass_impl, [], ['tcx]);
pub(super) fn late_lint_mod<'tcx, T: LateLintPass<'tcx> + 'tcx>( pub(super) fn late_lint_mod<'tcx, T: LateLintPass<'tcx> + 'tcx>(
tcx: TyCtxt<'tcx>, tcx: TyCtxt<'tcx>,
module_def_id: LocalDefId, module_def_id: LocalDefId,
@ -346,9 +322,8 @@ pub(super) fn late_lint_mod<'tcx, T: LateLintPass<'tcx> + 'tcx>(
let mut passes: Vec<_> = let mut passes: Vec<_> =
unerased_lint_store(tcx).late_module_passes.iter().map(|pass| (pass)(tcx)).collect(); unerased_lint_store(tcx).late_module_passes.iter().map(|pass| (pass)(tcx)).collect();
passes.push(Box::new(builtin_lints)); passes.push(Box::new(builtin_lints));
let pass = LateLintPassObjects { lints: &mut passes[..] };
let mut cx = LateContextAndPass { context, pass }; let mut cx = LateContextAndPasses { context, passes };
let (module, _span, hir_id) = tcx.hir().get_module(module_def_id); let (module, _span, hir_id) = tcx.hir().get_module(module_def_id);
cx.process_mod(module, hir_id); cx.process_mod(module, hir_id);
@ -377,9 +352,8 @@ fn late_lint_crate<'tcx, T: LateLintPass<'tcx> + 'tcx>(tcx: TyCtxt<'tcx>, builti
let mut passes = let mut passes =
unerased_lint_store(tcx).late_passes.iter().map(|p| (p)(tcx)).collect::<Vec<_>>(); unerased_lint_store(tcx).late_passes.iter().map(|p| (p)(tcx)).collect::<Vec<_>>();
passes.push(Box::new(builtin_lints)); passes.push(Box::new(builtin_lints));
let pass = LateLintPassObjects { lints: &mut passes[..] };
let mut cx = LateContextAndPass { context, pass }; let mut cx = LateContextAndPasses { context, passes };
// Visit the whole crate. // Visit the whole crate.
cx.with_lint_attrs(hir::CRATE_HIR_ID, |cx| { cx.with_lint_attrs(hir::CRATE_HIR_ID, |cx| {