Added Nitrado API retry
This commit is contained in:
1 file changed
+21
-13
+21
-13
@@ -3,10 +3,13 @@ const concat = require('concat-stream');
|
||||
const { Readable } = require('stream');
|
||||
const FormData = require('form-data');
|
||||
const fs = require('fs');
|
||||
const maxRetries = 5;
|
||||
const retryDelay = 5000; // 5 second
|
||||
|
||||
module.exports = {
|
||||
|
||||
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}`, {
|
||||
headers: {
|
||||
"Authorization": client.config.Nitrado.Auth
|
||||
@@ -24,10 +27,19 @@ module.exports = {
|
||||
const { body } = await fetch(res.data.token.url);
|
||||
await finished(Readable.fromWeb(body).pipe(stream));
|
||||
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) => {
|
||||
// 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`, {
|
||||
headers: {
|
||||
"Authorization": client.config.Nitrado.Auth
|
||||
@@ -59,16 +71,12 @@ module.exports = {
|
||||
}
|
||||
sendList();
|
||||
}));
|
||||
},
|
||||
|
||||
/*
|
||||
Allows more explicit function names outside this file;
|
||||
i.e BanPlayer() & UnbanPlayer() that both call to the
|
||||
parent function HandlePlayerBan() rather than write
|
||||
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),
|
||||
|
||||
} 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
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user