1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
extern crate futures;
use futures::prelude::*;
use futures::stream::iter_ok;
struct Join<T, U>(T, U);
impl<T: Stream, U> Stream for Join<T, U> {
type Item = T::Item;
type Error = T::Error;
fn poll(&mut self) -> Poll<Option<T::Item>, T::Error> {
self.0.poll()
}
}
impl<T, U: Sink> Sink for Join<T, U> {
type SinkItem = U::SinkItem;
type SinkError = U::SinkError;
fn start_send(&mut self, item: U::SinkItem)
-> StartSend<U::SinkItem, U::SinkError>
{
self.1.start_send(item)
}
fn poll_complete(&mut self) -> Poll<(), U::SinkError> {
self.1.poll_complete()
}
fn close(&mut self) -> Poll<(), U::SinkError> {
self.1.close()
}
}
#[test]
fn test_split() {
let mut dest = Vec::new();
{
let j = Join(iter_ok(vec![10, 20, 30]), &mut dest);
let (sink, stream) = j.split();
let j = sink.reunite(stream).expect("test_split: reunite error");
let (sink, stream) = j.split();
sink.send_all(stream).wait().unwrap();
}
assert_eq!(dest, vec![10, 20, 30]);
}
|
