use crate::client::resp::ClientResponse;
use crate::common::stream_handler::StreamHandlerInternal;
use crate::error;
use crate::result;
use crate::ClientRequest;
use crate::ErrorCode;
use crate::Headers;
use bytes::Bytes;
pub trait ClientStreamCreatedHandler: Send + 'static {
fn request_created(&mut self, req: ClientRequest, resp: ClientResponse) -> crate::Result<()>;
}
pub trait ClientResponseStreamHandler: Send + 'static {
fn headers(&mut self, headers: Headers, end_stream: bool) -> result::Result<()>;
fn data_frame(&mut self, data: Bytes, end_stream: bool) -> result::Result<()>;
fn trailers(&mut self, trailers: Headers) -> result::Result<()>;
fn rst(&mut self, error_code: ErrorCode) -> result::Result<()>;
fn error(&mut self, error: error::Error) -> result::Result<()>;
}
pub(crate) struct ClientResponseStreamHandlerHolder(
pub(crate) Box<dyn ClientResponseStreamHandler>,
);
impl StreamHandlerInternal for ClientResponseStreamHandlerHolder {
fn data_frame(&mut self, data: Bytes, end_stream: bool) -> result::Result<()> {
self.0.data_frame(data, end_stream)
}
fn trailers(&mut self, trailers: Headers) -> result::Result<()> {
self.0.trailers(trailers)
}
fn rst(mut self, error_code: ErrorCode) -> result::Result<()> {
self.0.rst(error_code)
}
fn error(mut self, error: error::Error) -> result::Result<()> {
self.0.error(error)
}
}