[
https://issues.apache.org/jira/browse/ZOOKEEPER-2736?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15941242#comment-15941242
]
ASF GitHub Bot commented on ZOOKEEPER-2736:
-------------------------------------------
Github user eribeiro commented on a diff in the pull request:
https://github.com/apache/zookeeper/pull/205#discussion_r108010303
--- Diff: src/java/main/org/apache/zookeeper/common/TokenBucket.java ---
@@ -0,0 +1,88 @@
+/**
+* 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.zookeeper.common;
+
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicLong;
+
+/**
+ * Simple rate limiter based on a token bucket.
+ */
+public class TokenBucket implements RateLimiter {
+
+ // VisibleForTesting
+ long refreshPeriodNanos = TimeUnit.SECONDS.toNanos(1L);
+ // VisibleForTesting
+ volatile long nextRefillTime;
+ private AtomicLong tokens;
+ private long capacity;
+ private long tokensPerPeriod;
+
+ public TokenBucket() { }
+
+ // VisibleForTesting
+ public TokenBucket(long capacity, long tokensPerSecond, long
initialTokens) {
+ this.tokens = new AtomicLong(initialTokens);
+ this.capacity = Math.max(capacity, tokensPerSecond);
+ this.tokensPerPeriod = tokensPerSecond;
+ this.nextRefillTime = System.nanoTime() + refreshPeriodNanos;
+ }
+
+ @Override
+ public boolean tryAquire() {
+ refill();
+ long current = tokens.get();
+ while (current > 0) {
+ if (tokens.compareAndSet(current, current - 1)) {
+ return true;
+ }
+ current = tokens.get();
+ }
+ return false;
+ }
+
+ private void refill() {
+ long currentTime = System.nanoTime();
+ if (currentTime >= nextRefillTime) {
+ synchronized (this) {
+ if (currentTime >= nextRefillTime) {
+ long elapsedNanos = currentTime - nextRefillTime +
refreshPeriodNanos;
+ long elapsedPeriods = elapsedNanos /
refreshPeriodNanos;
+ long newTokens = elapsedPeriods * tokensPerPeriod;
+ tokens.set(Math.min(capacity, tokens.get() +
newTokens));
+ nextRefillTime = currentTime + refreshPeriodNanos -
(elapsedNanos % refreshPeriodNanos);
+ }
+ }
+ }
+ }
+
+ // VisibleForTesting
+ long getTokenCount() {
--- End diff --
+1.
> Add a connection rate limiter
> -----------------------------
>
> Key: ZOOKEEPER-2736
> URL: https://issues.apache.org/jira/browse/ZOOKEEPER-2736
> Project: ZooKeeper
> Issue Type: Improvement
> Components: server
> Affects Versions: 3.4.9, 3.5.2
> Reporter: Vincent Poon
> Attachments: ZOOKEEPER-2736.v1.patch
>
>
> Currently the maxClientCnxns property only limits the aggregate number of
> connections from a client, but not the rate at which connections can be
> created.
> This patch adds a configurable connection rate limiter which limits the rate
> as well.
--
This message was sent by Atlassian JIRA
(v6.3.15#6346)
