chuan/WebAI2API
1
0
Fork 0
mirror of https://github.com/foxhui/WebAI2API.git synced 2026-06-16 21:03:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复 retries=0 时不下载的问题

Browse Source 
当 imgDlRetry 未配置时，retries=0 导致循环不执行直接返回失败。
修改为 maxAttempts = max(1, retries)，确保至少执行一次下载尝试。
This commit is contained in:
daidai
2026-03-28 20:23:47 +08:00
Unverified
parent 2ba57e094c
commit 0d196d87a6
17 changed files with 132 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/backend/utils/download.js
+7 -5
View File
@@ -27,16 +27,18 @@ function isRetryableError(message) {
*/
export async function useContextDownload(url, page, options = {}) {
const { timeout = 120000, retries = 3, retryDelay = 1000 } = options;
// 至少执行一次尝试（retries=0 表示不重试，但仍需下载一次）
const maxAttempts = Math.max(1, retries);
for (let attempt = 1; attempt <= retries; attempt++) {
for (let attempt = 1; attempt <= maxAttempts; attempt++) {
try {
const response = await page.request.get(url, { timeout });
if (!response.ok()) {
const status = response.status();
// 5xx 错误可重试
if (status >= 500 && attempt < retries) {
logger.warn('下载', `HTTP ${status}，重试 ${attempt}/${retries}...`);
if (status >= 500 && attempt < maxAttempts) {
logger.warn('下载', `HTTP ${status}，重试 ${attempt}/${maxAttempts}...`);
await new Promise(r => setTimeout(r, retryDelay * attempt));
continue;
}
@@ -50,8 +52,8 @@ export async function useContextDownload(url, page, options = {}) {
return { image: `data:${mimeType};base64,${base64}`, imageUrl: url };
} catch (e) {
if (isRetryableError(e.message) && attempt < retries) {
logger.warn('下载', `${e.message}，重试 ${attempt}/${retries}...`);
if (isRetryableError(e.message) && attempt < maxAttempts) {
logger.warn('下载', `${e.message}，重试 ${attempt}/${maxAttempts}...`);
await new Promise(r => setTimeout(r, retryDelay * attempt));
continue;
}