org.apache.pinot.broker.queryquota.MaxHitRateTracker Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.pinot.broker.queryquota;
import com.google.common.annotations.VisibleForTesting;
/**
* A stateful version of hit counter. Similar to the default hit counter, it maintains a list of buckets.
* Whereas it maintains an extra variable called _lastAccessTimestamp which tracks the last access time.
* If the stateful hit counter gets queried, it firstly compares the current timestamp and the last access timestamp,
* calculating the start index and end index among the buckets. Then, it traverses through all the valid candidate
* buckets.
* If the current timestamp has exceeded the current time range of all the buckets, this hit counter will use
* the current timestamp minus the default time queried time range to calculate the start time index.
*/
public class MaxHitRateTracker extends HitCounter {
private static final int ONE_SECOND_BUCKET_WIDTH_MS = 1000;
private static final int MAX_TIME_RANGE_FACTOR = 2;
private final long _maxTimeRangeMs;
private final long _defaultTimeRangeMs;
private volatile long _lastAccessTimestamp;
public MaxHitRateTracker(int timeRangeInSeconds) {
this(timeRangeInSeconds, timeRangeInSeconds * MAX_TIME_RANGE_FACTOR);
}
private MaxHitRateTracker(int defaultTimeRangeInSeconds, int maxTimeRangeInSeconds) {
super(maxTimeRangeInSeconds, (int) (maxTimeRangeInSeconds * 1000L / ONE_SECOND_BUCKET_WIDTH_MS));
_defaultTimeRangeMs = defaultTimeRangeInSeconds * 1000L;
_maxTimeRangeMs = maxTimeRangeInSeconds * 1000L;
}
/**
* Get the maximum count among the buckets
*/
public int getMaxCountPerBucket() {
return getMaxCountPerBucket(System.currentTimeMillis());
}
@VisibleForTesting
int getMaxCountPerBucket(long now) {
// Update the last access timestamp if the hit counter didn't get queried for more than _maxTimeRangeMs.
long then = _lastAccessTimestamp;
if (now - then > _maxTimeRangeMs) {
then = now - _defaultTimeRangeMs;
}
long startTimeUnits = then / _timeBucketWidthMs;
int startIndex = (int) (startTimeUnits % _bucketCount);
long numTimeUnits = now / _timeBucketWidthMs;
int endIndex = (int) (numTimeUnits % _bucketCount);
int maxCount = 0;
// Skipping the end index here as its bucket hasn't fully gathered all the hits yet.
for (int i = startIndex; i != endIndex; i = ((i + 1) % _bucketCount)) {
if (numTimeUnits - _bucketStartTime.get(i) < _bucketCount) {
maxCount = Math.max(_bucketHitCount.get(i), maxCount);
}
}
// Update the last access timestamp
_lastAccessTimestamp = now;
return maxCount;
}
}