refactor yields

Signed-off-by: ragnyll <[email protected]>
project-akri · Jan 26, 2022 · f333e78 · f333e78
1 parent 7e9cb58
commit f333e78
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/agent/src/util/device_plugin_builder.rs b/agent/src/util/device_plugin_builder.rs
@@ -141,7 +141,8 @@ impl DevicePluginBuilderInterface for DevicePluginBuilder {
 
                 async_stream::stream! {
                     loop {
-                        yield uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                        let item = uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                        yield item;
                     }
                 }
             };
@@ -271,7 +272,8 @@ pub mod tests {
 
             async_stream::stream! {
                 loop {
-                    yield uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                    let item = uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                    yield item;
                 }
             }
         };

diff --git a/agent/src/util/registration.rs b/agent/src/util/registration.rs
@@ -187,7 +187,8 @@ pub async fn internal_run_registration_server(
 
         async_stream::stream! {
             loop {
-                yield uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                let item = uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                yield item;
             }
         }
     };

diff --git a/discovery-utils/src/discovery/mod.rs b/discovery-utils/src/discovery/mod.rs
@@ -229,7 +229,8 @@ pub mod server {
 
                 async_stream::stream! {
                     loop {
-                        yield uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                        let item = uds.accept().map_ok(|(st, _)| unix_stream::UnixStream(st)).await;
+                        yield item;
                     }
                 }
             };