packages/testnet-faucets/src/ts/request_queue.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

import * as _ from 'lodash';
import * as timers from 'timers';

// HACK: web3 leaks XMLHttpRequest into the global scope and causes requests to hang
// because they are using the wrong XHR package.
// Filed issue: https://github.com/ethereum/web3.js/issues/844
// tslint:disable-next-line:ordered-imports
import * as Web3 from 'web3';

const MAX_QUEUE_SIZE = 500;
const DEFAULT_QUEUE_INTERVAL_MS = 1000;

export class RequestQueue {
    protected _queueIntervalMs: number;
    protected _queue: string[];
    protected _queueIntervalId?: NodeJS.Timer;
    protected _web3: Web3;
    constructor(web3: any) {
        this._queueIntervalMs = DEFAULT_QUEUE_INTERVAL_MS;
        this._queue = [];
        this._web3 = web3;
        this._start();
    }
    public add(recipientAddress: string): boolean {
        if (this.isFull()) {
            return false;
        }
        this._queue.push(recipientAddress);
        return true;
    }
    public size(): number {
        return this._queue.length;
    }
    public isFull(): boolean {
        return this.size() >= MAX_QUEUE_SIZE;
    }
    protected _start() {
        this._queueIntervalId = timers.setInterval(() => {
            const recipientAddress = this._queue.shift();
            if (_.isUndefined(recipientAddress)) {
                return;
            }
            // tslint:disable-next-line:no-floating-promises
            this._processNextRequestFireAndForgetAsync(recipientAddress);
        }, this._queueIntervalMs);
    }
    // tslint:disable-next-line:prefer-function-over-method
    protected async _processNextRequestFireAndForgetAsync(recipientAddress: string) {
        throw new Error('Expected processNextRequestFireAndForgetAsync to be implemented by a subclass');
    }
}