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

use crate::error;
use crate::result;
use crate::server::req_handler::ServerRequestStreamHandler;
use crate::server::req_handler::ServerRequestUnaryHandler;
use httpbis::ServerIncreaseInWindow;
use std::marker;

pub(crate) struct RequestHandlerUnaryToStream<M, H>
where
    H: ServerRequestUnaryHandler<M>,
    M: Send + 'static,
{
    pub(crate) increase_in_window: ServerIncreaseInWindow,
    pub(crate) handler: H,
    pub(crate) message: Option<M>,
    pub(crate) _marker: marker::PhantomData<M>,
}

impl<M, H> ServerRequestStreamHandler<M> for RequestHandlerUnaryToStream<M, H>
where
    H: ServerRequestUnaryHandler<M>,
    M: Send + 'static,
{
    fn grpc_message(&mut self, message: M, frame_size: u32) -> result::Result<()> {
        self.increase_in_window.data_frame_processed(frame_size);
        self.increase_in_window.increase_window_auto()?;
        if let Some(_) = self.message {
            return Err(error::Error::Other("more than one message in a stream"));
        }
        self.message = Some(message);
        Ok(())
    }

    fn end_stream(&mut self) -> result::Result<()> {
        match self.message.take() {
            Some(message) => self.handler.grpc_message(message),
            None => Err(error::Error::Other("no message, end of stream")),
        }
    }

    fn buffer_processed(&mut self, buffered: usize) -> result::Result<()> {
        // TODO: overflow check
        self.increase_in_window
            .increase_window_auto_above(buffered as u32)?;
        Ok(())
    }
}