@@ -185,21 +185,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
185
185
hir:: AsyncGeneratorKind :: Block ,
186
186
|this| this. with_new_scopes ( |this| this. lower_block_expr ( block) ) ,
187
187
) ,
188
- ExprKind :: Await ( expr) => {
189
- let dot_await_span = if expr. span . hi ( ) < e. span . hi ( ) {
190
- let span_with_whitespace = self
191
- . tcx
192
- . sess
193
- . source_map ( )
194
- . span_extend_while ( expr. span , char:: is_whitespace)
195
- . unwrap_or ( expr. span ) ;
196
- span_with_whitespace. shrink_to_hi ( ) . with_hi ( e. span . hi ( ) )
197
- } else {
198
- // this is a recovered `await expr`
199
- e. span
200
- } ;
201
- self . lower_expr_await ( dot_await_span, expr)
202
- }
188
+ ExprKind :: Await ( expr, await_kw_span) => self . lower_expr_await ( * await_kw_span, expr) ,
203
189
ExprKind :: Closure ( box Closure {
204
190
binder,
205
191
capture_clause,
@@ -710,18 +696,18 @@ impl<'hir> LoweringContext<'_, 'hir> {
710
696
/// }
711
697
/// }
712
698
/// ```
713
- fn lower_expr_await ( & mut self , dot_await_span : Span , expr : & Expr ) -> hir:: ExprKind < ' hir > {
714
- let full_span = expr. span . to ( dot_await_span ) ;
699
+ fn lower_expr_await ( & mut self , await_kw_span : Span , expr : & Expr ) -> hir:: ExprKind < ' hir > {
700
+ let full_span = expr. span . to ( await_kw_span ) ;
715
701
match self . generator_kind {
716
702
Some ( hir:: GeneratorKind :: Async ( _) ) => { }
717
703
Some ( hir:: GeneratorKind :: Gen ) | None => {
718
704
self . tcx . sess . emit_err ( AwaitOnlyInAsyncFnAndBlocks {
719
- dot_await_span ,
705
+ await_kw_span ,
720
706
item_span : self . current_item ,
721
707
} ) ;
722
708
}
723
709
}
724
- let span = self . mark_span_with_reason ( DesugaringKind :: Await , dot_await_span , None ) ;
710
+ let span = self . mark_span_with_reason ( DesugaringKind :: Await , await_kw_span , None ) ;
725
711
let gen_future_span = self . mark_span_with_reason (
726
712
DesugaringKind :: Await ,
727
713
full_span,
0 commit comments