continuwuity/src/core/utils/stream/expect.rs

use futures::{Stream, StreamExt, TryStream};

use crate::Result;

pub trait TryExpect<'a, Item> {
	fn expect_ok(self) -> impl Stream<Item = Item> + Send + 'a;
}

impl<'a, T, Item> TryExpect<'a, Item> for T
where
	T: Stream<Item = Result<Item>> + TryStream + Send + 'a,
{
	#[inline]
	fn expect_ok(self: T) -> impl Stream<Item = Item> + Send + 'a {
		self.map(|res| res.expect("stream expectation failure"))
	}
}