grpc 0.8.3

Rust implementation of gRPC
Documentation 
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use futures::future;
use futures::stream;

use futures::stream::Stream;
use futures::stream::StreamExt;

use crate::error::Error;
use futures::channel::mpsc;

use crate::futures_grpc::GrpcStream;
use crate::proto::metadata::Metadata;

/// gRPC request options.
#[derive(Debug, Default, Clone)]
pub struct RequestOptions {
    /// Request metadata.
    pub metadata: Metadata,
    /// Request is cacheable (not implemented).
    pub cachable: bool,
}

impl RequestOptions {
    /// Default request options.
    pub fn new() -> RequestOptions {
        Default::default()
    }
}

/// Excluding initial metadata which is passed separately
pub struct StreamingRequest<T: Send + 'static>(pub GrpcStream<T>);

impl<T: Send + 'static> StreamingRequest<T> {
    // constructors

    pub fn new<S>(stream: S) -> StreamingRequest<T>
    where
        S: Stream<Item = crate::Result<T>> + Send + 'static,
    {
        StreamingRequest(Box::pin(stream))
    }

    pub fn once(item: T) -> StreamingRequest<T> {
        StreamingRequest::new(stream::once(future::ok(item)))
    }

    pub fn iter<I>(iter: I) -> StreamingRequest<T>
    where
        I: IntoIterator<Item = T>,
        I::IntoIter: Send + 'static,
    {
        StreamingRequest::new(stream::iter(iter.into_iter()).map(Ok))
    }

    pub fn mpsc() -> (StreamingRequestSender<T>, StreamingRequest<T>) {
        let (tx, rx) = mpsc::channel(0);
        let tx = StreamingRequestSender { sender: Some(tx) };
        // TODO: handle drop of sender
        //let rx = StreamingRequest::new(rx.map_err(|_: mpsc::TryRecvError| error::Error::Other("sender died")));
        let rx = StreamingRequest::new(rx.map(Ok));
        (tx, rx)
    }

    pub fn single(item: T) -> StreamingRequest<T> {
        StreamingRequest::new(stream::once(future::ok(item)))
    }

    pub fn empty() -> StreamingRequest<T> {
        StreamingRequest::new(stream::empty())
    }

    pub fn err(err: Error) -> StreamingRequest<T> {
        StreamingRequest::new(stream::once(future::err(err)))
    }
}

pub struct StreamingRequestSender<T: Send + 'static> {
    sender: Option<mpsc::Sender<T>>,
}

/*
impl<T: Send + 'static> Sink<T> for StreamingRequestSender<T> {
    type Error = crate::Error;

    fn start_send(&mut self, item: T) -> StartSend<T, error::Error> {
        match self.sender.as_mut() {
            // TODO: error
            Some(sender) => sender
                .start_send(item)
                .map_err(|_send_error| error::Error::Other("channel closed")),
            None => Err(error::Error::Other("sender closed")),
        }
    }

    fn poll_complete(&mut self) -> Poll<(), error::Error> {
        match self.sender.as_mut() {
            Some(sender) => sender
                .poll_complete()
                .map_err(|_send_error| error::Error::Other("channel closed")),
            None => Err(error::Error::Other("sender closed")),
        }
    }

    fn close(&mut self) -> Poll<(), error::Error> {
        // Close of sender doesn't end receiver stream
        self.sender.take();
        Poll::Ready(Ok(()))
    }
}
*/

impl<T: Send + 'static> Drop for StreamingRequestSender<T> {
    fn drop(&mut self) {
        // TODO: cancel
    }
}