Preserve yield position during pretty printing

2025-03-12 16:27:52 -07:00
parent edf65e735c
commit 1c0916a2b3
12 changed files with 55 additions and 18 deletions
--- a/compiler/rustc_ast/src/util/classify.rs
+++ b/compiler/rustc_ast/src/util/classify.rs
@@ -182,7 +182,7 @@ pub fn expr_trailing_brace(mut expr: &ast::Expr) -> Option<TrailingBrace<'_>> {
            | Range(_, Some(e), _)
            | Ret(Some(e))
            | Unary(_, e)
-            | Yield(Some(e))
+            | Yield(Some(e), _)
            | Yeet(Some(e))
            | Become(e) => {
                expr = e;
@@ -217,7 +217,7 @@ pub fn expr_trailing_brace(mut expr: &ast::Expr) -> Option<TrailingBrace<'_>> {
            Break(_, None)
            | Range(_, None, _)
            | Ret(None)
-            | Yield(None)
+            | Yield(None, _)
            | Array(_)
            | Call(_, _)
            | MethodCall(_)