StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-15 05:41:51 +00:00
Code Releases Activity

Add bail-out after too many 429 errors

Browse source
This commit is contained in:
Dragory 2021-08-18 23:57:44 +03:00
parent 101367a103
commit 3c96cad8c6
No known key found for this signature in database
GPG key ID: 5F387BA66DF8AAC1
2 changed files with 33 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
backend/src/index.ts
View file

@ -18,8 +18,9 @@ import { RecoverablePluginError } from "./RecoverablePluginError";
import { SimpleError } from "./SimpleError";
import { ZeppelinGlobalConfig, ZeppelinGuildConfig } from "./types";
import { startUptimeCounter } from "./uptime";
import { errorMessage, isDiscordAPIError, isDiscordHTTPError, successMessage } from "./utils";
import { errorMessage, isDiscordAPIError, isDiscordHTTPError, SECONDS, successMessage } from "./utils";
import { loadYamlSafely } from "./utils/loadYamlSafely";
import { DecayingCounter } from "./utils/DecayingCounter";
if (!process.env.KEY) {
// tslint:disable-next-line:no-console
@ -178,6 +179,9 @@ connect().then(async () => {
});
client.setMaxListeners(200);
const safe429DecayInterval = 5 * SECONDS;
const safe429MaxCount = 5;
const safe429Counter = new DecayingCounter(safe429DecayInterval);
client.on(Constants.Events.DEBUG, errorText => {
if (!errorText.indexOf("429")) {
return;
@ -185,6 +189,13 @@ connect().then(async () => {
// tslint:disable-next-line:no-console
console.warn(errorText);
const value = safe429Counter.add(1);
if (value > safe429MaxCount) {
// tslint:disable-next-line:no-console
console.error(`Too many 429s (over ${safe429MaxCount} in ${safe429MaxCount * safe429DecayInterval}ms), exiting`);
process.exit(1);
}
});
client.on("error", err => {

21
backend/src/utils/DecayingCounter.ts Normal file
View file

@ -0,0 +1,21 @@
/**
* This is not related to Zeppelin's counters feature
*/
export class DecayingCounter {
protected value = 0;
constructor(protected decayInterval: number) {
setInterval(() => {
this.value = Math.max(0, this.value - 1);
}, decayInterval);
}
add(count = 1): number {
this.value += count;
return this.value;
}
get(): number {
return this.value;
}
}