@xcanwin/manyoyo 8.6.5 → 8.6.6

package/lib/download-verified.js

@@ -16,11 +16,16 @@ function sha256File(filePath) {
     });
 }
 
+const RETRYABLE_STATUS = new Set([408, 429, 500, 502, 503, 504]);
+
+class RetryableError extends Error {}
+
 /**
  * 下载并校验 SHA256。缓存命中（文件存在且哈希一致）直接复用；
  * 哈希不符一律删除并报错，绝不保留不可信文件。
+ * 5xx / 429 / 网络错误（含下载中途断开）按 1s、2s、4s 退避，最多重试 retries 次；4xx 与哈希不符不重试。
  */
-async function downloadVerified({ url, sha256, dest, fetchImpl = fetch, log = () => {} }) {
+async function downloadVerified({ url, sha256, dest, fetchImpl = fetch, log = () => {}, retries = 3, retryDelayMs = 1000 }) {
     if (!/^[0-9a-f]{64}$/.test(sha256 || '')) {
         throw new Error(`缺少有效的 SHA256: ${url}`);
     }
@@ -29,13 +34,20 @@ async function downloadVerified({ url, sha256, dest, fetchImpl = fetch, log = ()
         log(`缓存命中: ${path.basename(dest)}`);
         return dest;
     }
-    log(`下载: ${url}`);
-    const response = await fetchImpl(url, { redirect: 'follow' });
-    if (!response.ok || !response.body) {
-        throw new Error(`下载失败 (${response.status}): ${url}`);
-    }
     const partial = `${dest}.partial`;
-    await pipeline(Readable.fromWeb(response.body), fs.createWriteStream(partial));
+    for (let attempt = 0; ; attempt += 1) {
+        try {
+            await fetchOnce({ url, partial, fetchImpl, log });
+            break;
+        } catch (error) {
+            fs.rmSync(partial, { force: true });
+            const retryable = error instanceof RetryableError || !(error && error.status);
+            if (!retryable || attempt >= retries) throw error instanceof RetryableError ? error.cause : error;
+            const wait = retryDelayMs * (2 ** attempt);
+            log(`下载失败，${wait / 1000} 秒后重试（${attempt + 1}/${retries}）: ${error.message}`);
+            await new Promise(resolve => setTimeout(resolve, wait));
+        }
+    }
     const actual = await sha256File(partial);
     if (actual !== sha256) {
         fs.rmSync(partial, { force: true });
@@ -45,4 +57,15 @@ async function downloadVerified({ url, sha256, dest, fetchImpl = fetch, log = ()
     return dest;
 }
 
+async function fetchOnce({ url, partial, fetchImpl, log }) {
+    log(`下载: ${url}`);
+    const response = await fetchImpl(url, { redirect: 'follow' });
+    if (!response.ok || !response.body) {
+        const error = Object.assign(new Error(`下载失败 (${response.status}): ${url}`), { status: response.status });
+        if (RETRYABLE_STATUS.has(response.status)) throw Object.assign(new RetryableError(error.message), { cause: error });
+        throw error;
+    }
+    await pipeline(Readable.fromWeb(response.body), fs.createWriteStream(partial));
+}
+
 module.exports = { sha256File, downloadVerified };