more clean up

This commit is contained in:
Lucio Franco
2019-08-18 01:30:26 -04:00
parent b2a9ab97d7
commit ef87a558e7
12 changed files with 36 additions and 33 deletions
+3 -2
View File
@@ -4,13 +4,14 @@ use futures_core::{Stream, TryStream};
use futures_util::StreamExt;
use tokio_codec::Encoder;
pub fn encode<T, U>(mut encoder: T, mut source: U) -> impl TryStream<Ok = BytesBuf, Error = Status>
pub fn encode<T, U>(mut encoder: T, source: U) -> impl TryStream<Ok = BytesBuf, Error = Status>
where
T: Encoder<Error = Status>,
U: Stream<Item = Result<T::Item, Status>> + Unpin,
U: Stream<Item = Result<T::Item, Status>>,
{
async_stream::stream! {
let mut buf = BytesMut::with_capacity(1024);
futures_util::pin_mut!(source);
loop {
match source.next().await {