Skip to content

Commit 9ceb2b4

Browse files
compiler-errorscompiler-errors
committed
Split coroutine desugaring kind from source
1 parent 767453e commit 9ceb2b4

File tree

29 files changed

+438
-229
lines changed

29 files changed

+438
-229
lines changed

‎compiler/rustc_ast_lowering/src/expr.rs

+23-12
Original file line numberDiff line numberDiff line change
@@ -670,7 +670,10 @@ impl<'hir> LoweringContext<'_, 'hir> {
670670
let params = arena_vec![self; param];
671671

672672
let body = self.lower_body(move |this| {
673-
this.coroutine_kind = Some(hir::CoroutineKind::Async(async_coroutine_source));
673+
this.coroutine_kind = Some(hir::CoroutineKind::Desugared(
674+
hir::CoroutineDesugaring::Async,
675+
async_coroutine_source,
676+
));
674677

675678
let old_ctx = this.task_context;
676679
this.task_context = Some(task_context_hid);
@@ -724,7 +727,10 @@ impl<'hir> LoweringContext<'_, 'hir> {
724727
});
725728

726729
let body = self.lower_body(move |this| {
727-
this.coroutine_kind = Some(hir::CoroutineKind::Gen(coroutine_source));
730+
this.coroutine_kind = Some(hir::CoroutineKind::Desugared(
731+
hir::CoroutineDesugaring::Gen,
732+
coroutine_source,
733+
));
728734

729735
let res = body(this);
730736
(&[], res)
@@ -802,7 +808,10 @@ impl<'hir> LoweringContext<'_, 'hir> {
802808
let params = arena_vec![self; param];
803809

804810
let body = self.lower_body(move |this| {
805-
this.coroutine_kind = Some(hir::CoroutineKind::AsyncGen(async_coroutine_source));
811+
this.coroutine_kind = Some(hir::CoroutineKind::Desugared(
812+
hir::CoroutineDesugaring::AsyncGen,
813+
async_coroutine_source,
814+
));
806815

807816
let old_ctx = this.task_context;
808817
this.task_context = Some(task_context_hid);
@@ -877,9 +886,11 @@ impl<'hir> LoweringContext<'_, 'hir> {
877886
let full_span = expr.span.to(await_kw_span);
878887

879888
let is_async_gen = match self.coroutine_kind {
880-
Some(hir::CoroutineKind::Async(_)) => false,
881-
Some(hir::CoroutineKind::AsyncGen(_)) => true,
882-
Some(hir::CoroutineKind::Coroutine) | Some(hir::CoroutineKind::Gen(_)) | None => {
889+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Async, _)) => false,
890+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::AsyncGen, _)) => true,
891+
Some(hir::CoroutineKind::Coroutine)
892+
| Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Gen, _))
893+
| None => {
883894
return hir::ExprKind::Err(self.tcx.sess.emit_err(AwaitOnlyInAsyncFnAndBlocks {
884895
await_kw_span,
885896
item_span: self.current_item,
@@ -1097,9 +1108,9 @@ impl<'hir> LoweringContext<'_, 'hir> {
10971108
Some(movability)
10981109
}
10991110
Some(
1100-
hir::CoroutineKind::Gen(_)
1101-
| hir::CoroutineKind::Async(_)
1102-
| hir::CoroutineKind::AsyncGen(_),
1111+
hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Gen, _)
1112+
| hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Async, _)
1113+
| hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::AsyncGen, _),
11031114
) => {
11041115
panic!("non-`async`/`gen` closure body turned `async`/`gen` during lowering");
11051116
}
@@ -1612,9 +1623,9 @@ impl<'hir> LoweringContext<'_, 'hir> {
16121623

16131624
fn lower_expr_yield(&mut self, span: Span, opt_expr: Option<&Expr>) -> hir::ExprKind<'hir> {
16141625
let is_async_gen = match self.coroutine_kind {
1615-
Some(hir::CoroutineKind::Gen(_)) => false,
1616-
Some(hir::CoroutineKind::AsyncGen(_)) => true,
1617-
Some(hir::CoroutineKind::Async(_)) => {
1626+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Gen, _)) => false,
1627+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::AsyncGen, _)) => true,
1628+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Async, _)) => {
16181629
return hir::ExprKind::Err(
16191630
self.tcx.sess.emit_err(AsyncCoroutinesNotSupported { span }),
16201631
);

‎compiler/rustc_borrowck/src/diagnostics/conflict_errors.rs

+15-10
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
use either::Either;
2-
use hir::PatField;
32
use rustc_data_structures::captures::Captures;
43
use rustc_data_structures::fx::FxIndexSet;
54
use rustc_errors::{
@@ -8,6 +7,7 @@ use rustc_errors::{
87
use rustc_hir as hir;
98
use rustc_hir::def::{DefKind, Res};
109
use rustc_hir::intravisit::{walk_block, walk_expr, Visitor};
10+
use rustc_hir::{CoroutineDesugaring, PatField};
1111
use rustc_hir::{CoroutineKind, CoroutineSource, LangItem};
1212
use rustc_infer::traits::ObligationCause;
1313
use rustc_middle::hir::nested_filter::OnlyBodies;
@@ -2516,27 +2516,29 @@ impl<'cx, 'tcx> MirBorrowckCtxt<'cx, 'tcx> {
25162516
};
25172517
let kind = match use_span.coroutine_kind() {
25182518
Some(coroutine_kind) => match coroutine_kind {
2519-
CoroutineKind::Gen(kind) => match kind {
2519+
CoroutineKind::Desugared(CoroutineDesugaring::Gen, kind) => match kind {
25202520
CoroutineSource::Block => "gen block",
25212521
CoroutineSource::Closure => "gen closure",
25222522
CoroutineSource::Fn => {
25232523
bug!("gen block/closure expected, but gen function found.")
25242524
}
25252525
},
2526-
CoroutineKind::AsyncGen(kind) => match kind {
2526+
CoroutineKind::Desugared(CoroutineDesugaring::AsyncGen, kind) => match kind {
25272527
CoroutineSource::Block => "async gen block",
25282528
CoroutineSource::Closure => "async gen closure",
25292529
CoroutineSource::Fn => {
25302530
bug!("gen block/closure expected, but gen function found.")
25312531
}
25322532
},
2533-
CoroutineKind::Async(async_kind) => match async_kind {
2534-
CoroutineSource::Block => "async block",
2535-
CoroutineSource::Closure => "async closure",
2536-
CoroutineSource::Fn => {
2537-
bug!("async block/closure expected, but async function found.")
2533+
CoroutineKind::Desugared(CoroutineDesugaring::Async, async_kind) => {
2534+
match async_kind {
2535+
CoroutineSource::Block => "async block",
2536+
CoroutineSource::Closure => "async closure",
2537+
CoroutineSource::Fn => {
2538+
bug!("async block/closure expected, but async function found.")
2539+
}
25382540
}
2539-
},
2541+
}
25402542
CoroutineKind::Coroutine => "coroutine",
25412543
},
25422544
None => "closure",
@@ -2566,7 +2568,10 @@ impl<'cx, 'tcx> MirBorrowckCtxt<'cx, 'tcx> {
25662568
}
25672569
ConstraintCategory::CallArgument(_) => {
25682570
fr_name.highlight_region_name(&mut err);
2569-
if matches!(use_span.coroutine_kind(), Some(CoroutineKind::Async(_))) {
2571+
if matches!(
2572+
use_span.coroutine_kind(),
2573+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Async, _))
2574+
) {
25702575
err.note(
25712576
"async blocks are not executed immediately and must either take a \
25722577
reference or ownership of outside variables they use",

‎compiler/rustc_borrowck/src/diagnostics/region_errors.rs

+4-1
Original file line numberDiff line numberDiff line change
@@ -1049,7 +1049,10 @@ impl<'a, 'tcx> MirBorrowckCtxt<'a, 'tcx> {
10491049
..
10501050
}) => {
10511051
let body = map.body(*body);
1052-
if !matches!(body.coroutine_kind, Some(hir::CoroutineKind::Async(..))) {
1052+
if !matches!(
1053+
body.coroutine_kind,
1054+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Async, _))
1055+
) {
10531056
closure_span = Some(expr.span.shrink_to_lo());
10541057
}
10551058
}

‎compiler/rustc_borrowck/src/diagnostics/region_name.rs

+36-29
Original file line numberDiff line numberDiff line change
@@ -684,39 +684,46 @@ impl<'tcx> MirBorrowckCtxt<'_, 'tcx> {
684684
hir::FnRetTy::Return(hir_ty) => (fn_decl.output.span(), Some(hir_ty)),
685685
};
686686
let mir_description = match hir.body(body).coroutine_kind {
687-
Some(hir::CoroutineKind::Async(src)) => match src {
688-
hir::CoroutineSource::Block => " of async block",
689-
hir::CoroutineSource::Closure => " of async closure",
690-
hir::CoroutineSource::Fn => {
691-
let parent_item =
692-
tcx.hir_node_by_def_id(hir.get_parent_item(mir_hir_id).def_id);
693-
let output = &parent_item
694-
.fn_decl()
695-
.expect("coroutine lowered from async fn should be in fn")
696-
.output;
697-
span = output.span();
698-
if let hir::FnRetTy::Return(ret) = output {
699-
hir_ty = Some(self.get_future_inner_return_ty(*ret));
687+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Async, src)) => {
688+
match src {
689+
hir::CoroutineSource::Block => " of async block",
690+
hir::CoroutineSource::Closure => " of async closure",
691+
hir::CoroutineSource::Fn => {
692+
let parent_item =
693+
tcx.hir_node_by_def_id(hir.get_parent_item(mir_hir_id).def_id);
694+
let output = &parent_item
695+
.fn_decl()
696+
.expect("coroutine lowered from async fn should be in fn")
697+
.output;
698+
span = output.span();
699+
if let hir::FnRetTy::Return(ret) = output {
700+
hir_ty = Some(self.get_future_inner_return_ty(*ret));
701+
}
702+
" of async function"
700703
}
701-
" of async function"
702704
}
703-
},
704-
Some(hir::CoroutineKind::Gen(src)) => match src {
705-
hir::CoroutineSource::Block => " of gen block",
706-
hir::CoroutineSource::Closure => " of gen closure",
707-
hir::CoroutineSource::Fn => {
708-
let parent_item =
709-
tcx.hir_node_by_def_id(hir.get_parent_item(mir_hir_id).def_id);
710-
let output = &parent_item
711-
.fn_decl()
712-
.expect("coroutine lowered from gen fn should be in fn")
713-
.output;
714-
span = output.span();
715-
" of gen function"
705+
}
706+
Some(hir::CoroutineKind::Desugared(hir::CoroutineDesugaring::Gen, src)) => {
707+
match src {
708+
hir::CoroutineSource::Block => " of gen block",
709+
hir::CoroutineSource::Closure => " of gen closure",
710+
hir::CoroutineSource::Fn => {
711+
let parent_item =
712+
tcx.hir_node_by_def_id(hir.get_parent_item(mir_hir_id).def_id);
713+
let output = &parent_item
714+
.fn_decl()
715+
.expect("coroutine lowered from gen fn should be in fn")
716+
.output;
717+
span = output.span();
718+
" of gen function"
719+
}
716720
}
717-
},
721+
}
718722

719-
Some(hir::CoroutineKind::AsyncGen(src)) => match src {
723+
Some(hir::CoroutineKind::Desugared(
724+
hir::CoroutineDesugaring::AsyncGen,
725+
src,
726+
)) => match src {
720727
hir::CoroutineSource::Block => " of async gen block",
721728
hir::CoroutineSource::Closure => " of async gen closure",
722729
hir::CoroutineSource::Fn => {

‎compiler/rustc_codegen_ssa/src/debuginfo/type_names.rs

+26-10
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ use rustc_data_structures::fx::FxHashSet;
1515
use rustc_data_structures::stable_hasher::{Hash64, HashStable, StableHasher};
1616
use rustc_hir::def_id::DefId;
1717
use rustc_hir::definitions::{DefPathData, DefPathDataName, DisambiguatedDefPathData};
18-
use rustc_hir::{CoroutineKind, CoroutineSource, Mutability};
18+
use rustc_hir::{CoroutineDesugaring, CoroutineKind, CoroutineSource, Mutability};
1919
use rustc_middle::ty::layout::{IntegerExt, TyAndLayout};
2020
use rustc_middle::ty::{self, ExistentialProjection, ParamEnv, Ty, TyCtxt};
2121
use rustc_middle::ty::{GenericArgKind, GenericArgsRef};
@@ -560,15 +560,31 @@ pub fn push_item_name(tcx: TyCtxt<'_>, def_id: DefId, qualified: bool, output: &
560560

561561
fn coroutine_kind_label(coroutine_kind: Option<CoroutineKind>) -> &'static str {
562562
match coroutine_kind {
563-
Some(CoroutineKind::Gen(CoroutineSource::Block)) => "gen_block",
564-
Some(CoroutineKind::Gen(CoroutineSource::Closure)) => "gen_closure",
565-
Some(CoroutineKind::Gen(CoroutineSource::Fn)) => "gen_fn",
566-
Some(CoroutineKind::Async(CoroutineSource::Block)) => "async_block",
567-
Some(CoroutineKind::Async(CoroutineSource::Closure)) => "async_closure",
568-
Some(CoroutineKind::Async(CoroutineSource::Fn)) => "async_fn",
569-
Some(CoroutineKind::AsyncGen(CoroutineSource::Block)) => "async_gen_block",
570-
Some(CoroutineKind::AsyncGen(CoroutineSource::Closure)) => "async_gen_closure",
571-
Some(CoroutineKind::AsyncGen(CoroutineSource::Fn)) => "async_gen_fn",
563+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Gen, CoroutineSource::Block)) => {
564+
"gen_block"
565+
}
566+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Gen, CoroutineSource::Closure)) => {
567+
"gen_closure"
568+
}
569+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Gen, CoroutineSource::Fn)) => "gen_fn",
570+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Async, CoroutineSource::Block)) => {
571+
"async_block"
572+
}
573+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Async, CoroutineSource::Closure)) => {
574+
"async_closure"
575+
}
576+
Some(CoroutineKind::Desugared(CoroutineDesugaring::Async, CoroutineSource::Fn)) => {
577+
"async_fn"
578+
}
579+
Some(CoroutineKind::Desugared(CoroutineDesugaring::AsyncGen, CoroutineSource::Block)) => {
580+
"async_gen_block"
581+
}
582+
Some(CoroutineKind::Desugared(CoroutineDesugaring::AsyncGen, CoroutineSource::Closure)) => {
583+
"async_gen_closure"
584+
}
585+
Some(CoroutineKind::Desugared(CoroutineDesugaring::AsyncGen, CoroutineSource::Fn)) => {
586+
"async_gen_fn"
587+
}
572588
Some(CoroutineKind::Coroutine) => "coroutine",
573589
None => "closure",
574590
}

‎compiler/rustc_const_eval/src/transform/check_consts/check.rs

+6-2
Original file line numberDiff line numberDiff line change
@@ -464,8 +464,12 @@ impl<'tcx> Visitor<'tcx> for Checker<'_, 'tcx> {
464464

465465
Rvalue::Aggregate(kind, ..) => {
466466
if let AggregateKind::Coroutine(def_id, ..) = kind.as_ref()
467-
&& let Some(coroutine_kind @ hir::CoroutineKind::Async(..)) =
468-
self.tcx.coroutine_kind(def_id)
467+
&& let Some(
468+
coroutine_kind @ hir::CoroutineKind::Desugared(
469+
hir::CoroutineDesugaring::Async,
470+
_,
471+
),
472+
) = self.tcx.coroutine_kind(def_id)
469473
{
470474
self.check_op(ops::Coroutine(coroutine_kind));
471475
}

‎compiler/rustc_const_eval/src/transform/check_consts/ops.rs

+10-2
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,11 @@ impl<'tcx> NonConstOp<'tcx> for FnCallUnstable {
359359
pub struct Coroutine(pub hir::CoroutineKind);
360360
impl<'tcx> NonConstOp<'tcx> for Coroutine {
361361
fn status_in_item(&self, _: &ConstCx<'_, 'tcx>) -> Status {
362-
if let hir::CoroutineKind::Async(hir::CoroutineSource::Block) = self.0 {
362+
if let hir::CoroutineKind::Desugared(
363+
hir::CoroutineDesugaring::Async,
364+
hir::CoroutineSource::Block,
365+
) = self.0
366+
{
363367
Status::Unstable(sym::const_async_blocks)
364368
} else {
365369
Status::Forbidden
@@ -372,7 +376,11 @@ impl<'tcx> NonConstOp<'tcx> for Coroutine {
372376
span: Span,
373377
) -> DiagnosticBuilder<'tcx, ErrorGuaranteed> {
374378
let msg = format!("{:#}s are not allowed in {}s", self.0, ccx.const_kind());
375-
if let hir::CoroutineKind::Async(hir::CoroutineSource::Block) = self.0 {
379+
if let hir::CoroutineKind::Desugared(
380+
hir::CoroutineDesugaring::Async,
381+
hir::CoroutineSource::Block,
382+
) = self.0
383+
{
376384
ccx.tcx.sess.create_feature_err(
377385
errors::UnallowedOpInConstContext { span, msg },
378386
sym::const_async_blocks,

0 commit comments

Comments
 (0)