From fd9a6516821e35d7d551ac8bed657a076a4220b4 Mon Sep 17 00:00:00 2001 From: Ashley //// Date: Sat, 20 Apr 2024 19:05:04 +0000 Subject: [PATCH] use do while --- src/libpoketube/libpoketube-core.js | 34 +++++++++++++++++++++-------- 1 file changed, 25 insertions(+), 9 deletions(-) diff --git a/src/libpoketube/libpoketube-core.js b/src/libpoketube/libpoketube-core.js index 353116a3..124aee3d 100644 --- a/src/libpoketube/libpoketube-core.js +++ b/src/libpoketube/libpoketube-core.js @@ -75,19 +75,35 @@ class InnerTubePokeVidious { * @param {Object} headers - The headers to include in the fetch request. * @returns {Promise} A promise that resolves to the text content of the response. */ + /** + * Fetch data from the specified URL with the given headers, retrying once if status code is 500. + * @param {string} url - The URL to fetch data from. + * @param {Object} headers - The headers to include in the fetch request. + * @returns {Promise} A promise that resolves to the response object. + */ async function fetchData(url, headers) { + let response; + let retry = false; + /** - * @type {Response} + * Retry fetching if status code is 500. */ - const response = await fetch(url, { headers }); + do { + try { + response = await fetch(url, { headers }); + if (response.status === 500) { + console.log("Retrying due to status 500..."); + retry = true; + } else { + retry = false; + } + } catch (error) { + console.error("Error occurred during fetch:", error); + retry = false; + } + } while (retry); - if (response.status === 500) { - // If status is 500, fetch again - console.log("Retrying due to status 500..."); - return fetchData(url, headers); - } - - return response.text(); + return response; } const { fetch } = await import("undici");