occaecatidicta/packages/omelox-admin/lib/util/countDownLatch.ts

export class CountDownLatch {
    count: number;
    cb: () => void;
    /**
     * Count down to zero and invoke cb finally.
     */
    constructor(count: number, cb: () => void) {
        this.count = count;
        this.cb = cb;
    }

    /**
     * Call when a task finish to count down.
     *
     * @api public
     */
    done() {
        if (this.count <= 0) {
            throw new Error('illegal state.');
        }

        this.count--;
        if (this.count === 0) {
            this.cb();
        }
    }
}
/**
 * create a count down latch
 *
 * @api public
 */
export function createCountDownLatch(count: number, cb: () => void) {
    if (!count || count <= 0) {
        throw new Error('count should be positive.');
    }
    if (typeof cb !== 'function') {
        throw new Error('cb should be a function.');
    }

    return new CountDownLatch(count, cb);
}