httpbis/client/
stream_handler.rs

1use crate::client::resp::ClientResponse;
2use crate::common::stream_handler::StreamHandlerInternal;
3use crate::error;
4use crate::result;
5use crate::ClientRequest;
6use crate::ErrorCode;
7use crate::Headers;
8use bytes::Bytes;
9
10/// Called once when stream is created
11pub trait ClientStreamCreatedHandler: Send + 'static {
12    /// Called when stream is created
13    fn request_created(&mut self, req: ClientRequest, resp: ClientResponse) -> crate::Result<()>;
14}
15
16/// Synchrnous callback of incoming data
17pub trait ClientResponseStreamHandler: Send + 'static {
18    /// Response HEADERS frame received
19    fn headers(&mut self, headers: Headers, end_stream: bool) -> result::Result<()>;
20    /// DATA frame received
21    fn data_frame(&mut self, data: Bytes, end_stream: bool) -> result::Result<()>;
22    /// Trailers HEADERS received
23    fn trailers(&mut self, trailers: Headers) -> result::Result<()>;
24    /// RST_STREAM frame received
25    fn rst(&mut self, error_code: ErrorCode) -> result::Result<()>;
26    /// Any other error
27    fn error(&mut self, error: error::Error) -> result::Result<()>;
28}
29
30pub(crate) struct ClientResponseStreamHandlerHolder(
31    pub(crate) Box<dyn ClientResponseStreamHandler>,
32);
33
34impl StreamHandlerInternal for ClientResponseStreamHandlerHolder {
35    fn data_frame(&mut self, data: Bytes, end_stream: bool) -> result::Result<()> {
36        self.0.data_frame(data, end_stream)
37    }
38
39    fn trailers(&mut self, trailers: Headers) -> result::Result<()> {
40        self.0.trailers(trailers)
41    }
42
43    fn rst(mut self, error_code: ErrorCode) -> result::Result<()> {
44        self.0.rst(error_code)
45    }
46
47    fn error(mut self, error: error::Error) -> result::Result<()> {
48        self.0.error(error)
49    }
50}
httpbis/client/stream_handler.rs

httpbis/client/
stream_handler.rs
