Skip to content

Commit 022bb9c

Browse files
authored
Rollup merge of #133188 - maxcabrajac:walk_no_visit, r=petrochenkov
Add `visit` methods to ast nodes that already have `walk`s on ast visitors Some `walk` functions are called directly, because there were no correspondent visit functions. related to #128974 & #127615 r? `@petrochenkov`
2 parents d5dca43 + f6340f1 commit 022bb9c

File tree

2 files changed

+26
-16
lines changed

2 files changed

+26
-16
lines changed

compiler/rustc_ast/src/mut_visit.rs

+6-2
Original file line numberDiff line numberDiff line change
@@ -330,6 +330,10 @@ pub trait MutVisitor: Sized {
330330
fn visit_capture_by(&mut self, capture_by: &mut CaptureBy) {
331331
walk_capture_by(self, capture_by)
332332
}
333+
334+
fn visit_fn_ret_ty(&mut self, fn_ret_ty: &mut FnRetTy) {
335+
walk_fn_ret_ty(self, fn_ret_ty)
336+
}
333337
}
334338

335339
/// Use a map-style function (`FnOnce(T) -> T`) to overwrite a `&mut T`. Useful
@@ -609,7 +613,7 @@ fn walk_angle_bracketed_parameter_data<T: MutVisitor>(vis: &mut T, data: &mut An
609613
fn walk_parenthesized_parameter_data<T: MutVisitor>(vis: &mut T, args: &mut ParenthesizedArgs) {
610614
let ParenthesizedArgs { inputs, output, span, inputs_span } = args;
611615
visit_thin_vec(inputs, |input| vis.visit_ty(input));
612-
walk_fn_ret_ty(vis, output);
616+
vis.visit_fn_ret_ty(output);
613617
vis.visit_span(span);
614618
vis.visit_span(inputs_span);
615619
}
@@ -911,7 +915,7 @@ fn walk_fn<T: MutVisitor>(vis: &mut T, kind: FnKind<'_>) {
911915
fn walk_fn_decl<T: MutVisitor>(vis: &mut T, decl: &mut P<FnDecl>) {
912916
let FnDecl { inputs, output } = decl.deref_mut();
913917
inputs.flat_map_in_place(|param| vis.flat_map_param(param));
914-
walk_fn_ret_ty(vis, output);
918+
vis.visit_fn_ret_ty(output);
915919
}
916920

917921
fn walk_fn_ret_ty<T: MutVisitor>(vis: &mut T, fn_ret_ty: &mut FnRetTy) {

compiler/rustc_ast/src/visit.rs

+20-14
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,12 @@ pub trait Visitor<'ast>: Sized {
299299
fn visit_coroutine_kind(&mut self, _coroutine_kind: &'ast CoroutineKind) -> Self::Result {
300300
Self::Result::output()
301301
}
302+
fn visit_fn_decl(&mut self, fn_decl: &'ast FnDecl) -> Self::Result {
303+
walk_fn_decl(self, fn_decl)
304+
}
305+
fn visit_qself(&mut self, qs: &'ast Option<P<QSelf>>) -> Self::Result {
306+
walk_qself(self, qs)
307+
}
302308
}
303309

304310
pub fn walk_crate<'a, V: Visitor<'a>>(visitor: &mut V, krate: &'a Crate) -> V::Result {
@@ -434,13 +440,13 @@ impl WalkItemKind for ItemKind {
434440
body,
435441
from_glob: _,
436442
}) => {
437-
try_visit!(walk_qself(visitor, qself));
443+
try_visit!(visitor.visit_qself(qself));
438444
try_visit!(visitor.visit_path(path, *id));
439445
visit_opt!(visitor, visit_ident, rename);
440446
visit_opt!(visitor, visit_block, body);
441447
}
442448
ItemKind::DelegationMac(box DelegationMac { qself, prefix, suffixes, body }) => {
443-
try_visit!(walk_qself(visitor, qself));
449+
try_visit!(visitor.visit_qself(qself));
444450
try_visit!(visitor.visit_path(prefix, id));
445451
if let Some(suffixes) = suffixes {
446452
for (ident, rename) in suffixes {
@@ -518,10 +524,10 @@ pub fn walk_ty<'a, V: Visitor<'a>>(visitor: &mut V, typ: &'a Ty) -> V::Result {
518524
let BareFnTy { safety: _, ext: _, generic_params, decl, decl_span: _ } =
519525
&**function_declaration;
520526
walk_list!(visitor, visit_generic_param, generic_params);
521-
try_visit!(walk_fn_decl(visitor, decl));
527+
try_visit!(visitor.visit_fn_decl(decl));
522528
}
523529
TyKind::Path(maybe_qself, path) => {
524-
try_visit!(walk_qself(visitor, maybe_qself));
530+
try_visit!(visitor.visit_qself(maybe_qself));
525531
try_visit!(visitor.visit_path(path, *id));
526532
}
527533
TyKind::Pat(ty, pat) => {
@@ -652,16 +658,16 @@ pub fn walk_pat<'a, V: Visitor<'a>>(visitor: &mut V, pattern: &'a Pat) -> V::Res
652658
let Pat { id, kind, span: _, tokens: _ } = pattern;
653659
match kind {
654660
PatKind::TupleStruct(opt_qself, path, elems) => {
655-
try_visit!(walk_qself(visitor, opt_qself));
661+
try_visit!(visitor.visit_qself(opt_qself));
656662
try_visit!(visitor.visit_path(path, *id));
657663
walk_list!(visitor, visit_pat, elems);
658664
}
659665
PatKind::Path(opt_qself, path) => {
660-
try_visit!(walk_qself(visitor, opt_qself));
666+
try_visit!(visitor.visit_qself(opt_qself));
661667
try_visit!(visitor.visit_path(path, *id))
662668
}
663669
PatKind::Struct(opt_qself, path, fields, _rest) => {
664-
try_visit!(walk_qself(visitor, opt_qself));
670+
try_visit!(visitor.visit_qself(opt_qself));
665671
try_visit!(visitor.visit_path(path, *id));
666672
walk_list!(visitor, visit_pat_field, fields);
667673
}
@@ -846,13 +852,13 @@ pub fn walk_fn<'a, V: Visitor<'a>>(visitor: &mut V, kind: FnKind<'a>) -> V::Resu
846852
// Identifier and visibility are visited as a part of the item.
847853
try_visit!(visitor.visit_fn_header(header));
848854
try_visit!(visitor.visit_generics(generics));
849-
try_visit!(walk_fn_decl(visitor, decl));
855+
try_visit!(visitor.visit_fn_decl(decl));
850856
visit_opt!(visitor, visit_block, body);
851857
}
852858
FnKind::Closure(binder, coroutine_kind, decl, body) => {
853859
try_visit!(visitor.visit_closure_binder(binder));
854860
visit_opt!(visitor, visit_coroutine_kind, coroutine_kind.as_ref());
855-
try_visit!(walk_fn_decl(visitor, decl));
861+
try_visit!(visitor.visit_fn_decl(decl));
856862
try_visit!(visitor.visit_expr(body));
857863
}
858864
}
@@ -902,13 +908,13 @@ impl WalkItemKind for AssocItemKind {
902908
body,
903909
from_glob: _,
904910
}) => {
905-
try_visit!(walk_qself(visitor, qself));
911+
try_visit!(visitor.visit_qself(qself));
906912
try_visit!(visitor.visit_path(path, *id));
907913
visit_opt!(visitor, visit_ident, rename);
908914
visit_opt!(visitor, visit_block, body);
909915
}
910916
AssocItemKind::DelegationMac(box DelegationMac { qself, prefix, suffixes, body }) => {
911-
try_visit!(walk_qself(visitor, qself));
917+
try_visit!(visitor.visit_qself(qself));
912918
try_visit!(visitor.visit_path(prefix, id));
913919
if let Some(suffixes) = suffixes {
914920
for (ident, rename) in suffixes {
@@ -1023,7 +1029,7 @@ pub fn walk_inline_asm_sym<'a, V: Visitor<'a>>(
10231029
visitor: &mut V,
10241030
InlineAsmSym { id, qself, path }: &'a InlineAsmSym,
10251031
) -> V::Result {
1026-
try_visit!(walk_qself(visitor, qself));
1032+
try_visit!(visitor.visit_qself(qself));
10271033
visitor.visit_path(path, *id)
10281034
}
10291035

@@ -1055,7 +1061,7 @@ pub fn walk_expr<'a, V: Visitor<'a>>(visitor: &mut V, expression: &'a Expr) -> V
10551061
}
10561062
ExprKind::Struct(se) => {
10571063
let StructExpr { qself, path, fields, rest } = &**se;
1058-
try_visit!(walk_qself(visitor, qself));
1064+
try_visit!(visitor.visit_qself(qself));
10591065
try_visit!(visitor.visit_path(path, *id));
10601066
walk_list!(visitor, visit_expr_field, fields);
10611067
match rest {
@@ -1164,7 +1170,7 @@ pub fn walk_expr<'a, V: Visitor<'a>>(visitor: &mut V, expression: &'a Expr) -> V
11641170
}
11651171
ExprKind::Underscore => {}
11661172
ExprKind::Path(maybe_qself, path) => {
1167-
try_visit!(walk_qself(visitor, maybe_qself));
1173+
try_visit!(visitor.visit_qself(maybe_qself));
11681174
try_visit!(visitor.visit_path(path, *id));
11691175
}
11701176
ExprKind::Break(opt_label, opt_expr) => {

0 commit comments

Comments
 (0)