Skip to content

Commit f0906f5

Browse files
oli-obkoli-obk
committed
Error on using yield without also using gen on the closure
And suggest adding the `gen` to the closure or function
1 parent e8a6ccc commit f0906f5

19 files changed

+217
-62
lines changed

‎compiler/rustc_ast_lowering/messages.ftl‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,3 +161,6 @@ ast_lowering_underscore_expr_lhs_assign =
161161
.label = `_` not allowed here
162162
163163
ast_lowering_use_angle_brackets = use angle brackets instead
164+
ast_lowering_yield_in_closure =
165+
`yield` can only be used in `gen` closures or blocks
166+
.suggestion = use `gen` to make the closure a coroutine

‎compiler/rustc_ast_lowering/src/errors.rs‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -414,3 +414,12 @@ pub(crate) struct AsyncBoundOnlyForFnTraits {
414414
#[primary_span]
415415
pub span: Span,
416416
}
417+
418+
#[derive(Diagnostic)]
419+
#[diag(ast_lowering_yield_in_closure)]
420+
pub(crate) struct YieldInClosure {
421+
#[primary_span]
422+
pub span: Span,
423+
#[suggestion(code = "gen ", applicability = "maybe-incorrect", style = "verbose")]
424+
pub suggestion: Option<Span>,
425+
}

‎compiler/rustc_ast_lowering/src/expr.rs‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ use super::errors::{
88
};
99
use super::ResolverAstLoweringExt;
1010
use super::{ImplTraitContext, LoweringContext, ParamMode, ParenthesizedGenericArgs};
11+
use crate::errors::YieldInClosure;
1112
use crate::{FnDeclKind, ImplTraitPosition};
1213
use rustc_ast::ptr::P as AstP;
1314
use rustc_ast::*;
@@ -1578,7 +1579,12 @@ impl<'hir> LoweringContext<'_, 'hir> {
15781579
)
15791580
.emit();
15801581
}
1582+
if self.tcx.sess.edition().at_least_rust_2024() {
1583+
let suggestion = self.current_item.map(|s| s.shrink_to_lo());
1584+
self.dcx().emit_err(YieldInClosure { span, suggestion });
1585+
}
15811586
self.coroutine_kind = Some(hir::CoroutineKind::Coroutine(Movability::Movable));
1587+
15821588
false
15831589
}
15841590
};

‎compiler/rustc_ast_lowering/src/item.rs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
203203
body,
204204
..
205205
}) => {
206-
self.with_new_scopes(ident.span, |this| {
206+
self.with_new_scopes(span, |this| {
207207
// Note: we don't need to change the return type from `T` to
208208
// `impl Future<Output = T>` here because lower_body
209209
// only cares about the input argument patterns in the function

‎tests/ui/async-await/async-outside-of-await-issue-121096.stderr‎

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,15 @@
11
error[E0728]: `await` is only allowed inside `async` functions and blocks
22
--> $DIR/async-outside-of-await-issue-121096.rs:7:7
33
|
4-
LL | fn main() {
5-
| ---- this is not `async`
6-
...
7-
LL | }.await
8-
| ^^^^^ only allowed inside `async` functions and blocks
4+
LL | / fn main() {
5+
LL | | async {
6+
LL | | use std::ops::Add;
7+
LL | | let _ = 1.add(3);
8+
LL | | }.await
9+
| | ^^^^^ only allowed inside `async` functions and blocks
10+
LL | |
11+
LL | | }
12+
| |_- this is not `async`
913

1014
error: aborting due to 1 previous error
1115

‎tests/ui/async-await/await-keyword/incorrect-syntax-suggestions.stderr‎

Lines changed: 32 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -140,42 +140,54 @@ LL | }
140140
error[E0728]: `await` is only allowed inside `async` functions and blocks
141141
--> $DIR/incorrect-syntax-suggestions.rs:68:19
142142
|
143-
LL | fn foo13() -> Result<(), ()> {
144-
| ----- this is not `async`
145-
LL | let _ = bar().await();
146-
| ^^^^^ only allowed inside `async` functions and blocks
143+
LL | / fn foo13() -> Result<(), ()> {
144+
LL | | let _ = bar().await();
145+
| | ^^^^^ only allowed inside `async` functions and blocks
146+
LL | |
147+
LL | | Ok(())
148+
LL | | }
149+
| |_- this is not `async`
147150

148151
error[E0728]: `await` is only allowed inside `async` functions and blocks
149152
--> $DIR/incorrect-syntax-suggestions.rs:73:19
150153
|
151-
LL | fn foo14() -> Result<(), ()> {
152-
| ----- this is not `async`
153-
LL | let _ = bar().await()?;
154-
| ^^^^^ only allowed inside `async` functions and blocks
154+
LL | / fn foo14() -> Result<(), ()> {
155+
LL | | let _ = bar().await()?;
156+
| | ^^^^^ only allowed inside `async` functions and blocks
157+
LL | |
158+
LL | | Ok(())
159+
LL | | }
160+
| |_- this is not `async`
155161

156162
error[E0728]: `await` is only allowed inside `async` functions and blocks
157163
--> $DIR/incorrect-syntax-suggestions.rs:78:19
158164
|
159-
LL | fn foo15() -> Result<(), ()> {
160-
| ----- this is not `async`
161-
LL | let _ = bar().await;
162-
| ^^^^^ only allowed inside `async` functions and blocks
165+
LL | / fn foo15() -> Result<(), ()> {
166+
LL | | let _ = bar().await;
167+
| | ^^^^^ only allowed inside `async` functions and blocks
168+
LL | | Ok(())
169+
LL | | }
170+
| |_- this is not `async`
163171

164172
error[E0728]: `await` is only allowed inside `async` functions and blocks
165173
--> $DIR/incorrect-syntax-suggestions.rs:82:19
166174
|
167-
LL | fn foo16() -> Result<(), ()> {
168-
| ----- this is not `async`
169-
LL | let _ = bar().await?;
170-
| ^^^^^ only allowed inside `async` functions and blocks
175+
LL | / fn foo16() -> Result<(), ()> {
176+
LL | | let _ = bar().await?;
177+
| | ^^^^^ only allowed inside `async` functions and blocks
178+
LL | | Ok(())
179+
LL | | }
180+
| |_- this is not `async`
171181

172182
error[E0728]: `await` is only allowed inside `async` functions and blocks
173183
--> $DIR/incorrect-syntax-suggestions.rs:87:23
174184
|
175-
LL | fn foo() -> Result<(), ()> {
176-
| --- this is not `async`
177-
LL | let _ = bar().await?;
178-
| ^^^^^ only allowed inside `async` functions and blocks
185+
LL | / fn foo() -> Result<(), ()> {
186+
LL | | let _ = bar().await?;
187+
| | ^^^^^ only allowed inside `async` functions and blocks
188+
LL | | Ok(())
189+
LL | | }
190+
| |_____- this is not `async`
179191

180192
error[E0728]: `await` is only allowed inside `async` functions and blocks
181193
--> $DIR/incorrect-syntax-suggestions.rs:94:23

‎tests/ui/async-await/issues/issue-51751.stderr‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
error[E0728]: `await` is only allowed inside `async` functions and blocks
22
--> $DIR/issue-51751.rs:9:27
33
|
4-
LL | fn main() {
5-
| ---- this is not `async`
6-
LL | let result = inc(10000);
7-
LL | let finished = result.await;
8-
| ^^^^^ only allowed inside `async` functions and blocks
4+
LL | / fn main() {
5+
LL | | let result = inc(10000);
6+
LL | | let finished = result.await;
7+
| | ^^^^^ only allowed inside `async` functions and blocks
8+
LL | |
9+
LL | | }
10+
| |_- this is not `async`
911

1012
error: aborting due to 1 previous error
1113

‎tests/ui/async-await/issues/issue-62009-1.stderr‎

Lines changed: 30 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,28 +1,44 @@
11
error[E0728]: `await` is only allowed inside `async` functions and blocks
22
--> $DIR/issue-62009-1.rs:6:23
33
|
4-
LL | fn main() {
5-
| ---- this is not `async`
6-
LL | async { let (); }.await;
7-
| ^^^^^ only allowed inside `async` functions and blocks
4+
LL | / fn main() {
5+
LL | | async { let (); }.await;
6+
| | ^^^^^ only allowed inside `async` functions and blocks
7+
LL | |
8+
LL | | async {
9+
... |
10+
LL | |
11+
LL | | }
12+
| |_- this is not `async`
813

914
error[E0728]: `await` is only allowed inside `async` functions and blocks
1015
--> $DIR/issue-62009-1.rs:10:7
1116
|
12-
LL | fn main() {
13-
| ---- this is not `async`
14-
...
15-
LL | }.await;
16-
| ^^^^^ only allowed inside `async` functions and blocks
17+
LL | / fn main() {
18+
LL | | async { let (); }.await;
19+
LL | |
20+
LL | | async {
21+
LL | | let task1 = print_dur().await;
22+
LL | | }.await;
23+
| | ^^^^^ only allowed inside `async` functions and blocks
24+
... |
25+
LL | |
26+
LL | | }
27+
| |_- this is not `async`
1728

1829
error[E0728]: `await` is only allowed inside `async` functions and blocks
1930
--> $DIR/issue-62009-1.rs:12:16
2031
|
21-
LL | fn main() {
22-
| ---- this is not `async`
23-
...
24-
LL | (|_| 2333).await;
25-
| ^^^^^ only allowed inside `async` functions and blocks
32+
LL | / fn main() {
33+
LL | | async { let (); }.await;
34+
LL | |
35+
LL | | async {
36+
... |
37+
LL | | (|_| 2333).await;
38+
| | ^^^^^ only allowed inside `async` functions and blocks
39+
LL | |
40+
LL | | }
41+
| |_- this is not `async`
2642

2743
error: aborting due to 3 previous errors
2844

‎tests/ui/async-await/issues/issue-62009-2.stderr‎

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
error[E0728]: `await` is only allowed inside `async` functions and blocks
22
--> $DIR/issue-62009-2.rs:8:23
33
|
4-
LL | fn main() {
5-
| ---- this is not `async`
6-
LL | (async || 2333)().await;
7-
| ^^^^^ only allowed inside `async` functions and blocks
4+
LL | / fn main() {
5+
LL | | (async || 2333)().await;
6+
| | ^^^^^ only allowed inside `async` functions and blocks
7+
LL | |
8+
LL | | }
9+
| |_- this is not `async`
810

911
error: aborting due to 1 previous error
1012

‎tests/ui/async-await/issues/non-async-enclosing-span.stderr‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
error[E0728]: `await` is only allowed inside `async` functions and blocks
22
--> $DIR/non-async-enclosing-span.rs:9:28
33
|
4-
LL | fn main() {
5-
| ---- this is not `async`
6-
LL | let x = move || {};
7-
LL | let y = do_the_thing().await;
8-
| ^^^^^ only allowed inside `async` functions and blocks
4+
LL | / fn main() {
5+
LL | | let x = move || {};
6+
LL | | let y = do_the_thing().await;
7+
| | ^^^^^ only allowed inside `async` functions and blocks
8+
LL | |
9+
LL | | }
10+
| |_- this is not `async`
911

1012
error: aborting due to 1 previous error
1113

0 commit comments

Comments
 (0)