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 { 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),
|
|
||||||
|
|
||||||
}
|
}
|
||||||
Reference in new issue
Block a user