download-file 0.2.1

multi fiber download http file.
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use anyhow::Result;
use download_lib::DownloadFile;
use log::LevelFilter;
use std::path::PathBuf;
use std::time::Duration;
use structopt::StructOpt;

#[tokio::main]
async fn main() -> Result<()> {
    let opt = Opt::from_args();

    env_logger::builder()
        .filter_module("want", LevelFilter::Error)
        .filter_module("mio", LevelFilter::Error)
        .filter_module("rustls", LevelFilter::Error)
        .filter_level(LevelFilter::Trace)
        .init();

    match DownloadFile::start_download(opt.url, opt.save_path, opt.tasks, 1024 * 1024).await {
        Ok(download) => {
            let status = download.get_status();
            //  tokio::spawn(async move{
            //      while !status.is_finish() {
            //          tokio::time::sleep(Duration::from_secs(1)).await;
            //          log::info!("speed of progress:{}% {} K/s",status.get_percent_complete(),status.get_byte_sec()/1024);
            //      }
            //  });
            //
            //  while !download.is_finish() {
            //      let mut s="".to_string();
            //      std::io::stdin().read_line(&mut s).unwrap();
            //      if download.is_start() {
            //          download.suspend()
            //      }else{
            //          download.restart();
            //      }
            //  }

            while !status.is_finish() {
                tokio::time::sleep(Duration::from_secs(1)).await;
                log::info!(
                    "speed of progress:{}% {} K/s",
                    status.get_percent_complete(),
                    status.get_byte_sec() / 1024
                );
            }

            if !status.is_error() {
                log::info!(
                    "url {} download finish,save to {}",
                    status.url(),
                    download.get_real_file_path()
                );
            } else {
                log::info!(
                    "url {} download is error:{}",
                    status.url(),
                    status.get_error().unwrap()
                );
            }
        }
        Err(err) => {
            log::error!("down file fail:{}", err);
        }
    }

    Ok(())
}

// A basic example
#[derive(StructOpt, Debug)]
#[structopt(name = "basic")]
struct Opt {
    /// http url,http server need support range
    #[structopt(short = "u", long)]
    url: String,

    /// save file path
    #[structopt(short = "s", long, parse(from_os_str), default_value = "./")]
    save_path: PathBuf,

    /// number of concurrent download
    #[structopt(short = "t", long, default_value = "15")]
    tasks: u64,
}