Finalize PR#3299

smithy-lang · Dec 14, 2023 · dc6c4c9 · dc6c4c9
1 parent 9a24fff
commit dc6c4c9
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/rust-runtime/aws-smithy-async/src/future/pagination_stream.rs b/rust-runtime/aws-smithy-async/src/future/pagination_stream.rs
@@ -7,7 +7,7 @@
 
 use crate::future::pagination_stream::collect::sealed::Collectable;
 use std::future::Future;
-use std::pin::{pin, Pin};
+use std::pin::Pin;
 use std::task::{Context, Poll};
 
 pub mod collect;

diff --git a/rust-runtime/aws-smithy-types-convert/src/lib.rs b/rust-runtime/aws-smithy-types-convert/src/lib.rs
@@ -19,5 +19,5 @@
 #[cfg(any(feature = "convert-time", feature = "convert-chrono"))]
 pub mod date_time;
 
-#[cfg(feature = "convert-futures")]
-pub mod futures;
+#[cfg(feature = "convert-streams")]
+pub mod stream;
diff --git a/...e/aws-smithy-types-convert/src/futures.rs → ...me/aws-smithy-types-convert/src/stream.rs b/...e/aws-smithy-types-convert/src/futures.rs → ...me/aws-smithy-types-convert/src/stream.rs
@@ -36,3 +36,17 @@ where
         self.pagination_stream.poll_next(cx)
     }
 }
+
+/// Trait to convert PaginationStream into implementor of `Stream`
+pub trait PaginationStreamExt<Item> {
+    /// Convert PaginationStream into implementor of `Stream`
+    fn as_stream_01x(self) -> PaginationStreamImplStream<Item>;
+}
+
+impl<Item> PaginationStreamExt<Item> for PaginationStream<Item> {
+    fn as_stream_01x(self) -> PaginationStreamImplStream<Item> {
+        PaginationStreamImplStream {
+            pagination_stream: self,
+        }
+    }
+}