fix(Collector): increase and decrease max listeners dynamically

This commit is contained in:
SpaceEEC 2018-05-08 11:57:12 +02:00
parent ed8ab91782
commit 60288d0704
No known key found for this signature in database
GPG key ID: 8C8A2E338661B871
2 changed files with 4 additions and 0 deletions

View file

@ -33,6 +33,7 @@ class MessageCollector extends Collector {
*/
this.received = 0;
this.client.setMaxListeners(this.client.getMaxListeners() + 1);
this.client.on('message', this.listener);
// For backwards compatibility (remove in v12)
@ -92,6 +93,7 @@ class MessageCollector extends Collector {
cleanup() {
this.removeListener('collect', this._reEmitter);
this.client.removeListener('message', this.listener);
this.client.setMaxListeners(this.client.getMaxListeners() - 1);
}
}

View file

@ -39,6 +39,7 @@ class ReactionCollector extends Collector {
*/
this.total = 0;
this.client.setMaxListeners(this.client.getMaxListeners() + 1);
this.client.on('messageReactionAdd', this.listener);
}
@ -77,6 +78,7 @@ class ReactionCollector extends Collector {
*/
cleanup() {
this.client.removeListener('messageReactionAdd', this.listener);
this.client.setMaxListeners(this.client.getMaxListeners() - 1);
}
}