From cd00ebc584b121a575418221f19e517835c5d9c1 Mon Sep 17 00:00:00 2001 From: baiqing Date: Tue, 5 May 2026 12:59:01 +0800 Subject: [PATCH] =?UTF-8?q?fix(local-asr):=20=E4=BF=AE=20PR=20#262=20revie?= =?UTF-8?q?w=20=E6=8F=90=E7=9A=84=E4=B8=A4=E4=B8=AA=20bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (1) 小文件路径 try_download_range_append 的 retry 污染： 原实现用 append 模式打开 .partial。若上一次 attempt 已写了部分字节后失败， retry 拿到的还是完整 chunk → append → 文件比应有大小多 N 字节，永久损坏。 改成 truncate(effective_start) + seek 写入，每次 attempt 从干净起点重写。 小文件（≤ 32MB）每个 chunk 是整个文件，重写无成本。 (2) 多文件并发一个失败不 abort 其它 worker： 原代码只 log "signaling others to stop"，没真 set cancel flag。剩下的 workers 会继续吃带宽直到全部完成才把错误冒上来。 现在错误时主动 cancel.store(true)，再用 self_aborted 标记区分这是"我们 因错误 abort"还是"用户主动 cancel"——前者 emit Failed，后者 emit Cancelled， 避免错误被误报成用户取消。 跟踪 PR: #262 review by github-actions PR Reviewer Guide --- .../app/src-tauri/src/asr/local/download.rs | 29 ++++++++++++++----- 1 file changed, 22 insertions(+), 7 deletions(-) diff --git a/openless-all/app/src-tauri/src/asr/local/download.rs b/openless-all/app/src-tauri/src/asr/local/download.rs index e4955bd5..5a6fc91d 100644 --- a/openless-all/app/src-tauri/src/asr/local/download.rs +++ b/openless-all/app/src-tauri/src/asr/local/download.rs @@ -384,7 +384,11 @@ async fn run_download( })); } + // 区分"用户主动取消" vs "我们因为某个 worker 失败了主动 abort 其它 worker"： + // 都共用同一个 cancel AtomicBool（worker 端只看一个 flag 就够），但外层用 + // `self_aborted` 记是哪种情况，决定最后 emit Cancelled 还是 Failed。 let mut first_err: Option = None; + let mut self_aborted = false; while let Some(joined) = futs.next().await { match joined { Ok(Ok(())) => {} @@ -392,9 +396,12 @@ async fn run_download( if first_err.is_none() { first_err = Some(e); } - // 取消其它正在跑的：cancel flag 设为 true，让所有 spawn task 早退 + // 一个 worker 失败 → 让其它 worker 立即停，免得它们继续吃带宽 + // 然后用户还得等到所有任务完成才看到失败。 if !cancel.load(Ordering::SeqCst) { - log::warn!("[local-asr] one file failed; signaling others to stop"); + log::warn!("[local-asr] one file failed; aborting other workers"); + cancel.store(true, Ordering::SeqCst); + self_aborted = true; } } Err(e) => { @@ -405,7 +412,8 @@ async fn run_download( } } - if cancel.load(Ordering::SeqCst) { + // 用户主动 cancel（不是我们因为错误自己 set 的）→ Cancelled + if cancel.load(Ordering::SeqCst) && !self_aborted { emit_cancelled(app, model_id, "", 0, file_count, total_bytes); return Ok(()); } @@ -684,17 +692,24 @@ async fn try_download_range_append( if status.as_u16() != 200 && status.as_u16() != 206 { anyhow::bail!("HTTP {status} for {url}"); } - if status.as_u16() == 200 && range_start > 0 { - let _ = std::fs::remove_file(partial); - } let effective_start = if status.as_u16() == 200 { 0 } else { range_start }; + // 截断 partial 到本次 attempt 的起点，再 seek 写入。 + // 老 append 实现的 bug：若上一次 attempt 已写了部分字节后失败，retry 拿到的还是 + // 完整 chunk → append → 文件比应有大小多 N 字节 → 永久损坏。 + // 小文件路径每个 chunk 是整个文件（≤ 32MB），用 truncate 重写最直白。 let mut file = tokio::fs::OpenOptions::new() .create(true) - .append(true) + .write(true) .open(partial) .await .with_context(|| format!("open partial failed: {}", partial.display()))?; + file.set_len(effective_start) + .await + .with_context(|| format!("truncate partial failed: {}", partial.display()))?; + file.seek(std::io::SeekFrom::Start(effective_start)) + .await + .with_context(|| format!("seek partial failed: {}", partial.display()))?; let mut stream = resp.bytes_stream(); let mut written: u64 = 0;