Transform the inner expression of yield expressions (#66)

tokio-rs · Mar 12, 2022 · e1a6ae6 · e1a6ae6
1 parent cb8a508
commit e1a6ae6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/async-stream-impl/src/lib.rs b/async-stream-impl/src/lib.rs
@@ -114,6 +114,8 @@ impl VisitMut for Scrub<'_> {
             syn::Expr::Yield(yield_expr) => {
                 self.has_yielded = true;
 
+                syn::visit_mut::visit_expr_yield_mut(self, yield_expr);
+
                 let value_expr = yield_expr.expr.as_ref().unwrap_or(&self.unit);
 
                 // let ident = &self.yielder;

diff --git a/async-stream/tests/try_stream.rs b/async-stream/tests/try_stream.rs
@@ -78,3 +78,10 @@ async fn multi_try() {
         values
     );
 }
+
+#[allow(unused)]
+fn issue_65() -> impl Stream<Item = Result<u32, ()>> {
+    try_stream! {
+        yield Err(())?;
+    }
+}