Skip to content

Commit f9d1f92

Browse files
eholkeholk
committed
Option<CoroutineKind>
1 parent 48d5f1f commit f9d1f92

File tree

22 files changed

+117
-122
lines changed

22 files changed

+117
-122
lines changed

‎compiler/rustc_ast/src/ast.rs

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1311,7 +1311,7 @@ pub struct Closure {
13111311
pub binder: ClosureBinder,
13121312
pub capture_clause: CaptureBy,
13131313
pub constness: Const,
1314-
pub coro_kind: CoroutineKind,
1314+
pub coro_kind: Option<CoroutineKind>,
13151315
pub movability: Movability,
13161316
pub fn_decl: P<FnDecl>,
13171317
pub body: P<Expr>,
@@ -2417,8 +2417,6 @@ pub enum CoroutineKind {
24172417
Async { span: Span, closure_id: NodeId, return_impl_trait_id: NodeId },
24182418
/// `gen`, which evaluates to `impl Iterator`
24192419
Gen { span: Span, closure_id: NodeId, return_impl_trait_id: NodeId },
2420-
/// Neither `async` nor `gen`
2421-
None,
24222420
}
24232421

24242422
impl CoroutineKind {
@@ -2430,14 +2428,12 @@ impl CoroutineKind {
24302428
matches!(self, CoroutineKind::Gen { .. })
24312429
}
24322430

2433-
/// In this case this is an `async` return, the `NodeId` for the generated `impl Trait` item.
2434-
pub fn opt_return_id(self) -> Option<(NodeId, Span)> {
2431+
/// In this case this is an `async` or `gen` return, the `NodeId` for the generated `impl Trait`
2432+
/// item.
2433+
pub fn return_id(self) -> (NodeId, Span) {
24352434
match self {
24362435
CoroutineKind::Async { return_impl_trait_id, span, .. }
2437-
| CoroutineKind::Gen { return_impl_trait_id, span, .. } => {
2438-
Some((return_impl_trait_id, span))
2439-
}
2440-
CoroutineKind::None => None,
2436+
| CoroutineKind::Gen { return_impl_trait_id, span, .. } => (return_impl_trait_id, span),
24412437
}
24422438
}
24432439
}
@@ -2842,7 +2838,7 @@ pub struct FnHeader {
28422838
/// The `unsafe` keyword, if any
28432839
pub unsafety: Unsafe,
28442840
/// Whether this is `async`, `gen`, or nothing.
2845-
pub coro_kind: CoroutineKind,
2841+
pub coro_kind: Option<CoroutineKind>,
28462842
/// The `const` keyword, if any
28472843
pub constness: Const,
28482844
/// The `extern` keyword and corresponding ABI string, if any
@@ -2854,20 +2850,15 @@ impl FnHeader {
28542850
pub fn has_qualifiers(&self) -> bool {
28552851
let Self { unsafety, coro_kind, constness, ext } = self;
28562852
matches!(unsafety, Unsafe::Yes(_))
2857-
|| !matches!(coro_kind, CoroutineKind::None)
2853+
|| coro_kind.is_some()
28582854
|| matches!(constness, Const::Yes(_))
28592855
|| !matches!(ext, Extern::None)
28602856
}
28612857
}
28622858

28632859
impl Default for FnHeader {
28642860
fn default() -> FnHeader {
2865-
FnHeader {
2866-
unsafety: Unsafe::No,
2867-
coro_kind: CoroutineKind::None,
2868-
constness: Const::No,
2869-
ext: Extern::None,
2870-
}
2861+
FnHeader { unsafety: Unsafe::No, coro_kind: None, constness: Const::No, ext: Extern::None }
28712862
}
28722863
}
28732864

‎compiler/rustc_ast/src/mut_visit.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -878,7 +878,6 @@ pub fn noop_visit_coro_kind<T: MutVisitor>(coro_kind: &mut CoroutineKind, vis: &
878878
vis.visit_id(closure_id);
879879
vis.visit_id(return_impl_trait_id);
880880
}
881-
CoroutineKind::None => {}
882881
}
883882
}
884883

@@ -1173,7 +1172,7 @@ fn visit_const_item<T: MutVisitor>(
11731172
pub fn noop_visit_fn_header<T: MutVisitor>(header: &mut FnHeader, vis: &mut T) {
11741173
let FnHeader { unsafety, coro_kind, constness, ext: _ } = header;
11751174
visit_constness(constness, vis);
1176-
vis.visit_coro_kind(coro_kind);
1175+
coro_kind.as_mut().map(|coro_kind| vis.visit_coro_kind(coro_kind));
11771176
visit_unsafety(unsafety, vis);
11781177
}
11791178

@@ -1416,7 +1415,7 @@ pub fn noop_visit_expr<T: MutVisitor>(
14161415
}) => {
14171416
vis.visit_closure_binder(binder);
14181417
visit_constness(constness, vis);
1419-
vis.visit_coro_kind(coro_kind);
1418+
coro_kind.as_mut().map(|coro_kind| vis.visit_coro_kind(coro_kind));
14201419
vis.visit_capture_by(capture_clause);
14211420
vis.visit_fn_decl(fn_decl);
14221421
vis.visit_expr(body);

‎compiler/rustc_ast_lowering/src/expr.rs

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -202,8 +202,10 @@ impl<'hir> LoweringContext<'_, 'hir> {
202202
fn_decl_span,
203203
fn_arg_span,
204204
}) => match coro_kind {
205-
CoroutineKind::Async { closure_id, .. }
206-
| CoroutineKind::Gen { closure_id, .. } => self.lower_expr_async_closure(
205+
Some(
206+
CoroutineKind::Async { closure_id, .. }
207+
| CoroutineKind::Gen { closure_id, .. },
208+
) => self.lower_expr_async_closure(
207209
binder,
208210
*capture_clause,
209211
e.id,
@@ -214,7 +216,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
214216
*fn_decl_span,
215217
*fn_arg_span,
216218
),
217-
CoroutineKind::None => self.lower_expr_closure(
219+
None => self.lower_expr_closure(
218220
binder,
219221
*capture_clause,
220222
e.id,
@@ -933,13 +935,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
933935

934936
let bound_generic_params = self.lower_lifetime_binder(closure_id, generic_params);
935937
// Lower outside new scope to preserve `is_in_loop_condition`.
936-
let fn_decl = self.lower_fn_decl(
937-
decl,
938-
closure_id,
939-
fn_decl_span,
940-
FnDeclKind::Closure,
941-
CoroutineKind::None,
942-
);
938+
let fn_decl = self.lower_fn_decl(decl, closure_id, fn_decl_span, FnDeclKind::Closure, None);
943939

944940
let c = self.arena.alloc(hir::Closure {
945941
def_id: self.local_def_id(closure_id),
@@ -1054,13 +1050,8 @@ impl<'hir> LoweringContext<'_, 'hir> {
10541050
// We need to lower the declaration outside the new scope, because we
10551051
// have to conserve the state of being inside a loop condition for the
10561052
// closure argument types.
1057-
let fn_decl = self.lower_fn_decl(
1058-
&outer_decl,
1059-
closure_id,
1060-
fn_decl_span,
1061-
FnDeclKind::Closure,
1062-
CoroutineKind::None,
1063-
);
1053+
let fn_decl =
1054+
self.lower_fn_decl(&outer_decl, closure_id, fn_decl_span, FnDeclKind::Closure, None);
10641055

10651056
let c = self.arena.alloc(hir::Closure {
10661057
def_id: self.local_def_id(closure_id),

‎compiler/rustc_ast_lowering/src/item.rs

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -602,7 +602,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
602602
i.id,
603603
sig.span,
604604
FnDeclKind::ExternFn,
605-
CoroutineKind::None,
605+
None,
606606
),
607607
this.lower_fn_params_to_names(fdec),
608608
)
@@ -841,7 +841,6 @@ impl<'hir> LoweringContext<'_, 'hir> {
841841
},
842842
),
843843
AssocItemKind::Fn(box Fn { sig, generics, body, .. }) => {
844-
self.current_item = Some(i.span);
845844
let body_id = self.lower_maybe_coroutine_body(
846845
i.span,
847846
hir_id,
@@ -1025,15 +1024,16 @@ impl<'hir> LoweringContext<'_, 'hir> {
10251024
span: Span,
10261025
fn_id: hir::HirId,
10271026
decl: &FnDecl,
1028-
coro_kind: CoroutineKind,
1027+
coro_kind: Option<CoroutineKind>,
10291028
body: Option<&Block>,
10301029
) -> hir::BodyId {
1031-
let (closure_id, body) = match (coro_kind, body) {
1032-
(
1033-
CoroutineKind::Async { closure_id, .. } | CoroutineKind::Gen { closure_id, .. },
1034-
Some(body),
1035-
) => (closure_id, body),
1036-
_ => return self.lower_fn_body_block(span, decl, body),
1030+
let (Some(coro_kind), Some(body)) = (coro_kind, body) else {
1031+
return self.lower_fn_body_block(span, decl, body);
1032+
};
1033+
let closure_id = match coro_kind {
1034+
CoroutineKind::Async { closure_id, .. } | CoroutineKind::Gen { closure_id, .. } => {
1035+
closure_id
1036+
}
10371037
};
10381038

10391039
self.lower_body(|this| {
@@ -1218,7 +1218,6 @@ impl<'hir> LoweringContext<'_, 'hir> {
12181218
hir::CoroutineSource::Fn,
12191219
mkbody,
12201220
),
1221-
CoroutineKind::None => unreachable!("we must have either an async fn or a gen fn"),
12221221
};
12231222

12241223
let hir_id = this.lower_node_id(closure_id);
@@ -1235,7 +1234,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
12351234
sig: &FnSig,
12361235
id: NodeId,
12371236
kind: FnDeclKind,
1238-
coro_kind: CoroutineKind,
1237+
coro_kind: Option<CoroutineKind>,
12391238
) -> (&'hir hir::Generics<'hir>, hir::FnSig<'hir>) {
12401239
let header = self.lower_fn_header(sig.header);
12411240
let itctx = ImplTraitContext::Universal;
@@ -1247,7 +1246,7 @@ impl<'hir> LoweringContext<'_, 'hir> {
12471246
}
12481247

12491248
fn lower_fn_header(&mut self, h: FnHeader) -> hir::FnHeader {
1250-
let asyncness = if let CoroutineKind::Async { span, .. } = h.coro_kind {
1249+
let asyncness = if let Some(CoroutineKind::Async { span, .. }) = h.coro_kind {
12511250
hir::IsAsync::Async(span)
12521251
} else {
12531252
hir::IsAsync::NotAsync

‎compiler/rustc_ast_lowering/src/lib.rs

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1359,13 +1359,7 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
13591359
generic_params,
13601360
unsafety: self.lower_unsafety(f.unsafety),
13611361
abi: self.lower_extern(f.ext),
1362-
decl: self.lower_fn_decl(
1363-
&f.decl,
1364-
t.id,
1365-
t.span,
1366-
FnDeclKind::Pointer,
1367-
CoroutineKind::None,
1368-
),
1362+
decl: self.lower_fn_decl(&f.decl, t.id, t.span, FnDeclKind::Pointer, None),
13691363
param_names: self.lower_fn_params_to_names(&f.decl),
13701364
}))
13711365
}
@@ -1800,7 +1794,7 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
18001794
fn_node_id: NodeId,
18011795
fn_span: Span,
18021796
kind: FnDeclKind,
1803-
coro: CoroutineKind,
1797+
coro: Option<CoroutineKind>,
18041798
) -> &'hir hir::FnDecl<'hir> {
18051799
let c_variadic = decl.c_variadic();
18061800

@@ -1830,11 +1824,11 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
18301824
}));
18311825

18321826
let output = match coro {
1833-
CoroutineKind::Async { .. } | CoroutineKind::Gen { .. } => {
1827+
Some(coro) => {
18341828
let fn_def_id = self.local_def_id(fn_node_id);
18351829
self.lower_coroutine_fn_ret_ty(&decl.output, fn_def_id, coro, kind, fn_span)
18361830
}
1837-
CoroutineKind::None => match &decl.output {
1831+
None => match &decl.output {
18381832
FnRetTy::Ty(ty) => {
18391833
let context = if kind.return_impl_trait_allowed() {
18401834
let fn_def_id = self.local_def_id(fn_node_id);
@@ -1911,9 +1905,6 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
19111905
let opaque_ty_node_id = match coro {
19121906
CoroutineKind::Async { return_impl_trait_id, .. }
19131907
| CoroutineKind::Gen { return_impl_trait_id, .. } => return_impl_trait_id,
1914-
CoroutineKind::None => {
1915-
unreachable!("lower_coroutine_fn_ret_ty must be called with either Async or Gen")
1916-
}
19171908
};
19181909

19191910
let captured_lifetimes: Vec<_> = self
@@ -1971,7 +1962,6 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
19711962
let (symbol, lang_item) = match coro {
19721963
CoroutineKind::Async { .. } => (hir::FN_OUTPUT_NAME, hir::LangItem::Future),
19731964
CoroutineKind::Gen { .. } => (hir::ITERATOR_ITEM_NAME, hir::LangItem::Iterator),
1974-
CoroutineKind::None => panic!("attemping to lower output type of non-coroutine fn"),
19751965
};
19761966

19771967
let future_args = self.arena.alloc(hir::GenericArgs {

‎compiler/rustc_ast_passes/src/ast_validation.rs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1268,13 +1268,18 @@ impl<'a> Visitor<'a> for AstValidator<'a> {
12681268

12691269
self.check_c_variadic_type(fk);
12701270

1271-
// Functions cannot both be `const async`
1271+
// Functions cannot both be `const async` or `const gen`
12721272
if let Some(&FnHeader {
12731273
constness: Const::Yes(cspan),
1274-
coro_kind: CoroutineKind::Async { span: aspan, .. },
1274+
coro_kind:
1275+
Some(
1276+
CoroutineKind::Async { span: aspan, .. }
1277+
| CoroutineKind::Gen { span: aspan, .. },
1278+
),
12751279
..
12761280
}) = fk.header()
12771281
{
1282+
// FIXME(eholk): Report a different error for `const gen`
12781283
self.err_handler().emit_err(errors::ConstAndAsync {
12791284
spans: vec![cspan, aspan],
12801285
cspan,

‎compiler/rustc_ast_pretty/src/pprust/state.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1492,7 +1492,6 @@ impl<'a> State<'a> {
14921492

14931493
fn print_coro_kind(&mut self, coro_kind: ast::CoroutineKind) {
14941494
match coro_kind {
1495-
ast::CoroutineKind::None => {}
14961495
ast::CoroutineKind::Gen { .. } => {
14971496
self.word_nbsp("gen");
14981497
}
@@ -1691,7 +1690,7 @@ impl<'a> State<'a> {
16911690

16921691
fn print_fn_header_info(&mut self, header: ast::FnHeader) {
16931692
self.print_constness(header.constness);
1694-
self.print_coro_kind(header.coro_kind);
1693+
header.coro_kind.map(|coro_kind| self.print_coro_kind(coro_kind));
16951694
self.print_unsafety(header.unsafety);
16961695

16971696
match header.ext {

‎compiler/rustc_ast_pretty/src/pprust/state/expr.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ impl<'a> State<'a> {
423423
self.print_closure_binder(binder);
424424
self.print_constness(*constness);
425425
self.print_movability(*movability);
426-
self.print_coro_kind(*coro_kind);
426+
coro_kind.map(|coro_kind| self.print_coro_kind(coro_kind));
427427
self.print_capture_clause(*capture_clause);
428428

429429
self.print_fn_params_and_ret(fn_decl, true);

‎compiler/rustc_builtin_macros/src/test.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@ fn check_test_signature(
541541
return Err(sd.emit_err(errors::TestBadFn { span: i.span, cause: span, kind: "unsafe" }));
542542
}
543543

544-
if let ast::CoroutineKind::Async { span, .. } = f.sig.header.coro_kind {
544+
if let Some(ast::CoroutineKind::Async { span, .. }) = f.sig.header.coro_kind {
545545
return Err(sd.emit_err(errors::TestBadFn { span: i.span, cause: span, kind: "async" }));
546546
}
547547

‎compiler/rustc_expand/src/build.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -547,7 +547,7 @@ impl<'a> ExtCtxt<'a> {
547547
binder: ast::ClosureBinder::NotPresent,
548548
capture_clause: ast::CaptureBy::Ref,
549549
constness: ast::Const::No,
550-
coro_kind: ast::CoroutineKind::None,
550+
coro_kind: None,
551551
movability: ast::Movability::Movable,
552552
fn_decl,
553553
body,

0 commit comments

Comments
 (0)