Added Nitrado API retry

This commit is contained in:
IrPgFKS0 committed 2023-09-02 22:09:26 -07:00
1 parent a76ed204c9
commit 1b8d802d06
1 file changed
+21 -13
+21 -13
View File
@@ -3,10 +3,13 @@ const concat = require('concat-stream');
const { Readable } = require('stream'); const { Readable } = require('stream');
const FormData = require('form-data'); const FormData = require('form-data');
const fs = require('fs'); const fs = require('fs');
const maxRetries = 5;
const retryDelay = 5000; // 5 second
module.exports = { module.exports = {
DownloadNitradoFile: async (client, filename, outputDir) => { DownloadNitradoFile: async (client, filename, outputDir) => {
for (let retries = 0; retries <= maxRetries; retries++) {
try {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/download?file=${filename}`, { const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers/file_server/download?file=${filename}`, {
headers: { headers: {
"Authorization": client.config.Nitrado.Auth "Authorization": client.config.Nitrado.Auth
@@ -24,10 +27,19 @@ module.exports = {
const { body } = await fetch(res.data.token.url); const { body } = await fetch(res.data.token.url);
await finished(Readable.fromWeb(body).pipe(stream)); await finished(Readable.fromWeb(body).pipe(stream));
return 0; return 0;
} catch (error) {
if (retries === maxRetries) {
throw new Error(`Failed to fetch data after ${maxRetries} retries`);
}
}
await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
}
}, },
HandlePlayerBan: async (client, gamertag, ban) => { HandlePlayerBan: async (client, gamertag, ban) => {
// get current bans // get current bans
for (let retries = 0; retries <= maxRetries; retries++) {
try {
const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers`, { const res = await fetch(`https://api.nitrado.net/services/${client.config.Nitrado.ServerID}/gameservers`, {
headers: { headers: {
"Authorization": client.config.Nitrado.Auth "Authorization": client.config.Nitrado.Auth
@@ -59,16 +71,12 @@ module.exports = {
} }
sendList(); sendList();
})); }));
}, } catch (error) {
if (retries === maxRetries) {
/* throw new Error(`Failed to fetch data after ${maxRetries} retries`);
Allows more explicit function names outside this file; }
i.e BanPlayer() & UnbanPlayer() that both call to the }
parent function HandlePlayerBan() rather than write await new Promise(resolve => setTimeout(resolve, retryDelay)); // Delay before retrying
two whole different functions for each. }
*/ }
BanPlayer: async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, true),
UnbanPlayer: async (client, gamertag) => module.exports.HandlePlayerBan(client, gamertag, false),
} }