chain.rs - source

Files

async_trait

args.rs expand.rs lib.rs lifetime.rs parse.rs receiver.rs

bitflags

byteorder

bytes

buf

buf_impl.rs buf_mut.rs chain.rs iter.rs limit.rs mod.rs reader.rs take.rs uninit_slice.rs vec_deque.rs writer.rs

fmt

debug.rs hex.rs mod.rs

bytes.rs bytes_mut.rs lib.rs loom.rs

cfg_if

enum_primitive

futures

futures_channel

mpsc

mod.rs queue.rs sink_impl.rs

lib.rs lock.rs oneshot.rs

futures_core

task

__internal

atomic_waker.rs mod.rs

future.rs lib.rs stream.rs

futures_executor

enter.rs lib.rs local_pool.rs

futures_io

futures_macro

join.rs lib.rs select.rs

futures_sink

futures_task

arc_wake.rs future_obj.rs lib.rs noop_waker.rs spawn.rs waker.rs waker_ref.rs

futures_util

async_await

join_mod.rs mod.rs pending.rs poll.rs random.rs select_mod.rs

future

io

lock

bilock.rs mod.rs mutex.rs

sink

buffer.rs close.rs drain.rs err_into.rs fanout.rs feed.rs flush.rs map_err.rs mod.rs send.rs send_all.rs unfold.rs with.rs with_flat_map.rs

stream

task

mod.rs spawn.rs

fns.rs lib.rs never.rs unfold_state.rs

instant

lib.rs native.rs

libc

unix

linux_like

linux

arch

generic

gnu

b64

x86_64

align.rs mod.rs not_x32.rs

align.rs mod.rs

align.rs mod.rs

align.rs mod.rs

fixed_width_ints.rs lib.rs macros.rs

lock_api

lib.rs mutex.rs remutex.rs rwlock.rs

log

lib.rs macros.rs

memchr

x86

avx.rs mod.rs sse2.rs

fallback.rs iter.rs lib.rs naive.rs

mio

event

event.rs events.rs mod.rs source.rs

net

tcp

listener.rs mod.rs socket.rs stream.rs

uds

datagram.rs listener.rs mod.rs stream.rs

sys

unix

selector

epoll.rs mod.rs

uds

datagram.rs listener.rs mod.rs socketaddr.rs stream.rs

mod.rs net.rs pipe.rs sourcefd.rs tcp.rs udp.rs waker.rs

interest.rs io_source.rs lib.rs macros.rs poll.rs token.rs waker.rs

nix

num_cpus

lib.rs linux.rs

num_traits

ops

checked.rs inv.rs mod.rs mul_add.rs overflowing.rs saturating.rs wrapping.rs

bounds.rs cast.rs float.rs identities.rs int.rs lib.rs macros.rs pow.rs real.rs sign.rs

once_cell

imp_std.rs lib.rs race.rs

parking_lot

condvar.rs deadlock.rs elision.rs fair_mutex.rs lib.rs mutex.rs once.rs raw_fair_mutex.rs raw_mutex.rs raw_rwlock.rs remutex.rs rwlock.rs util.rs

parking_lot_core

thread_parker

linux.rs mod.rs

lib.rs parking_lot.rs spinwait.rs util.rs word_lock.rs

pin_project_lite

pin_utils

lib.rs projection.rs stack_pin.rs

proc_macro2

detection.rs fallback.rs lib.rs marker.rs parse.rs wrapper.rs

proc_macro_hack

error.rs iter.rs lib.rs parse.rs quote.rs

proc_macro_nested

quote

ext.rs format.rs ident_fragment.rs lib.rs runtime.rs spanned.rs to_tokens.rs

rs9p

error.rs fcall.rs lib.rs serialize.rs srv.rs utils.rs

scopeguard

signal_hook_registry

half_lock.rs lib.rs

slab

smallvec

syn

tokio

fs

future

block_on.rs maybe_done.rs mod.rs poll_fn.rs ready.rs try_join.rs

io

loom

std

atomic_ptr.rs atomic_u16.rs atomic_u32.rs atomic_u64.rs atomic_u8.rs atomic_usize.rs mod.rs mutex.rs parking_lot.rs unsafe_cell.rs

macros

cfg.rs join.rs loom.rs mod.rs pin.rs ready.rs scoped_tls.rs select.rs support.rs thread_local.rs try_join.rs

net

tcp

listener.rs mod.rs socket.rs split.rs split_owned.rs stream.rs

unix

datagram

mod.rs socket.rs

listener.rs mod.rs socketaddr.rs split.rs split_owned.rs stream.rs ucred.rs

addr.rs lookup_host.rs mod.rs udp.rs

park

either.rs mod.rs thread.rs

process

unix

driver.rs mod.rs orphan.rs reap.rs

runtime

signal

unix

ctrl_c.rs mod.rs registry.rs reusable_box.rs unix.rs

sync

task

blocking.rs local.rs mod.rs spawn.rs task_local.rs unconstrained.rs yield_now.rs

time

driver

wheel

level.rs mod.rs

entry.rs handle.rs mod.rs sleep.rs

clock.rs error.rs instant.rs interval.rs mod.rs timeout.rs

util

bit.rs error.rs linked_list.rs mod.rs rand.rs slab.rs trace.rs try_lock.rs wake.rs

blocking.rs coop.rs lib.rs

tokio_macros

entry.rs lib.rs select.rs

tokio_stream

tokio_util

unicode_xid

lib.rs tables.rs

use crate::stream_ext::Fuse;
use crate::Stream;

use core::pin::Pin;
use core::task::{Context, Poll};
use pin_project_lite::pin_project;

pin_project! {
    /// Stream returned by the [`chain`](super::StreamExt::chain) method.
    pub struct Chain<T, U> {
        #[pin]
        a: Fuse<T>,
        #[pin]
        b: U,
    }
}

impl<T, U> Chain<T, U> {
    pub(super) fn new(a: T, b: U) -> Chain<T, U>
    where
        T: Stream,
        U: Stream,
    {
        Chain { a: Fuse::new(a), b }
    }
}

impl<T, U> Stream for Chain<T, U>
where
    T: Stream,
    U: Stream<Item = T::Item>,
{
    type Item = T::Item;

    fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<T::Item>> {
        use Poll::Ready;

        let me = self.project();

        if let Some(v) = ready!(me.a.poll_next(cx)) {
            return Ready(Some(v));
        }

        me.b.poll_next(cx)
    }

    fn size_hint(&self) -> (usize, Option<usize>) {
        super::merge_size_hints(self.a.size_hint(), self.b.size_hint())
    }
}