feat: 添加写入方法选项支持 mmap 和标准写入

在 64 位系统上，新增 `--write-method` 参数，允许用户选择使用 `mmap` 或 `std` 方式进行文件写入。

Author: share121

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "fast-down-cli"
-version = "2.7.1"
+version = "2.7.2"
 edition = "2024"
 license = "MIT"
 authors = ["share121 <2854631158@qq.com>"]

Cargo.toml

@@ -1,9 +1,15 @@
-use clap::{Parser, Subcommand};
+use clap::{Parser, Subcommand, ValueEnum};
 use color_eyre::{Result, eyre::ContextCompat};
 use crossterm::terminal;
 use reqwest::header::{HeaderMap, HeaderName};
 use std::{path::PathBuf, str::FromStr, time::Duration};
 
+#[derive(Debug, Clone, ValueEnum)]
+pub enum WriteMethod {
+    Mmap,
+    Std,
+}
+
 /// 超级快的下载器
 #[derive(Parser, Debug)]
 #[command(name = "fast-down")]
@@ -103,6 +109,9 @@ struct DownloadCli {
     /// 最大投机线程数
     #[arg(long, default_value_t = 3)]
     max_speculative: usize,
+    /// 写入方法 (mmap 速度快, std 兼容性好)
+    #[arg(long, default_value = "mmap")]
+    write_method: WriteMethod,
 }
 
 #[derive(Debug)]
@@ -138,6 +147,7 @@ pub struct DownloadArgs {
     pub interface: bool,
     pub ips: Vec<String>,
     pub max_speculative: usize,
+    pub write_method: WriteMethod,
 }
 
 impl Args {
@@ -179,6 +189,7 @@ impl Args {
                         interface: cli.interface,
                         ips: cli.ips,
                         max_speculative: cli.max_speculative,
+                        write_method: cli.write_method,
                     };
                     for header in cli.headers {
                         let mut parts = header.splitn(2, ':').map(|t| t.trim());

src/args.rs

@@ -1,3 +1,5 @@
+#[cfg(target_pointer_width = "64")]
+use crate::args::WriteMethod;
 use crate::{
     args::DownloadArgs,
     fmt,
@@ -232,21 +234,25 @@ pub async fn download(mut args: DownloadArgs) -> Result<()> {
     {
         return Err(err.into());
     }
+    let get_std_pusher = async {
+        let file = OpenOptions::new()
+            .create(true)
+            .write(true)
+            .read(true)
+            .truncate(false)
+            .open(&save_path)
+            .await?;
+        let file_pusher = FilePusher::new(file, info.size, args.write_buffer_size).await?;
+        Ok::<_, color_eyre::Report>(BoxPusher::new(file_pusher))
+    };
     let result = if info.fast_download {
         #[cfg(target_pointer_width = "64")]
-        let pusher = MmapFilePusher::new(&save_path, info.size).await?;
-        #[cfg(not(target_pointer_width = "64"))]
-        let pusher = {
-            let file = OpenOptions::new()
-                .create(true)
-                .write(true)
-                .read(true)
-                .truncate(false)
-                .open(&save_path)
-                .await?;
-            FilePusher::new(file, info.size, args.write_buffer_size).await?
+        let pusher = match args.write_method {
+            WriteMethod::Mmap => BoxPusher::new(MmapFilePusher::new(&save_path, info.size).await?),
+            WriteMethod::Std => get_std_pusher.await?,
         };
-        let pusher = BoxPusher::new(pusher);
+        #[cfg(not(target_pointer_width = "64"))]
+        let pusher = get_std_pusher.await?;
         download_multi(
             puller,
             pusher,
@@ -261,14 +267,7 @@ pub async fn download(mut args: DownloadArgs) -> Result<()> {
             },
         )
     } else {
-        let file = OpenOptions::new()
-            .write(true)
-            .create(true)
-            .truncate(false)
-            .open(&save_path)
-            .await?;
-        let pusher = FilePusher::new(file, info.size, args.write_buffer_size).await?;
-        let pusher = BoxPusher::new(pusher);
+        let pusher = get_std_pusher.await?;
         download_single(
             puller,
             pusher,