Function getTokenBucket

lib/api/apiUtils/rateLimit/tokenBucket.js:189–213

* Get or create token bucket for a bucket * * @param {string} resourceClass - "bucket" or "account" * @param {string} resourceId - bucket name or account canonicalId * @param {string} measure - measure id e.g. "rps" * @param {object} limitConfig - Rate limit configuration * @param {object} log

(resourceClass, resourceId, measure, limitConfig, log)

 * @returns {WorkerTokenBucket}
 */
function getTokenBucket(resourceClass, resourceId, measure, limitConfig, log) {
    const cacheKey = `${resourceClass}:${resourceId}:${measure}`;
    let bucket = tokenBuckets.get(cacheKey);
    if (!bucket) {
        bucket = new WorkerTokenBucket(resourceClass, resourceId, measure, limitConfig, log);
        tokenBuckets.set(cacheKey, bucket);

        log.debug('Created token bucket', {
            cacheKey,
            bufferSize: bucket.bufferSize,
            refillThreshold: bucket.refillThreshold,
        });
    } else {
        const { updated, oldConfig } = bucket.updateLimit(limitConfig);
        if (updated) {
            log.debug('Updated token bucket limit config', {
                cacheKey,
                old: oldConfig,
                new: limitConfig,
            });
        }
    }

    return bucket;
}

/**
 * Get all active token buckets