asyncawait/src/async/semaphore.ts

async/await for node.js

export = Semaphore;


/** A simple abstraction for limiting concurrent function calls to a specific upper bound. */
class Semaphore {

    static unlimited = new Semaphore(10000000);

    constructor(private n: number) {
        this._avail = n;
    }

    enter(fn: () => void) {
        if (this._avail > 0) {
            --this._avail;
            fn();
        } else {
            this._queued.push(fn);
        }
    }

    leave() {
        if (this._queued.length > 0) {
            var fn = this._queued.pop();
            fn();
        } else {
            ++this._avail;
        }
    }

    private _avail: number;
    private _queued: Function[] = [];
}