Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions checkstyle/import-control-server.xml
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,7 @@

<subpackage name="network">
<allow pkg="com.fasterxml.jackson" />
<allow pkg="org.apache.kafka.network.metrics" />
</subpackage>

</import-control>
19 changes: 9 additions & 10 deletions core/src/main/java/kafka/server/ClientRequestQuotaManager.java
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,14 @@
*/
package kafka.server;

import kafka.network.RequestChannel;

import org.apache.kafka.common.MetricName;
import org.apache.kafka.common.internals.Plugin;
import org.apache.kafka.common.metrics.Metrics;
import org.apache.kafka.common.metrics.QuotaViolationException;
import org.apache.kafka.common.metrics.Sensor;
import org.apache.kafka.common.metrics.stats.Rate;
import org.apache.kafka.common.utils.Time;
import org.apache.kafka.network.Request;
import org.apache.kafka.server.config.ClientQuotaManagerConfig;
import org.apache.kafka.server.quota.ClientQuotaCallback;
import org.apache.kafka.server.quota.ClientQuotaManager;
Expand Down Expand Up @@ -76,22 +75,22 @@ private void recordExempt(double value) {
* @param request client request
* @return Number of milliseconds to throttle in case of quota violation. Zero otherwise
*/
public int maybeRecordAndGetThrottleTimeMs(RequestChannel.Request request, long timeMs) {
public int maybeRecordAndGetThrottleTimeMs(Request request, long timeMs) {
if (quotasEnabled()) {
request.setRecordNetworkThreadTimeCallback(timeNanos -> {
recordNoThrottle(request.session(), request.header().clientId(), nanosToPercentage(Long.parseLong(timeNanos.toString())));
});
request.setRecordNetworkThreadTimeCallback(timeNanos ->
recordNoThrottle(request.session(), request.header().clientId(), nanosToPercentage(timeNanos))
);
return recordAndGetThrottleTimeMs(request.session(), request.header().clientId(), nanosToPercentage(request.requestThreadTimeNanos()), timeMs);
} else {
return 0;
}
}

public void maybeRecordExempt(RequestChannel.Request request) {
public void maybeRecordExempt(Request request) {
if (quotasEnabled()) {
request.setRecordNetworkThreadTimeCallback(timeNanos -> {
recordExempt(nanosToPercentage(Long.parseLong(timeNanos.toString())));
});
request.setRecordNetworkThreadTimeCallback(timeNanos ->
recordExempt(nanosToPercentage(timeNanos))
);
recordExempt(nanosToPercentage(request.requestThreadTimeNanos()));
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@

package kafka.server.handlers;

import kafka.network.RequestChannel;
import kafka.server.AuthHelper;
import kafka.server.KafkaConfig;

Expand All @@ -31,6 +30,7 @@
import org.apache.kafka.common.requests.DescribeTopicPartitionsRequest;
import org.apache.kafka.common.resource.Resource;
import org.apache.kafka.metadata.MetadataCache;
import org.apache.kafka.network.Request;

import java.util.HashSet;
import java.util.List;
Expand All @@ -55,7 +55,7 @@ public DescribeTopicPartitionsRequestHandler(
this.config = config;
}

public DescribeTopicPartitionsResponseData handleDescribeTopicPartitionsRequest(RequestChannel.Request abstractRequest) {
public DescribeTopicPartitionsResponseData handleDescribeTopicPartitionsRequest(Request abstractRequest) {
DescribeTopicPartitionsRequestData request = ((DescribeTopicPartitionsRequest) abstractRequest.loggableRequest()).data();
Set<String> topics = new HashSet<>();
boolean fetchAllTopics = request.topics().isEmpty();
Expand Down
Loading
Loading