[ChunksTimeout] Consumes the stream and return the buffered items immediately

[muhamadazmy]

[ChunksTimeout] Consumes the stream and return the buffered items immediately

Summary:
When the underlying stream is an exclusive reference (&mut stream), and we need to drop the ChunksTimout stream without losing the buffered items.

[muhamadazmy]

Use case

use std::time::Duration;

use tokio::sync::mpsc;
use tokio_stream::{StreamExt, wrappers::ReceiverStream};

#[tokio::main]
async fn main() {
    let (tx, rx) = mpsc::channel::<u64>(20);

    let mut receiver = ReceiverStream::new(rx);

    let chunked = (&mut receiver).chunks_timeout(3, Duration::from_secs(3));
    tokio::pin!(chunked);

    tx.send(10).await.unwrap();
    tx.send(20).await.unwrap();

    tokio::select! {
        Some(batch) = chunked.next() => {
            println!("Got: {:?}", batch);
        }
        _ = tokio::time::sleep(Duration::from_secs(1)) => {
            // another condition
        }
    }

    let remaining = chunked.remaining();

    // use `receiver` here
}