Skip to content

Commit 8a6640b

Browse files
borsbors
committed
Auto merge of #127827 - compiler-errors:async-closure-closure-async, r=<try>
[EXPERIMENT] Rewrite closure-of-async to async-closure to see what the fallout is uwuwuwu
2 parents 16b5690 + 8f4f214 commit 8a6640b

File tree

6 files changed

+111
-55
lines changed

6 files changed

+111
-55
lines changed

‎compiler/rustc_ast/src/ast.rs

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1214,6 +1214,30 @@ impl Expr {
12141214
}
12151215
}
12161216

1217+
pub fn peel_uwu(&self) -> &Expr {
1218+
let mut expr = self;
1219+
loop {
1220+
match &expr.kind {
1221+
ExprKind::Block(blk, None) => {
1222+
if blk.stmts.len() == 1
1223+
&& let StmtKind::Expr(blk) = &blk.stmts[0].kind
1224+
{
1225+
expr = blk;
1226+
} else {
1227+
break;
1228+
}
1229+
}
1230+
ExprKind::Paren(paren) => {
1231+
expr = paren;
1232+
}
1233+
_ => {
1234+
break;
1235+
}
1236+
}
1237+
}
1238+
expr
1239+
}
1240+
12171241
pub fn peel_parens(&self) -> &Expr {
12181242
let mut expr = self;
12191243
while let ExprKind::Paren(inner) = &expr.kind {
@@ -1614,15 +1638,16 @@ pub struct QSelf {
16141638
}
16151639

16161640
/// A capture clause used in closures and `async` blocks.
1617-
#[derive(Clone, Copy, PartialEq, Encodable, Decodable, Debug, HashStable_Generic)]
1641+
#[derive(Clone, Copy, PartialEq, Ord, Eq, PartialOrd, Debug)]
1642+
#[derive(Encodable, Decodable, HashStable_Generic)]
16181643
pub enum CaptureBy {
1644+
/// `move` keyword was not specified.
1645+
Ref,
16191646
/// `move |x| y + x`.
16201647
Value {
16211648
/// The span of the `move` keyword.
16221649
move_kw: Span,
16231650
},
1624-
/// `move` keyword was not specified.
1625-
Ref,
16261651
}
16271652

16281653
/// Closure lifetime binder, `for<'a, 'b>` in `for<'a, 'b> |_: &'a (), _: &'b ()|`.

‎compiler/rustc_ast_lowering/src/expr.rs

Lines changed: 59 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -213,18 +213,65 @@ impl<'hir> LoweringContext<'_, 'hir> {
213213
*fn_decl_span,
214214
*fn_arg_span,
215215
),
216-
None => self.lower_expr_closure(
217-
binder,
218-
*capture_clause,
219-
e.id,
220-
hir_id,
221-
*constness,
222-
*movability,
223-
fn_decl,
224-
body,
225-
*fn_decl_span,
226-
*fn_arg_span,
227-
),
216+
None => {
217+
let peeled = body.peel_uwu();
218+
if let ast::ExprKind::Gen(
219+
gen_capture_clause,
220+
block,
221+
gen_kind @ ast::GenBlockKind::Async,
222+
span,
223+
) = &peeled.kind
224+
{
225+
let coroutine_kind = match gen_kind {
226+
GenBlockKind::Async => CoroutineKind::Async {
227+
span: *span,
228+
closure_id: peeled.node_id(),
229+
return_impl_trait_id: self.next_node_id(),
230+
},
231+
GenBlockKind::Gen => CoroutineKind::Gen {
232+
span: *span,
233+
closure_id: peeled.node_id(),
234+
return_impl_trait_id: self.next_node_id(),
235+
},
236+
GenBlockKind::AsyncGen => CoroutineKind::AsyncGen {
237+
span: *span,
238+
closure_id: peeled.node_id(),
239+
return_impl_trait_id: self.next_node_id(),
240+
},
241+
};
242+
let id = self.next_node_id();
243+
self.lower_expr_coroutine_closure(
244+
binder,
245+
(*capture_clause).max(*gen_capture_clause),
246+
e.id,
247+
hir_id,
248+
coroutine_kind,
249+
fn_decl,
250+
&ast::Expr {
251+
id,
252+
span: *span,
253+
kind: ExprKind::Block(block.clone(), None),
254+
attrs: thin_vec![],
255+
tokens: None,
256+
},
257+
*fn_decl_span,
258+
*fn_arg_span,
259+
)
260+
} else {
261+
self.lower_expr_closure(
262+
binder,
263+
*capture_clause,
264+
e.id,
265+
hir_id,
266+
*constness,
267+
*movability,
268+
fn_decl,
269+
body,
270+
*fn_decl_span,
271+
*fn_arg_span,
272+
)
273+
}
274+
}
228275
},
229276
ExprKind::Gen(capture_clause, block, genblock_kind, decl_span) => {
230277
let desugaring_kind = match genblock_kind {

‎tests/ui/async-await/async-closures/wrong-fn-kind.stderr

Lines changed: 2 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,3 @@
1-
error[E0525]: expected a closure that implements the `async Fn` trait, but this closure only implements `async FnOnce`
2-
--> $DIR/wrong-fn-kind.rs:17:20
3-
|
4-
LL | needs_async_fn(move || async move {
5-
| -------------- -^^^^^^
6-
| | |
7-
| _____|______________this closure implements `async FnOnce`, not `async Fn`
8-
| | |
9-
| | required by a bound introduced by this call
10-
LL | |
11-
LL | | println!("{x}");
12-
| | - closure is `async FnOnce` because it moves the variable `x` out of its environment
13-
LL | | });
14-
| |_____- the requirement to implement `async Fn` derives from here
15-
|
16-
note: required by a bound in `needs_async_fn`
17-
--> $DIR/wrong-fn-kind.rs:5:27
18-
|
19-
LL | fn needs_async_fn(_: impl async Fn()) {}
20-
| ^^^^^^^^^^ required by this bound in `needs_async_fn`
21-
221
error[E0596]: cannot borrow `x` as mutable, as it is a captured variable in a `Fn` closure
232
--> $DIR/wrong-fn-kind.rs:9:20
243
|
@@ -35,7 +14,6 @@ LL |
3514
LL | x += 1;
3615
| - mutable borrow occurs due to use of `x` in closure
3716

38-
error: aborting due to 2 previous errors
17+
error: aborting due to 1 previous error
3918

40-
Some errors have detailed explanations: E0525, E0596.
41-
For more information about an error, try `rustc --explain E0525`.
19+
For more information about this error, try `rustc --explain E0596`.

‎tests/ui/async-await/issue-69446-fnmut-capture.stderr

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,16 @@
1-
error: captured variable cannot escape `FnMut` closure body
2-
--> $DIR/issue-69446-fnmut-capture.rs:19:17
1+
error: async closure does not implement `FnMut` because it captures state from its environment
2+
--> $DIR/issue-69446-fnmut-capture.rs:19:9
33
|
4-
LL | let mut x = Foo;
5-
| ----- variable defined here
6-
LL | bar(move || async {
7-
| _______________-_^
8-
| | |
9-
| | inferred to be a `FnMut` closure
10-
LL | | x.foo();
11-
| | - variable captured here
12-
LL | | });
13-
| |_____^ returns an `async` block that contains a reference to a captured variable, which then escapes the closure body
4+
LL | bar(move || async {
5+
| --- ^^^^^^^
6+
| |
7+
| required by a bound introduced by this call
148
|
15-
= note: `FnMut` closures only have access to their captured variables while they are executing...
16-
= note: ...therefore, they cannot allow references to captured variables to escape
9+
note: required by a bound in `bar`
10+
--> $DIR/issue-69446-fnmut-capture.rs:12:25
11+
|
12+
LL | async fn bar<T>(_: impl FnMut() -> T)
13+
| ^^^^^^^^^^^^ required by this bound in `bar`
1714

1815
error: aborting due to 1 previous error
1916

‎tests/ui/async-await/issue-70935-complex-spans.stderr

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ note: required because it appears within the type `NotSync`
1313
LL | struct NotSync(PhantomData<*mut ()>);
1414
| ^^^^^^^
1515
= note: required for `&NotSync` to implement `Send`
16+
= note: required because it appears within the type `(&NotSync,)`
1617
note: required because it's used within this closure
1718
--> $DIR/issue-70935-complex-spans.rs:19:13
1819
|
@@ -46,6 +47,7 @@ note: required because it appears within the type `NotSync`
4647
LL | struct NotSync(PhantomData<*mut ()>);
4748
| ^^^^^^^
4849
= note: required for `&NotSync` to implement `Send`
50+
= note: required because it appears within the type `(&NotSync,)`
4951
note: required because it's used within this closure
5052
--> $DIR/issue-70935-complex-spans.rs:19:13
5153
|

‎tests/ui/impl-trait/opaque-cast-field-access-in-future.stderr

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
error[E0282]: type annotations needed
2+
--> $DIR/opaque-cast-field-access-in-future.rs:7:14
3+
|
4+
LL | &mut foo.bar;
5+
| ^^^ cannot infer type
6+
17
error[E0283]: type annotations needed
28
--> $DIR/opaque-cast-field-access-in-future.rs:22:17
39
|
@@ -6,6 +12,7 @@ LL | fn run() -> Foo<impl Future<Output = ()>> {
612
|
713
= note: cannot satisfy `_: Future`
814

9-
error: aborting due to 1 previous error
15+
error: aborting due to 2 previous errors
1016

11-
For more information about this error, try `rustc --explain E0283`.
17+
Some errors have detailed explanations: E0282, E0283.
18+
For more information about an error, try `rustc --explain E0282`.

0 commit comments

Comments
 (0)