bytehaul 0.1.8

Async HTTP download library with resume, multi-connection, rate limiting, and checksum verification
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

pub(crate) mod request;
pub(crate) mod response;
pub(crate) mod worker;

use std::time::Duration;

use bytes::Bytes;
use http_body_util::{BodyExt, Empty};
use hyper::body::Incoming;
use hyper::Response;

use crate::error::{DownloadError, TransportError};

pub(crate) type HttpRequestBody = Empty<Bytes>;
pub(crate) type HttpResponse = Response<Incoming>;

pub(crate) async fn next_data_chunk(
	body: &mut Incoming,
	read_timeout: Duration,
) -> Result<Option<Bytes>, DownloadError> {
	loop {
		let frame = tokio::time::timeout(read_timeout, body.frame())
			.await
			.map_err(|_| DownloadError::timeout("response body timed out"))?;

		match frame {
			Some(Ok(frame)) => match frame.into_data() {
				Ok(data) => return Ok(Some(data)),
				Err(_) => continue,
			},
			Some(Err(error)) => return Err(TransportError::from(error).into()),
			None => return Ok(None),
		}
	}
}