scuffle-http 0.3.2

A high-performance HTTP server supporting HTTP/1.1, HTTP/2, and HTTP/3.
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

use bytes::{Buf, Bytes};
use h3::quic::SendStream;
use h3::server::RequestStream;
use http_body::Body;

use crate::service::{HttpService, HttpServiceFactory};

/// Copy the response body to the given stream.
pub(crate) async fn copy_response_body<S, F>(
    mut send: RequestStream<S, Bytes>,
    body: <F::Service as HttpService>::ResBody,
) -> Result<(), crate::error::HttpError<F>>
where
    F: HttpServiceFactory,
    F::Error: std::error::Error,
    <F::Service as HttpService>::Error: std::error::Error,
    S: SendStream<Bytes>,
    <F::Service as HttpService>::ResBody: http_body::Body,
    <<F::Service as HttpService>::ResBody as http_body::Body>::Error: std::error::Error,
{
    let mut body = std::pin::pin!(body);

    while let Some(frame) = std::future::poll_fn(|cx| body.as_mut().poll_frame(cx)).await {
        match frame
            .map_err(crate::error::HttpError::ResBodyError)?
            .into_data()
            .map_err(|f| f.into_trailers())
        {
            Ok(mut data) => send.send_data(data.copy_to_bytes(data.remaining())).await?,
            Err(Ok(trailers)) => {
                send.send_trailers(trailers).await?;
                return Ok(());
            }
            Err(Err(_)) => continue,
        }
    }

    send.finish().await?;

    Ok(())
}