bytefetch 0.2.0

A Rust library that makes HTTP file downloads easier to implement.
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use std::{sync::Arc, time::Duration};

use super::{HttpDownloaderSetupErrors, throttle::ThrottleConfig};

const DEFAULT_TASKS_COUNT: u8 = 8;
const MIN_TASKS_COUNT: u8 = 1;
const MAX_TASKS_COUNT: u8 = 64;
const DEFAULT_TIMEOUT: Duration = Duration::from_secs(10);

pub(super) struct HttpDownloadConfig {
    pub(super) tasks_count: u8,
    pub(super) split_result: Option<(u64, u64)>,
    pub(super) throttle_config: Arc<ThrottleConfig>,
    pub(super) is_new: bool,
    pub(super) timeout: Duration,
}

impl HttpDownloadConfig {
    pub(super) fn default() -> Self {
        Self {
            tasks_count: 0,
            split_result: None,
            throttle_config: Arc::new(ThrottleConfig::default()),
            is_new: true,
            timeout: DEFAULT_TIMEOUT,
        }
    }

    pub(super) fn try_set_tasks_count(
        mut self,
        tasks_count: Option<u8>,
    ) -> Result<Self, HttpDownloaderSetupErrors> {
        self.tasks_count = match tasks_count {
            Some(v) if v >= MIN_TASKS_COUNT && v <= MAX_TASKS_COUNT => v,
            Some(_) => return Err(HttpDownloaderSetupErrors::InvalidThreadsCount),
            None => DEFAULT_TASKS_COUNT,
        };
        Ok(self)
    }

    pub(super) fn set_tasks_count(mut self, task_count: u8) -> Self {
        self.tasks_count = task_count;
        self
    }

    pub(super) fn set_throttle_speed(self, throttle_speed: Option<u64>) -> Self {
        let task_speed = throttle_speed.unwrap_or_default() / self.tasks_count as u64;
        self.throttle_config.set_task_speed(task_speed);
        self
    }

    pub(super) fn set_timeout(mut self, timeout: Option<Duration>) -> Self {
        if let Some(t) = timeout {
            self.timeout = t;
        }
        self
    }

    pub(super) fn mark_resumed(mut self) -> Self {
        self.is_new = false;
        self
    }
}