2018-09-06 20:41:12 +08:00
|
|
|
//@ run-pass
|
|
|
|
|
2020-04-28 14:22:08 +08:00
|
|
|
//@ revisions: default nomiropt
|
|
|
|
//@[nomiropt]compile-flags: -Z mir-opt-level=0
|
|
|
|
|
2024-04-11 21:15:34 +08:00
|
|
|
#![feature(coroutines, coroutine_trait, stmt_expr_attributes)]
|
2017-07-08 07:12:44 +08:00
|
|
|
|
2023-10-20 00:06:43 +08:00
|
|
|
use std::ops::{CoroutineState, Coroutine};
|
2018-10-05 02:49:38 +08:00
|
|
|
use std::pin::Pin;
|
2017-07-08 07:12:44 +08:00
|
|
|
|
|
|
|
fn finish<T>(mut amt: usize, mut t: T) -> T::Return
|
2023-10-20 00:06:43 +08:00
|
|
|
where T: Coroutine<(), Yield = ()> + Unpin,
|
2017-07-08 07:12:44 +08:00
|
|
|
{
|
|
|
|
loop {
|
2020-01-26 03:03:10 +08:00
|
|
|
match Pin::new(&mut t).resume(()) {
|
2023-10-20 00:06:43 +08:00
|
|
|
CoroutineState::Yielded(()) => amt = amt.checked_sub(1).unwrap(),
|
|
|
|
CoroutineState::Complete(ret) => {
|
2017-07-08 07:12:44 +08:00
|
|
|
assert_eq!(amt, 0);
|
|
|
|
return ret
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
2024-04-11 21:15:34 +08:00
|
|
|
finish(1, #[coroutine] || yield);
|
|
|
|
finish(8, #[coroutine] || {
|
2017-07-08 07:12:44 +08:00
|
|
|
for _ in 0..8 {
|
|
|
|
yield;
|
|
|
|
}
|
|
|
|
});
|
2024-04-11 21:15:34 +08:00
|
|
|
finish(1, #[coroutine] || {
|
2017-07-08 07:12:44 +08:00
|
|
|
if true {
|
|
|
|
yield;
|
|
|
|
} else {
|
|
|
|
}
|
|
|
|
});
|
2024-04-11 21:15:34 +08:00
|
|
|
finish(1, #[coroutine] || {
|
2017-07-08 07:12:44 +08:00
|
|
|
if false {
|
|
|
|
} else {
|
|
|
|
yield;
|
|
|
|
}
|
|
|
|
});
|
2024-04-11 21:15:34 +08:00
|
|
|
finish(2, #[coroutine] || {
|
2017-07-08 07:12:44 +08:00
|
|
|
if { yield; false } {
|
|
|
|
yield;
|
|
|
|
panic!()
|
|
|
|
}
|
|
|
|
yield
|
|
|
|
});
|
|
|
|
}
|