36 lines
1.1 KiB
TypeScript
36 lines
1.1 KiB
TypeScript
import { EventEmitter } from "stream";
|
|
|
|
// Class to ratelimit a resource (chatting, logging in, etc)
|
|
export default class RateLimiter extends EventEmitter {
|
|
private limit : number;
|
|
private interval : number;
|
|
private requestCount : number;
|
|
private limiter? : NodeJS.Timer;
|
|
private limiterSet : boolean;
|
|
constructor(limit : number, interval : number) {
|
|
super();
|
|
this.limit = limit;
|
|
this.interval = interval;
|
|
this.requestCount = 0;
|
|
this.limiterSet = false;
|
|
}
|
|
// Return value is whether or not the action should be continued
|
|
request() : boolean {
|
|
this.requestCount++;
|
|
if (this.requestCount === this.limit) {
|
|
this.emit('limit');
|
|
clearTimeout(this.limiter);
|
|
this.limiterSet = false;
|
|
this.requestCount = 0;
|
|
return false;
|
|
}
|
|
if (!this.limiterSet) {
|
|
this.limiter = setTimeout(() => {
|
|
this.limiterSet = false;
|
|
this.requestCount = 0;
|
|
}, this.interval * 1000);
|
|
this.limiterSet = true;
|
|
}
|
|
return true;
|
|
}
|
|
} |