zookeeper-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "ASF GitHub Bot (JIRA)" <j...@apache.org>
Subject [jira] [Commented] (ZOOKEEPER-2736) Add a connection rate limiter
Date Fri, 24 Mar 2017 18:56:41 GMT

    [ https://issues.apache.org/jira/browse/ZOOKEEPER-2736?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15940937#comment-15940937
] 

ASF GitHub Bot commented on ZOOKEEPER-2736:
-------------------------------------------

Github user afine commented on a diff in the pull request:

    https://github.com/apache/zookeeper/pull/205#discussion_r107976351
  
    --- 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() {
    --- End diff --
    
    would things be easier if we just made this synchronized?


> 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)

Mime
View raw message