renovate/lib/workers/global/limits.ts
Sergei Zharinov 746d170824
feat(limits): Add "branchConcurrentLimit" option (#8046)
Co-authored-by: Michael Kriese <michael.kriese@visualon.de>
2021-01-10 13:29:14 +01:00

41 lines
946 B
TypeScript

import { logger } from '../../logger';
export enum Limit {
Commits = 'Commits',
PullRequests = 'PullRequests',
Branches = 'Branches',
}
interface LimitValue {
max: number | null;
current: number;
}
const limits = new Map<Limit, LimitValue>();
export function resetAllLimits(): void {
limits.clear();
}
export function setMaxLimit(key: Limit, val: unknown): void {
const max = typeof val === 'number' ? Math.max(0, val) : null;
limits.set(key, { current: 0, max });
logger.debug(`${key} limit = ${max}`);
}
export function incLimitedValue(key: Limit, incBy = 1): void {
const limit = limits.get(key) || { max: null, current: 0 };
limits.set(key, {
...limit,
current: limit.current + incBy,
});
}
export function isLimitReached(key: Limit): boolean {
const limit = limits.get(key);
if (!limit || limit.max === null) {
return false;
}
const { max, current } = limit;
return max - current <= 0;
}