mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-04 12:18:16 +00:00 
			
		
		
		
	Merge pull request #38320 from liggitt/golang-ratelimit
Automatic merge from submit-queue (batch tested with PRs 59158, 38320, 59059, 55516, 59357). If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>. Switch from juju/ratelimit to golang.org/x/time/rate Replaces juju/ratelimit with golang.org/x/time/rate xref https://github.com/kubernetes/steering/issues/21 Requires removing the Saturation() method on the rate limiter. In the process of attempting to contribute it to the `golang.org/x/time/rate` implementation, it became clear that what it was calculating was not very useful when combined with periodic polling. See discussion in https://go-review.googlesource.com/c/time/+/29958#message-4caffc11669cadd90e2da4c05122cfec50ea6a22 ```release-note NONE ```
This commit is contained in:
		
							
								
								
									
										4
									
								
								Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										4
									
								
								Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							@@ -2016,10 +2016,6 @@
 | 
			
		||||
			"Comment": "v3.0.7-72-ga0ff256",
 | 
			
		||||
			"Rev": "a0ff2567cfb70903282db057e799fd826784d41d"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/kardianos/osext",
 | 
			
		||||
			"Rev": "8fef92e41e22a70e700a96b29f066cda30ea24ef"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										199
									
								
								Godeps/LICENSES
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										199
									
								
								Godeps/LICENSES
									
									
									
										generated
									
									
									
								
							@@ -70015,205 +70015,6 @@ http://creativecommons.org/publicdomain/zero/1.0
 | 
			
		||||
================================================================================
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
================================================================================
 | 
			
		||||
= vendor/github.com/juju/ratelimit licensed under: =
 | 
			
		||||
 | 
			
		||||
All files in this repository are licensed as follows. If you contribute
 | 
			
		||||
to this repository, it is assumed that you license your contribution
 | 
			
		||||
under the same license unless you state otherwise.
 | 
			
		||||
 | 
			
		||||
All files Copyright (C) 2015 Canonical Ltd. unless otherwise specified in the file.
 | 
			
		||||
 | 
			
		||||
This software is licensed under the LGPLv3, included below.
 | 
			
		||||
 | 
			
		||||
As a special exception to the GNU Lesser General Public License version 3
 | 
			
		||||
("LGPL3"), the copyright holders of this Library give you permission to
 | 
			
		||||
convey to a third party a Combined Work that links statically or dynamically
 | 
			
		||||
to this Library without providing any Minimal Corresponding Source or
 | 
			
		||||
Minimal Application Code as set out in 4d or providing the installation
 | 
			
		||||
information set out in section 4e, provided that you comply with the other
 | 
			
		||||
provisions of LGPL3 and provided that you meet, for the Application the
 | 
			
		||||
terms and conditions of the license(s) which apply to the Application.
 | 
			
		||||
 | 
			
		||||
Except as stated in this special exception, the provisions of LGPL3 will
 | 
			
		||||
continue to comply in full to this Library. If you modify this Library, you
 | 
			
		||||
may apply this exception to your version of this Library, but you are not
 | 
			
		||||
obliged to do so. If you do not wish to do so, delete this exception
 | 
			
		||||
statement from your version. This exception does not (and cannot) modify any
 | 
			
		||||
license terms which apply to the Application, with which you must still
 | 
			
		||||
comply.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
                   GNU LESSER GENERAL PUBLIC LICENSE
 | 
			
		||||
                       Version 3, 29 June 2007
 | 
			
		||||
 | 
			
		||||
 Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
 | 
			
		||||
 Everyone is permitted to copy and distribute verbatim copies
 | 
			
		||||
 of this license document, but changing it is not allowed.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
  This version of the GNU Lesser General Public License incorporates
 | 
			
		||||
the terms and conditions of version 3 of the GNU General Public
 | 
			
		||||
License, supplemented by the additional permissions listed below.
 | 
			
		||||
 | 
			
		||||
  0. Additional Definitions.
 | 
			
		||||
 | 
			
		||||
  As used herein, "this License" refers to version 3 of the GNU Lesser
 | 
			
		||||
General Public License, and the "GNU GPL" refers to version 3 of the GNU
 | 
			
		||||
General Public License.
 | 
			
		||||
 | 
			
		||||
  "The Library" refers to a covered work governed by this License,
 | 
			
		||||
other than an Application or a Combined Work as defined below.
 | 
			
		||||
 | 
			
		||||
  An "Application" is any work that makes use of an interface provided
 | 
			
		||||
by the Library, but which is not otherwise based on the Library.
 | 
			
		||||
Defining a subclass of a class defined by the Library is deemed a mode
 | 
			
		||||
of using an interface provided by the Library.
 | 
			
		||||
 | 
			
		||||
  A "Combined Work" is a work produced by combining or linking an
 | 
			
		||||
Application with the Library.  The particular version of the Library
 | 
			
		||||
with which the Combined Work was made is also called the "Linked
 | 
			
		||||
Version".
 | 
			
		||||
 | 
			
		||||
  The "Minimal Corresponding Source" for a Combined Work means the
 | 
			
		||||
Corresponding Source for the Combined Work, excluding any source code
 | 
			
		||||
for portions of the Combined Work that, considered in isolation, are
 | 
			
		||||
based on the Application, and not on the Linked Version.
 | 
			
		||||
 | 
			
		||||
  The "Corresponding Application Code" for a Combined Work means the
 | 
			
		||||
object code and/or source code for the Application, including any data
 | 
			
		||||
and utility programs needed for reproducing the Combined Work from the
 | 
			
		||||
Application, but excluding the System Libraries of the Combined Work.
 | 
			
		||||
 | 
			
		||||
  1. Exception to Section 3 of the GNU GPL.
 | 
			
		||||
 | 
			
		||||
  You may convey a covered work under sections 3 and 4 of this License
 | 
			
		||||
without being bound by section 3 of the GNU GPL.
 | 
			
		||||
 | 
			
		||||
  2. Conveying Modified Versions.
 | 
			
		||||
 | 
			
		||||
  If you modify a copy of the Library, and, in your modifications, a
 | 
			
		||||
facility refers to a function or data to be supplied by an Application
 | 
			
		||||
that uses the facility (other than as an argument passed when the
 | 
			
		||||
facility is invoked), then you may convey a copy of the modified
 | 
			
		||||
version:
 | 
			
		||||
 | 
			
		||||
   a) under this License, provided that you make a good faith effort to
 | 
			
		||||
   ensure that, in the event an Application does not supply the
 | 
			
		||||
   function or data, the facility still operates, and performs
 | 
			
		||||
   whatever part of its purpose remains meaningful, or
 | 
			
		||||
 | 
			
		||||
   b) under the GNU GPL, with none of the additional permissions of
 | 
			
		||||
   this License applicable to that copy.
 | 
			
		||||
 | 
			
		||||
  3. Object Code Incorporating Material from Library Header Files.
 | 
			
		||||
 | 
			
		||||
  The object code form of an Application may incorporate material from
 | 
			
		||||
a header file that is part of the Library.  You may convey such object
 | 
			
		||||
code under terms of your choice, provided that, if the incorporated
 | 
			
		||||
material is not limited to numerical parameters, data structure
 | 
			
		||||
layouts and accessors, or small macros, inline functions and templates
 | 
			
		||||
(ten or fewer lines in length), you do both of the following:
 | 
			
		||||
 | 
			
		||||
   a) Give prominent notice with each copy of the object code that the
 | 
			
		||||
   Library is used in it and that the Library and its use are
 | 
			
		||||
   covered by this License.
 | 
			
		||||
 | 
			
		||||
   b) Accompany the object code with a copy of the GNU GPL and this license
 | 
			
		||||
   document.
 | 
			
		||||
 | 
			
		||||
  4. Combined Works.
 | 
			
		||||
 | 
			
		||||
  You may convey a Combined Work under terms of your choice that,
 | 
			
		||||
taken together, effectively do not restrict modification of the
 | 
			
		||||
portions of the Library contained in the Combined Work and reverse
 | 
			
		||||
engineering for debugging such modifications, if you also do each of
 | 
			
		||||
the following:
 | 
			
		||||
 | 
			
		||||
   a) Give prominent notice with each copy of the Combined Work that
 | 
			
		||||
   the Library is used in it and that the Library and its use are
 | 
			
		||||
   covered by this License.
 | 
			
		||||
 | 
			
		||||
   b) Accompany the Combined Work with a copy of the GNU GPL and this license
 | 
			
		||||
   document.
 | 
			
		||||
 | 
			
		||||
   c) For a Combined Work that displays copyright notices during
 | 
			
		||||
   execution, include the copyright notice for the Library among
 | 
			
		||||
   these notices, as well as a reference directing the user to the
 | 
			
		||||
   copies of the GNU GPL and this license document.
 | 
			
		||||
 | 
			
		||||
   d) Do one of the following:
 | 
			
		||||
 | 
			
		||||
       0) Convey the Minimal Corresponding Source under the terms of this
 | 
			
		||||
       License, and the Corresponding Application Code in a form
 | 
			
		||||
       suitable for, and under terms that permit, the user to
 | 
			
		||||
       recombine or relink the Application with a modified version of
 | 
			
		||||
       the Linked Version to produce a modified Combined Work, in the
 | 
			
		||||
       manner specified by section 6 of the GNU GPL for conveying
 | 
			
		||||
       Corresponding Source.
 | 
			
		||||
 | 
			
		||||
       1) Use a suitable shared library mechanism for linking with the
 | 
			
		||||
       Library.  A suitable mechanism is one that (a) uses at run time
 | 
			
		||||
       a copy of the Library already present on the user's computer
 | 
			
		||||
       system, and (b) will operate properly with a modified version
 | 
			
		||||
       of the Library that is interface-compatible with the Linked
 | 
			
		||||
       Version.
 | 
			
		||||
 | 
			
		||||
   e) Provide Installation Information, but only if you would otherwise
 | 
			
		||||
   be required to provide such information under section 6 of the
 | 
			
		||||
   GNU GPL, and only to the extent that such information is
 | 
			
		||||
   necessary to install and execute a modified version of the
 | 
			
		||||
   Combined Work produced by recombining or relinking the
 | 
			
		||||
   Application with a modified version of the Linked Version. (If
 | 
			
		||||
   you use option 4d0, the Installation Information must accompany
 | 
			
		||||
   the Minimal Corresponding Source and Corresponding Application
 | 
			
		||||
   Code. If you use option 4d1, you must provide the Installation
 | 
			
		||||
   Information in the manner specified by section 6 of the GNU GPL
 | 
			
		||||
   for conveying Corresponding Source.)
 | 
			
		||||
 | 
			
		||||
  5. Combined Libraries.
 | 
			
		||||
 | 
			
		||||
  You may place library facilities that are a work based on the
 | 
			
		||||
Library side by side in a single library together with other library
 | 
			
		||||
facilities that are not Applications and are not covered by this
 | 
			
		||||
License, and convey such a combined library under terms of your
 | 
			
		||||
choice, if you do both of the following:
 | 
			
		||||
 | 
			
		||||
   a) Accompany the combined library with a copy of the same work based
 | 
			
		||||
   on the Library, uncombined with any other library facilities,
 | 
			
		||||
   conveyed under the terms of this License.
 | 
			
		||||
 | 
			
		||||
   b) Give prominent notice with the combined library that part of it
 | 
			
		||||
   is a work based on the Library, and explaining where to find the
 | 
			
		||||
   accompanying uncombined form of the same work.
 | 
			
		||||
 | 
			
		||||
  6. Revised Versions of the GNU Lesser General Public License.
 | 
			
		||||
 | 
			
		||||
  The Free Software Foundation may publish revised and/or new versions
 | 
			
		||||
of the GNU Lesser General Public License from time to time. Such new
 | 
			
		||||
versions will be similar in spirit to the present version, but may
 | 
			
		||||
differ in detail to address new problems or concerns.
 | 
			
		||||
 | 
			
		||||
  Each version is given a distinguishing version number. If the
 | 
			
		||||
Library as you received it specifies that a certain numbered version
 | 
			
		||||
of the GNU Lesser General Public License "or any later version"
 | 
			
		||||
applies to it, you have the option of following the terms and
 | 
			
		||||
conditions either of that published version or of any later version
 | 
			
		||||
published by the Free Software Foundation. If the Library as you
 | 
			
		||||
received it does not specify a version number of the GNU Lesser
 | 
			
		||||
General Public License, you may choose any version of the GNU Lesser
 | 
			
		||||
General Public License ever published by the Free Software Foundation.
 | 
			
		||||
 | 
			
		||||
  If the Library as you received it specifies that a proxy can decide
 | 
			
		||||
whether future versions of the GNU Lesser General Public License shall
 | 
			
		||||
apply, that proxy's public statement of acceptance of any version is
 | 
			
		||||
permanent authorization for you to choose that version for the
 | 
			
		||||
Library.
 | 
			
		||||
 | 
			
		||||
= vendor/github.com/juju/ratelimit/LICENSE 2d1c30374313ae40df7772dc92ef9fd5
 | 
			
		||||
================================================================================
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
================================================================================
 | 
			
		||||
= vendor/github.com/kardianos/osext licensed under: =
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -15,7 +15,7 @@ go_library(
 | 
			
		||||
    deps = [
 | 
			
		||||
        "//pkg/controller:go_default_library",
 | 
			
		||||
        "//vendor/github.com/golang/glog:go_default_library",
 | 
			
		||||
        "//vendor/github.com/juju/ratelimit:go_default_library",
 | 
			
		||||
        "//vendor/golang.org/x/time/rate:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/api/certificates/v1beta1:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/api/errors:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/util/runtime:go_default_library",
 | 
			
		||||
 
 | 
			
		||||
@@ -22,6 +22,9 @@ import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/golang/glog"
 | 
			
		||||
	"golang.org/x/time/rate"
 | 
			
		||||
 | 
			
		||||
	certificates "k8s.io/api/certificates/v1beta1"
 | 
			
		||||
	"k8s.io/apimachinery/pkg/api/errors"
 | 
			
		||||
	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 | 
			
		||||
@@ -34,9 +37,6 @@ import (
 | 
			
		||||
	"k8s.io/client-go/tools/record"
 | 
			
		||||
	"k8s.io/client-go/util/workqueue"
 | 
			
		||||
	"k8s.io/kubernetes/pkg/controller"
 | 
			
		||||
 | 
			
		||||
	"github.com/golang/glog"
 | 
			
		||||
	"github.com/juju/ratelimit"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type CertificateController struct {
 | 
			
		||||
@@ -65,7 +65,7 @@ func NewCertificateController(
 | 
			
		||||
		queue: workqueue.NewNamedRateLimitingQueue(workqueue.NewMaxOfRateLimiter(
 | 
			
		||||
			workqueue.NewItemExponentialFailureRateLimiter(200*time.Millisecond, 1000*time.Second),
 | 
			
		||||
			// 10 qps, 100 bucket size.  This is only for retry speed and its only the overall factor (not per item)
 | 
			
		||||
			&workqueue.BucketRateLimiter{Bucket: ratelimit.NewBucketWithRate(float64(10), int64(100))},
 | 
			
		||||
			&workqueue.BucketRateLimiter{Limiter: rate.NewLimiter(rate.Limit(10), 100)},
 | 
			
		||||
		), "certificate"),
 | 
			
		||||
		handler: handler,
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -293,20 +293,15 @@ func (q *RateLimitedTimedQueue) SwapLimiter(newQPS float32) {
 | 
			
		||||
		newLimiter = flowcontrol.NewFakeNeverRateLimiter()
 | 
			
		||||
	} else {
 | 
			
		||||
		newLimiter = flowcontrol.NewTokenBucketRateLimiter(newQPS, EvictionRateLimiterBurst)
 | 
			
		||||
	}
 | 
			
		||||
	// If we're currently waiting on limiter, we drain the new one - this is a good approach when Burst value is 1
 | 
			
		||||
	// TODO: figure out if we need to support higher Burst values and decide on the drain logic, should we keep:
 | 
			
		||||
	// - saturation (percentage of used tokens)
 | 
			
		||||
	// - number of used tokens
 | 
			
		||||
	// - number of available tokens
 | 
			
		||||
	// - something else
 | 
			
		||||
	for q.limiter.Saturation() > newLimiter.Saturation() {
 | 
			
		||||
		// Check if we're not using fake limiter
 | 
			
		||||
		previousSaturation := newLimiter.Saturation()
 | 
			
		||||
		newLimiter.TryAccept()
 | 
			
		||||
		// It's a fake limiter
 | 
			
		||||
		if newLimiter.Saturation() == previousSaturation {
 | 
			
		||||
			break
 | 
			
		||||
 | 
			
		||||
		// If we're currently waiting on limiter, we drain the new one - this is a good approach when Burst value is 1
 | 
			
		||||
		// TODO: figure out if we need to support higher Burst values and decide on the drain logic, should we keep:
 | 
			
		||||
		// - saturation (percentage of used tokens)
 | 
			
		||||
		// - number of used tokens
 | 
			
		||||
		// - number of available tokens
 | 
			
		||||
		// - something else
 | 
			
		||||
		if q.limiter.TryAccept() == false {
 | 
			
		||||
			newLimiter.TryAccept()
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	q.limiter.Stop()
 | 
			
		||||
 
 | 
			
		||||
@@ -61,6 +61,7 @@ type fakeTimer struct {
 | 
			
		||||
 | 
			
		||||
func newFakeTimer() *fakeTimer {
 | 
			
		||||
	ft := &fakeTimer{
 | 
			
		||||
		now:     time.Date(2000, 0, 0, 0, 0, 0, 0, time.UTC),
 | 
			
		||||
		c:       make(chan time.Time),
 | 
			
		||||
		updated: make(chan timerUpdate),
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -13,7 +13,6 @@ go_library(
 | 
			
		||||
    deps = [
 | 
			
		||||
        "//vendor/github.com/golang/glog:go_default_library",
 | 
			
		||||
        "//vendor/github.com/prometheus/client_golang/prometheus:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/util/wait:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/client-go/util/flowcontrol:go_default_library",
 | 
			
		||||
    ],
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -21,7 +21,6 @@ import (
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"k8s.io/apimachinery/pkg/util/wait"
 | 
			
		||||
	"k8s.io/client-go/util/flowcontrol"
 | 
			
		||||
 | 
			
		||||
	"github.com/golang/glog"
 | 
			
		||||
@@ -71,11 +70,13 @@ func RegisterMetricAndTrackRateLimiterUsage(ownerName string, rateLimiter flowco
 | 
			
		||||
	if err := registerRateLimiterMetric(ownerName); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	go wait.Until(func() {
 | 
			
		||||
		metricsLock.Lock()
 | 
			
		||||
		defer metricsLock.Unlock()
 | 
			
		||||
		rateLimiterMetrics[ownerName].metric.Set(rateLimiter.Saturation())
 | 
			
		||||
	}, updatePeriod, rateLimiterMetrics[ownerName].stopCh)
 | 
			
		||||
	// TODO: determine how to track rate limiter saturation
 | 
			
		||||
	// See discussion at https://go-review.googlesource.com/c/time/+/29958#message-4caffc11669cadd90e2da4c05122cfec50ea6a22
 | 
			
		||||
	// go wait.Until(func() {
 | 
			
		||||
	//   metricsLock.Lock()
 | 
			
		||||
	//   defer metricsLock.Unlock()
 | 
			
		||||
	//   rateLimiterMetrics[ownerName].metric.Set()
 | 
			
		||||
	// }, updatePeriod, rateLimiterMetrics[ownerName].stopCh)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -230,10 +230,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/mailru/easyjson/buffer",
 | 
			
		||||
			"Rev": "2f5df55504ebc322e4d52d34df6a1f5b503bf26d"
 | 
			
		||||
@@ -406,6 +402,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/width",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "google.golang.org/genproto/googleapis/api/annotations",
 | 
			
		||||
			"Rev": "09f6ed296fc66555a25fe4ce95173148778dfa85"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										4
									
								
								staging/src/k8s.io/apiserver/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										4
									
								
								staging/src/k8s.io/apiserver/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							@@ -558,10 +558,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/mailru/easyjson/buffer",
 | 
			
		||||
			"Rev": "2f5df55504ebc322e4d52d34df6a1f5b503bf26d"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										8
									
								
								staging/src/k8s.io/client-go/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										8
									
								
								staging/src/k8s.io/client-go/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							@@ -186,10 +186,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/pmezard/go-difflib/difflib",
 | 
			
		||||
			"Rev": "d8ed2627bdf02c080bf22230dbb337003b7aba2d"
 | 
			
		||||
@@ -274,6 +270,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/unicode/norm",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "gopkg.in/inf.v0",
 | 
			
		||||
			"Rev": "3887ee99ecf07df5b447e9b00d9c0b2adaa9f3e4"
 | 
			
		||||
 
 | 
			
		||||
@@ -25,7 +25,7 @@ go_library(
 | 
			
		||||
    ],
 | 
			
		||||
    importpath = "k8s.io/client-go/util/flowcontrol",
 | 
			
		||||
    deps = [
 | 
			
		||||
        "//vendor/github.com/juju/ratelimit:go_default_library",
 | 
			
		||||
        "//vendor/golang.org/x/time/rate:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/util/clock:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/client-go/util/integer:go_default_library",
 | 
			
		||||
    ],
 | 
			
		||||
 
 | 
			
		||||
@@ -18,8 +18,9 @@ package flowcontrol
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/juju/ratelimit"
 | 
			
		||||
	"golang.org/x/time/rate"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type RateLimiter interface {
 | 
			
		||||
@@ -30,17 +31,13 @@ type RateLimiter interface {
 | 
			
		||||
	Accept()
 | 
			
		||||
	// Stop stops the rate limiter, subsequent calls to CanAccept will return false
 | 
			
		||||
	Stop()
 | 
			
		||||
	// Saturation returns a percentage number which describes how saturated
 | 
			
		||||
	// this rate limiter is.
 | 
			
		||||
	// Usually we use token bucket rate limiter. In that case,
 | 
			
		||||
	// 1.0 means no tokens are available; 0.0 means we have a full bucket of tokens to use.
 | 
			
		||||
	Saturation() float64
 | 
			
		||||
	// QPS returns QPS of this rate limiter
 | 
			
		||||
	QPS() float32
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type tokenBucketRateLimiter struct {
 | 
			
		||||
	limiter *ratelimit.Bucket
 | 
			
		||||
	limiter *rate.Limiter
 | 
			
		||||
	clock   Clock
 | 
			
		||||
	qps     float32
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -50,42 +47,48 @@ type tokenBucketRateLimiter struct {
 | 
			
		||||
// The bucket is initially filled with 'burst' tokens, and refills at a rate of 'qps'.
 | 
			
		||||
// The maximum number of tokens in the bucket is capped at 'burst'.
 | 
			
		||||
func NewTokenBucketRateLimiter(qps float32, burst int) RateLimiter {
 | 
			
		||||
	limiter := ratelimit.NewBucketWithRate(float64(qps), int64(burst))
 | 
			
		||||
	return newTokenBucketRateLimiter(limiter, qps)
 | 
			
		||||
	limiter := rate.NewLimiter(rate.Limit(qps), burst)
 | 
			
		||||
	return newTokenBucketRateLimiter(limiter, realClock{}, qps)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// An injectable, mockable clock interface.
 | 
			
		||||
type Clock interface {
 | 
			
		||||
	ratelimit.Clock
 | 
			
		||||
	Now() time.Time
 | 
			
		||||
	Sleep(time.Duration)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type realClock struct{}
 | 
			
		||||
 | 
			
		||||
func (realClock) Now() time.Time {
 | 
			
		||||
	return time.Now()
 | 
			
		||||
}
 | 
			
		||||
func (realClock) Sleep(d time.Duration) {
 | 
			
		||||
	time.Sleep(d)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewTokenBucketRateLimiterWithClock is identical to NewTokenBucketRateLimiter
 | 
			
		||||
// but allows an injectable clock, for testing.
 | 
			
		||||
func NewTokenBucketRateLimiterWithClock(qps float32, burst int, clock Clock) RateLimiter {
 | 
			
		||||
	limiter := ratelimit.NewBucketWithRateAndClock(float64(qps), int64(burst), clock)
 | 
			
		||||
	return newTokenBucketRateLimiter(limiter, qps)
 | 
			
		||||
func NewTokenBucketRateLimiterWithClock(qps float32, burst int, c Clock) RateLimiter {
 | 
			
		||||
	limiter := rate.NewLimiter(rate.Limit(qps), burst)
 | 
			
		||||
	return newTokenBucketRateLimiter(limiter, c, qps)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func newTokenBucketRateLimiter(limiter *ratelimit.Bucket, qps float32) RateLimiter {
 | 
			
		||||
func newTokenBucketRateLimiter(limiter *rate.Limiter, c Clock, qps float32) RateLimiter {
 | 
			
		||||
	return &tokenBucketRateLimiter{
 | 
			
		||||
		limiter: limiter,
 | 
			
		||||
		clock:   c,
 | 
			
		||||
		qps:     qps,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *tokenBucketRateLimiter) TryAccept() bool {
 | 
			
		||||
	return t.limiter.TakeAvailable(1) == 1
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *tokenBucketRateLimiter) Saturation() float64 {
 | 
			
		||||
	capacity := t.limiter.Capacity()
 | 
			
		||||
	avail := t.limiter.Available()
 | 
			
		||||
	return float64(capacity-avail) / float64(capacity)
 | 
			
		||||
	return t.limiter.AllowN(t.clock.Now(), 1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Accept will block until a token becomes available
 | 
			
		||||
func (t *tokenBucketRateLimiter) Accept() {
 | 
			
		||||
	t.limiter.Wait(1)
 | 
			
		||||
	now := t.clock.Now()
 | 
			
		||||
	t.clock.Sleep(t.limiter.ReserveN(now, 1).DelayFrom(now))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *tokenBucketRateLimiter) Stop() {
 | 
			
		||||
@@ -105,10 +108,6 @@ func (t *fakeAlwaysRateLimiter) TryAccept() bool {
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *fakeAlwaysRateLimiter) Saturation() float64 {
 | 
			
		||||
	return 0
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *fakeAlwaysRateLimiter) Stop() {}
 | 
			
		||||
 | 
			
		||||
func (t *fakeAlwaysRateLimiter) Accept() {}
 | 
			
		||||
@@ -131,10 +130,6 @@ func (t *fakeNeverRateLimiter) TryAccept() bool {
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *fakeNeverRateLimiter) Saturation() float64 {
 | 
			
		||||
	return 1
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (t *fakeNeverRateLimiter) Stop() {
 | 
			
		||||
	t.wg.Done()
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -17,7 +17,6 @@ limitations under the License.
 | 
			
		||||
package flowcontrol
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"math"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"testing"
 | 
			
		||||
	"time"
 | 
			
		||||
@@ -116,29 +115,6 @@ func TestThrottle(t *testing.T) {
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestRateLimiterSaturation(t *testing.T) {
 | 
			
		||||
	const e = 0.000001
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		capacity int
 | 
			
		||||
		take     int
 | 
			
		||||
 | 
			
		||||
		expectedSaturation float64
 | 
			
		||||
	}{
 | 
			
		||||
		{1, 1, 1},
 | 
			
		||||
		{10, 3, 0.3},
 | 
			
		||||
	}
 | 
			
		||||
	for i, tt := range tests {
 | 
			
		||||
		rl := NewTokenBucketRateLimiter(1, tt.capacity)
 | 
			
		||||
		for i := 0; i < tt.take; i++ {
 | 
			
		||||
			rl.Accept()
 | 
			
		||||
		}
 | 
			
		||||
		if math.Abs(rl.Saturation()-tt.expectedSaturation) > e {
 | 
			
		||||
			t.Fatalf("#%d: Saturation rate difference isn't within tolerable range\n want=%f, get=%f",
 | 
			
		||||
				i, tt.expectedSaturation, rl.Saturation())
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestAlwaysFake(t *testing.T) {
 | 
			
		||||
	rl := NewFakeAlwaysRateLimiter()
 | 
			
		||||
	if !rl.TryAccept() {
 | 
			
		||||
 
 | 
			
		||||
@@ -34,7 +34,7 @@ go_library(
 | 
			
		||||
    ],
 | 
			
		||||
    importpath = "k8s.io/client-go/util/workqueue",
 | 
			
		||||
    deps = [
 | 
			
		||||
        "//vendor/github.com/juju/ratelimit:go_default_library",
 | 
			
		||||
        "//vendor/golang.org/x/time/rate:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/util/clock:go_default_library",
 | 
			
		||||
        "//vendor/k8s.io/apimachinery/pkg/util/runtime:go_default_library",
 | 
			
		||||
    ],
 | 
			
		||||
 
 | 
			
		||||
@@ -21,7 +21,7 @@ import (
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/juju/ratelimit"
 | 
			
		||||
	"golang.org/x/time/rate"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type RateLimiter interface {
 | 
			
		||||
@@ -40,19 +40,19 @@ func DefaultControllerRateLimiter() RateLimiter {
 | 
			
		||||
	return NewMaxOfRateLimiter(
 | 
			
		||||
		NewItemExponentialFailureRateLimiter(5*time.Millisecond, 1000*time.Second),
 | 
			
		||||
		// 10 qps, 100 bucket size.  This is only for retry speed and its only the overall factor (not per item)
 | 
			
		||||
		&BucketRateLimiter{Bucket: ratelimit.NewBucketWithRate(float64(10), int64(100))},
 | 
			
		||||
		&BucketRateLimiter{Limiter: rate.NewLimiter(rate.Limit(10), 100)},
 | 
			
		||||
	)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// BucketRateLimiter adapts a standard bucket to the workqueue ratelimiter API
 | 
			
		||||
type BucketRateLimiter struct {
 | 
			
		||||
	*ratelimit.Bucket
 | 
			
		||||
	*rate.Limiter
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var _ RateLimiter = &BucketRateLimiter{}
 | 
			
		||||
 | 
			
		||||
func (r *BucketRateLimiter) When(item interface{}) time.Duration {
 | 
			
		||||
	return r.Bucket.Take(1)
 | 
			
		||||
	return r.Limiter.Reserve().Delay()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *BucketRateLimiter) NumRequeues(item interface{}) int {
 | 
			
		||||
 
 | 
			
		||||
@@ -210,10 +210,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/mailru/easyjson/buffer",
 | 
			
		||||
			"Rev": "2f5df55504ebc322e4d52d34df6a1f5b503bf26d"
 | 
			
		||||
@@ -382,6 +378,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/width",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "google.golang.org/genproto/googleapis/api/annotations",
 | 
			
		||||
			"Rev": "09f6ed296fc66555a25fe4ce95173148778dfa85"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										12
									
								
								staging/src/k8s.io/metrics/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										12
									
								
								staging/src/k8s.io/metrics/Godeps/Godeps.json
									
									
									
										generated
									
									
									
								
							@@ -62,14 +62,14 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/spf13/pflag",
 | 
			
		||||
			"Rev": "4c012f6dcd9546820e378d0bdda4d8fc772cdfea"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/net/context",
 | 
			
		||||
			"Rev": "1c05540f6879653db88113bc4a2b70aec4bd491f"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/net/http2",
 | 
			
		||||
			"Rev": "1c05540f6879653db88113bc4a2b70aec4bd491f"
 | 
			
		||||
@@ -102,6 +102,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/unicode/norm",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "gopkg.in/inf.v0",
 | 
			
		||||
			"Rev": "3887ee99ecf07df5b447e9b00d9c0b2adaa9f3e4"
 | 
			
		||||
 
 | 
			
		||||
@@ -202,10 +202,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/mailru/easyjson/buffer",
 | 
			
		||||
			"Rev": "2f5df55504ebc322e4d52d34df6a1f5b503bf26d"
 | 
			
		||||
@@ -366,6 +362,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/width",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "google.golang.org/genproto/googleapis/api/annotations",
 | 
			
		||||
			"Rev": "09f6ed296fc66555a25fe4ce95173148778dfa85"
 | 
			
		||||
 
 | 
			
		||||
@@ -86,10 +86,6 @@
 | 
			
		||||
			"ImportPath": "github.com/json-iterator/go",
 | 
			
		||||
			"Rev": "13f86432b882000a51c6e610c620974462691a97"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/juju/ratelimit",
 | 
			
		||||
			"Rev": "5b9ff866471762aa2ab2dced63c9fb6f53921342"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "github.com/spf13/pflag",
 | 
			
		||||
			"Rev": "4c012f6dcd9546820e378d0bdda4d8fc772cdfea"
 | 
			
		||||
@@ -142,6 +138,10 @@
 | 
			
		||||
			"ImportPath": "golang.org/x/text/unicode/norm",
 | 
			
		||||
			"Rev": "b19bf474d317b857955b12035d2c5acb57ce8b01"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "golang.org/x/time/rate",
 | 
			
		||||
			"Rev": "f51c12702a4d776e4c1fa9b0fabab841babae631"
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			"ImportPath": "gopkg.in/inf.v0",
 | 
			
		||||
			"Rev": "3887ee99ecf07df5b447e9b00d9c0b2adaa9f3e4"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										191
									
								
								vendor/github.com/juju/ratelimit/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										191
									
								
								vendor/github.com/juju/ratelimit/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -1,191 +0,0 @@
 | 
			
		||||
All files in this repository are licensed as follows. If you contribute
 | 
			
		||||
to this repository, it is assumed that you license your contribution
 | 
			
		||||
under the same license unless you state otherwise.
 | 
			
		||||
 | 
			
		||||
All files Copyright (C) 2015 Canonical Ltd. unless otherwise specified in the file.
 | 
			
		||||
 | 
			
		||||
This software is licensed under the LGPLv3, included below.
 | 
			
		||||
 | 
			
		||||
As a special exception to the GNU Lesser General Public License version 3
 | 
			
		||||
("LGPL3"), the copyright holders of this Library give you permission to
 | 
			
		||||
convey to a third party a Combined Work that links statically or dynamically
 | 
			
		||||
to this Library without providing any Minimal Corresponding Source or
 | 
			
		||||
Minimal Application Code as set out in 4d or providing the installation
 | 
			
		||||
information set out in section 4e, provided that you comply with the other
 | 
			
		||||
provisions of LGPL3 and provided that you meet, for the Application the
 | 
			
		||||
terms and conditions of the license(s) which apply to the Application.
 | 
			
		||||
 | 
			
		||||
Except as stated in this special exception, the provisions of LGPL3 will
 | 
			
		||||
continue to comply in full to this Library. If you modify this Library, you
 | 
			
		||||
may apply this exception to your version of this Library, but you are not
 | 
			
		||||
obliged to do so. If you do not wish to do so, delete this exception
 | 
			
		||||
statement from your version. This exception does not (and cannot) modify any
 | 
			
		||||
license terms which apply to the Application, with which you must still
 | 
			
		||||
comply.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
                   GNU LESSER GENERAL PUBLIC LICENSE
 | 
			
		||||
                       Version 3, 29 June 2007
 | 
			
		||||
 | 
			
		||||
 Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
 | 
			
		||||
 Everyone is permitted to copy and distribute verbatim copies
 | 
			
		||||
 of this license document, but changing it is not allowed.
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
  This version of the GNU Lesser General Public License incorporates
 | 
			
		||||
the terms and conditions of version 3 of the GNU General Public
 | 
			
		||||
License, supplemented by the additional permissions listed below.
 | 
			
		||||
 | 
			
		||||
  0. Additional Definitions.
 | 
			
		||||
 | 
			
		||||
  As used herein, "this License" refers to version 3 of the GNU Lesser
 | 
			
		||||
General Public License, and the "GNU GPL" refers to version 3 of the GNU
 | 
			
		||||
General Public License.
 | 
			
		||||
 | 
			
		||||
  "The Library" refers to a covered work governed by this License,
 | 
			
		||||
other than an Application or a Combined Work as defined below.
 | 
			
		||||
 | 
			
		||||
  An "Application" is any work that makes use of an interface provided
 | 
			
		||||
by the Library, but which is not otherwise based on the Library.
 | 
			
		||||
Defining a subclass of a class defined by the Library is deemed a mode
 | 
			
		||||
of using an interface provided by the Library.
 | 
			
		||||
 | 
			
		||||
  A "Combined Work" is a work produced by combining or linking an
 | 
			
		||||
Application with the Library.  The particular version of the Library
 | 
			
		||||
with which the Combined Work was made is also called the "Linked
 | 
			
		||||
Version".
 | 
			
		||||
 | 
			
		||||
  The "Minimal Corresponding Source" for a Combined Work means the
 | 
			
		||||
Corresponding Source for the Combined Work, excluding any source code
 | 
			
		||||
for portions of the Combined Work that, considered in isolation, are
 | 
			
		||||
based on the Application, and not on the Linked Version.
 | 
			
		||||
 | 
			
		||||
  The "Corresponding Application Code" for a Combined Work means the
 | 
			
		||||
object code and/or source code for the Application, including any data
 | 
			
		||||
and utility programs needed for reproducing the Combined Work from the
 | 
			
		||||
Application, but excluding the System Libraries of the Combined Work.
 | 
			
		||||
 | 
			
		||||
  1. Exception to Section 3 of the GNU GPL.
 | 
			
		||||
 | 
			
		||||
  You may convey a covered work under sections 3 and 4 of this License
 | 
			
		||||
without being bound by section 3 of the GNU GPL.
 | 
			
		||||
 | 
			
		||||
  2. Conveying Modified Versions.
 | 
			
		||||
 | 
			
		||||
  If you modify a copy of the Library, and, in your modifications, a
 | 
			
		||||
facility refers to a function or data to be supplied by an Application
 | 
			
		||||
that uses the facility (other than as an argument passed when the
 | 
			
		||||
facility is invoked), then you may convey a copy of the modified
 | 
			
		||||
version:
 | 
			
		||||
 | 
			
		||||
   a) under this License, provided that you make a good faith effort to
 | 
			
		||||
   ensure that, in the event an Application does not supply the
 | 
			
		||||
   function or data, the facility still operates, and performs
 | 
			
		||||
   whatever part of its purpose remains meaningful, or
 | 
			
		||||
 | 
			
		||||
   b) under the GNU GPL, with none of the additional permissions of
 | 
			
		||||
   this License applicable to that copy.
 | 
			
		||||
 | 
			
		||||
  3. Object Code Incorporating Material from Library Header Files.
 | 
			
		||||
 | 
			
		||||
  The object code form of an Application may incorporate material from
 | 
			
		||||
a header file that is part of the Library.  You may convey such object
 | 
			
		||||
code under terms of your choice, provided that, if the incorporated
 | 
			
		||||
material is not limited to numerical parameters, data structure
 | 
			
		||||
layouts and accessors, or small macros, inline functions and templates
 | 
			
		||||
(ten or fewer lines in length), you do both of the following:
 | 
			
		||||
 | 
			
		||||
   a) Give prominent notice with each copy of the object code that the
 | 
			
		||||
   Library is used in it and that the Library and its use are
 | 
			
		||||
   covered by this License.
 | 
			
		||||
 | 
			
		||||
   b) Accompany the object code with a copy of the GNU GPL and this license
 | 
			
		||||
   document.
 | 
			
		||||
 | 
			
		||||
  4. Combined Works.
 | 
			
		||||
 | 
			
		||||
  You may convey a Combined Work under terms of your choice that,
 | 
			
		||||
taken together, effectively do not restrict modification of the
 | 
			
		||||
portions of the Library contained in the Combined Work and reverse
 | 
			
		||||
engineering for debugging such modifications, if you also do each of
 | 
			
		||||
the following:
 | 
			
		||||
 | 
			
		||||
   a) Give prominent notice with each copy of the Combined Work that
 | 
			
		||||
   the Library is used in it and that the Library and its use are
 | 
			
		||||
   covered by this License.
 | 
			
		||||
 | 
			
		||||
   b) Accompany the Combined Work with a copy of the GNU GPL and this license
 | 
			
		||||
   document.
 | 
			
		||||
 | 
			
		||||
   c) For a Combined Work that displays copyright notices during
 | 
			
		||||
   execution, include the copyright notice for the Library among
 | 
			
		||||
   these notices, as well as a reference directing the user to the
 | 
			
		||||
   copies of the GNU GPL and this license document.
 | 
			
		||||
 | 
			
		||||
   d) Do one of the following:
 | 
			
		||||
 | 
			
		||||
       0) Convey the Minimal Corresponding Source under the terms of this
 | 
			
		||||
       License, and the Corresponding Application Code in a form
 | 
			
		||||
       suitable for, and under terms that permit, the user to
 | 
			
		||||
       recombine or relink the Application with a modified version of
 | 
			
		||||
       the Linked Version to produce a modified Combined Work, in the
 | 
			
		||||
       manner specified by section 6 of the GNU GPL for conveying
 | 
			
		||||
       Corresponding Source.
 | 
			
		||||
 | 
			
		||||
       1) Use a suitable shared library mechanism for linking with the
 | 
			
		||||
       Library.  A suitable mechanism is one that (a) uses at run time
 | 
			
		||||
       a copy of the Library already present on the user's computer
 | 
			
		||||
       system, and (b) will operate properly with a modified version
 | 
			
		||||
       of the Library that is interface-compatible with the Linked
 | 
			
		||||
       Version.
 | 
			
		||||
 | 
			
		||||
   e) Provide Installation Information, but only if you would otherwise
 | 
			
		||||
   be required to provide such information under section 6 of the
 | 
			
		||||
   GNU GPL, and only to the extent that such information is
 | 
			
		||||
   necessary to install and execute a modified version of the
 | 
			
		||||
   Combined Work produced by recombining or relinking the
 | 
			
		||||
   Application with a modified version of the Linked Version. (If
 | 
			
		||||
   you use option 4d0, the Installation Information must accompany
 | 
			
		||||
   the Minimal Corresponding Source and Corresponding Application
 | 
			
		||||
   Code. If you use option 4d1, you must provide the Installation
 | 
			
		||||
   Information in the manner specified by section 6 of the GNU GPL
 | 
			
		||||
   for conveying Corresponding Source.)
 | 
			
		||||
 | 
			
		||||
  5. Combined Libraries.
 | 
			
		||||
 | 
			
		||||
  You may place library facilities that are a work based on the
 | 
			
		||||
Library side by side in a single library together with other library
 | 
			
		||||
facilities that are not Applications and are not covered by this
 | 
			
		||||
License, and convey such a combined library under terms of your
 | 
			
		||||
choice, if you do both of the following:
 | 
			
		||||
 | 
			
		||||
   a) Accompany the combined library with a copy of the same work based
 | 
			
		||||
   on the Library, uncombined with any other library facilities,
 | 
			
		||||
   conveyed under the terms of this License.
 | 
			
		||||
 | 
			
		||||
   b) Give prominent notice with the combined library that part of it
 | 
			
		||||
   is a work based on the Library, and explaining where to find the
 | 
			
		||||
   accompanying uncombined form of the same work.
 | 
			
		||||
 | 
			
		||||
  6. Revised Versions of the GNU Lesser General Public License.
 | 
			
		||||
 | 
			
		||||
  The Free Software Foundation may publish revised and/or new versions
 | 
			
		||||
of the GNU Lesser General Public License from time to time. Such new
 | 
			
		||||
versions will be similar in spirit to the present version, but may
 | 
			
		||||
differ in detail to address new problems or concerns.
 | 
			
		||||
 | 
			
		||||
  Each version is given a distinguishing version number. If the
 | 
			
		||||
Library as you received it specifies that a certain numbered version
 | 
			
		||||
of the GNU Lesser General Public License "or any later version"
 | 
			
		||||
applies to it, you have the option of following the terms and
 | 
			
		||||
conditions either of that published version or of any later version
 | 
			
		||||
published by the Free Software Foundation. If the Library as you
 | 
			
		||||
received it does not specify a version number of the GNU Lesser
 | 
			
		||||
General Public License, you may choose any version of the GNU Lesser
 | 
			
		||||
General Public License ever published by the Free Software Foundation.
 | 
			
		||||
 | 
			
		||||
  If the Library as you received it specifies that a proxy can decide
 | 
			
		||||
whether future versions of the GNU Lesser General Public License shall
 | 
			
		||||
apply, that proxy's public statement of acceptance of any version is
 | 
			
		||||
permanent authorization for you to choose that version for the
 | 
			
		||||
Library.
 | 
			
		||||
							
								
								
									
										117
									
								
								vendor/github.com/juju/ratelimit/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										117
									
								
								vendor/github.com/juju/ratelimit/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -1,117 +0,0 @@
 | 
			
		||||
# ratelimit
 | 
			
		||||
--
 | 
			
		||||
    import "github.com/juju/ratelimit"
 | 
			
		||||
 | 
			
		||||
The ratelimit package provides an efficient token bucket implementation. See
 | 
			
		||||
http://en.wikipedia.org/wiki/Token_bucket.
 | 
			
		||||
 | 
			
		||||
## Usage
 | 
			
		||||
 | 
			
		||||
#### func  Reader
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func Reader(r io.Reader, bucket *Bucket) io.Reader
 | 
			
		||||
```
 | 
			
		||||
Reader returns a reader that is rate limited by the given token bucket. Each
 | 
			
		||||
token in the bucket represents one byte.
 | 
			
		||||
 | 
			
		||||
#### func  Writer
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func Writer(w io.Writer, bucket *Bucket) io.Writer
 | 
			
		||||
```
 | 
			
		||||
Writer returns a writer that is rate limited by the given token bucket. Each
 | 
			
		||||
token in the bucket represents one byte.
 | 
			
		||||
 | 
			
		||||
#### type Bucket
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
type Bucket struct {
 | 
			
		||||
}
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
Bucket represents a token bucket that fills at a predetermined rate. Methods on
 | 
			
		||||
Bucket may be called concurrently.
 | 
			
		||||
 | 
			
		||||
#### func  NewBucket
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func NewBucket(fillInterval time.Duration, capacity int64) *Bucket
 | 
			
		||||
```
 | 
			
		||||
NewBucket returns a new token bucket that fills at the rate of one token every
 | 
			
		||||
fillInterval, up to the given maximum capacity. Both arguments must be positive.
 | 
			
		||||
The bucket is initially full.
 | 
			
		||||
 | 
			
		||||
#### func  NewBucketWithQuantum
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func NewBucketWithQuantum(fillInterval time.Duration, capacity, quantum int64) *Bucket
 | 
			
		||||
```
 | 
			
		||||
NewBucketWithQuantum is similar to NewBucket, but allows the specification of
 | 
			
		||||
the quantum size - quantum tokens are added every fillInterval.
 | 
			
		||||
 | 
			
		||||
#### func  NewBucketWithRate
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func NewBucketWithRate(rate float64, capacity int64) *Bucket
 | 
			
		||||
```
 | 
			
		||||
NewBucketWithRate returns a token bucket that fills the bucket at the rate of
 | 
			
		||||
rate tokens per second up to the given maximum capacity. Because of limited
 | 
			
		||||
clock resolution, at high rates, the actual rate may be up to 1% different from
 | 
			
		||||
the specified rate.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) Rate
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) Rate() float64
 | 
			
		||||
```
 | 
			
		||||
Rate returns the fill rate of the bucket, in tokens per second.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) Take
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) Take(count int64) time.Duration
 | 
			
		||||
```
 | 
			
		||||
Take takes count tokens from the bucket without blocking. It returns the time
 | 
			
		||||
that the caller should wait until the tokens are actually available.
 | 
			
		||||
 | 
			
		||||
Note that if the request is irrevocable - there is no way to return tokens to
 | 
			
		||||
the bucket once this method commits us to taking them.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) TakeAvailable
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) TakeAvailable(count int64) int64
 | 
			
		||||
```
 | 
			
		||||
TakeAvailable takes up to count immediately available tokens from the bucket. It
 | 
			
		||||
returns the number of tokens removed, or zero if there are no available tokens.
 | 
			
		||||
It does not block.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) TakeMaxDuration
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) TakeMaxDuration(count int64, maxWait time.Duration) (time.Duration, bool)
 | 
			
		||||
```
 | 
			
		||||
TakeMaxDuration is like Take, except that it will only take tokens from the
 | 
			
		||||
bucket if the wait time for the tokens is no greater than maxWait.
 | 
			
		||||
 | 
			
		||||
If it would take longer than maxWait for the tokens to become available, it does
 | 
			
		||||
nothing and reports false, otherwise it returns the time that the caller should
 | 
			
		||||
wait until the tokens are actually available, and reports true.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) Wait
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) Wait(count int64)
 | 
			
		||||
```
 | 
			
		||||
Wait takes count tokens from the bucket, waiting until they are available.
 | 
			
		||||
 | 
			
		||||
#### func (*Bucket) WaitMaxDuration
 | 
			
		||||
 | 
			
		||||
```go
 | 
			
		||||
func (tb *Bucket) WaitMaxDuration(count int64, maxWait time.Duration) bool
 | 
			
		||||
```
 | 
			
		||||
WaitMaxDuration is like Wait except that it will only take tokens from the
 | 
			
		||||
bucket if it needs to wait for no greater than maxWait. It reports whether any
 | 
			
		||||
tokens have been removed from the bucket If no tokens have been removed, it
 | 
			
		||||
returns immediately.
 | 
			
		||||
							
								
								
									
										284
									
								
								vendor/github.com/juju/ratelimit/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										284
									
								
								vendor/github.com/juju/ratelimit/ratelimit.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -1,284 +0,0 @@
 | 
			
		||||
// Copyright 2014 Canonical Ltd.
 | 
			
		||||
// Licensed under the LGPLv3 with static-linking exception.
 | 
			
		||||
// See LICENCE file for details.
 | 
			
		||||
 | 
			
		||||
// Package ratelimit provides an efficient token bucket implementation
 | 
			
		||||
// that can be used to limit the rate of arbitrary things.
 | 
			
		||||
// See http://en.wikipedia.org/wiki/Token_bucket.
 | 
			
		||||
package ratelimit
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"math"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Bucket represents a token bucket that fills at a predetermined rate.
 | 
			
		||||
// Methods on Bucket may be called concurrently.
 | 
			
		||||
type Bucket struct {
 | 
			
		||||
	startTime    time.Time
 | 
			
		||||
	capacity     int64
 | 
			
		||||
	quantum      int64
 | 
			
		||||
	fillInterval time.Duration
 | 
			
		||||
	clock        Clock
 | 
			
		||||
 | 
			
		||||
	// The mutex guards the fields following it.
 | 
			
		||||
	mu sync.Mutex
 | 
			
		||||
 | 
			
		||||
	// avail holds the number of available tokens
 | 
			
		||||
	// in the bucket, as of availTick ticks from startTime.
 | 
			
		||||
	// It will be negative when there are consumers
 | 
			
		||||
	// waiting for tokens.
 | 
			
		||||
	avail     int64
 | 
			
		||||
	availTick int64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Clock is used to inject testable fakes.
 | 
			
		||||
type Clock interface {
 | 
			
		||||
	Now() time.Time
 | 
			
		||||
	Sleep(d time.Duration)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// realClock implements Clock in terms of standard time functions.
 | 
			
		||||
type realClock struct{}
 | 
			
		||||
 | 
			
		||||
// Now is identical to time.Now.
 | 
			
		||||
func (realClock) Now() time.Time {
 | 
			
		||||
	return time.Now()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Sleep is identical to time.Sleep.
 | 
			
		||||
func (realClock) Sleep(d time.Duration) {
 | 
			
		||||
	time.Sleep(d)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewBucket returns a new token bucket that fills at the
 | 
			
		||||
// rate of one token every fillInterval, up to the given
 | 
			
		||||
// maximum capacity. Both arguments must be
 | 
			
		||||
// positive. The bucket is initially full.
 | 
			
		||||
func NewBucket(fillInterval time.Duration, capacity int64) *Bucket {
 | 
			
		||||
	return NewBucketWithClock(fillInterval, capacity, realClock{})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewBucketWithClock is identical to NewBucket but injects a testable clock
 | 
			
		||||
// interface.
 | 
			
		||||
func NewBucketWithClock(fillInterval time.Duration, capacity int64, clock Clock) *Bucket {
 | 
			
		||||
	return NewBucketWithQuantumAndClock(fillInterval, capacity, 1, clock)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// rateMargin specifes the allowed variance of actual
 | 
			
		||||
// rate from specified rate. 1% seems reasonable.
 | 
			
		||||
const rateMargin = 0.01
 | 
			
		||||
 | 
			
		||||
// NewBucketWithRate returns a token bucket that fills the bucket
 | 
			
		||||
// at the rate of rate tokens per second up to the given
 | 
			
		||||
// maximum capacity. Because of limited clock resolution,
 | 
			
		||||
// at high rates, the actual rate may be up to 1% different from the
 | 
			
		||||
// specified rate.
 | 
			
		||||
func NewBucketWithRate(rate float64, capacity int64) *Bucket {
 | 
			
		||||
	return NewBucketWithRateAndClock(rate, capacity, realClock{})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewBucketWithRateAndClock is identical to NewBucketWithRate but injects a
 | 
			
		||||
// testable clock interface.
 | 
			
		||||
func NewBucketWithRateAndClock(rate float64, capacity int64, clock Clock) *Bucket {
 | 
			
		||||
	for quantum := int64(1); quantum < 1<<50; quantum = nextQuantum(quantum) {
 | 
			
		||||
		fillInterval := time.Duration(1e9 * float64(quantum) / rate)
 | 
			
		||||
		if fillInterval <= 0 {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
		tb := NewBucketWithQuantumAndClock(fillInterval, capacity, quantum, clock)
 | 
			
		||||
		if diff := math.Abs(tb.Rate() - rate); diff/rate <= rateMargin {
 | 
			
		||||
			return tb
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	panic("cannot find suitable quantum for " + strconv.FormatFloat(rate, 'g', -1, 64))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// nextQuantum returns the next quantum to try after q.
 | 
			
		||||
// We grow the quantum exponentially, but slowly, so we
 | 
			
		||||
// get a good fit in the lower numbers.
 | 
			
		||||
func nextQuantum(q int64) int64 {
 | 
			
		||||
	q1 := q * 11 / 10
 | 
			
		||||
	if q1 == q {
 | 
			
		||||
		q1++
 | 
			
		||||
	}
 | 
			
		||||
	return q1
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewBucketWithQuantum is similar to NewBucket, but allows
 | 
			
		||||
// the specification of the quantum size - quantum tokens
 | 
			
		||||
// are added every fillInterval.
 | 
			
		||||
func NewBucketWithQuantum(fillInterval time.Duration, capacity, quantum int64) *Bucket {
 | 
			
		||||
	return NewBucketWithQuantumAndClock(fillInterval, capacity, quantum, realClock{})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// NewBucketWithQuantumAndClock is identical to NewBucketWithQuantum but injects
 | 
			
		||||
// a testable clock interface.
 | 
			
		||||
func NewBucketWithQuantumAndClock(fillInterval time.Duration, capacity, quantum int64, clock Clock) *Bucket {
 | 
			
		||||
	if fillInterval <= 0 {
 | 
			
		||||
		panic("token bucket fill interval is not > 0")
 | 
			
		||||
	}
 | 
			
		||||
	if capacity <= 0 {
 | 
			
		||||
		panic("token bucket capacity is not > 0")
 | 
			
		||||
	}
 | 
			
		||||
	if quantum <= 0 {
 | 
			
		||||
		panic("token bucket quantum is not > 0")
 | 
			
		||||
	}
 | 
			
		||||
	return &Bucket{
 | 
			
		||||
		clock:        clock,
 | 
			
		||||
		startTime:    clock.Now(),
 | 
			
		||||
		capacity:     capacity,
 | 
			
		||||
		quantum:      quantum,
 | 
			
		||||
		avail:        capacity,
 | 
			
		||||
		fillInterval: fillInterval,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Wait takes count tokens from the bucket, waiting until they are
 | 
			
		||||
// available.
 | 
			
		||||
func (tb *Bucket) Wait(count int64) {
 | 
			
		||||
	if d := tb.Take(count); d > 0 {
 | 
			
		||||
		tb.clock.Sleep(d)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// WaitMaxDuration is like Wait except that it will
 | 
			
		||||
// only take tokens from the bucket if it needs to wait
 | 
			
		||||
// for no greater than maxWait. It reports whether
 | 
			
		||||
// any tokens have been removed from the bucket
 | 
			
		||||
// If no tokens have been removed, it returns immediately.
 | 
			
		||||
func (tb *Bucket) WaitMaxDuration(count int64, maxWait time.Duration) bool {
 | 
			
		||||
	d, ok := tb.TakeMaxDuration(count, maxWait)
 | 
			
		||||
	if d > 0 {
 | 
			
		||||
		tb.clock.Sleep(d)
 | 
			
		||||
	}
 | 
			
		||||
	return ok
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const infinityDuration time.Duration = 0x7fffffffffffffff
 | 
			
		||||
 | 
			
		||||
// Take takes count tokens from the bucket without blocking. It returns
 | 
			
		||||
// the time that the caller should wait until the tokens are actually
 | 
			
		||||
// available.
 | 
			
		||||
//
 | 
			
		||||
// Note that if the request is irrevocable - there is no way to return
 | 
			
		||||
// tokens to the bucket once this method commits us to taking them.
 | 
			
		||||
func (tb *Bucket) Take(count int64) time.Duration {
 | 
			
		||||
	d, _ := tb.take(tb.clock.Now(), count, infinityDuration)
 | 
			
		||||
	return d
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TakeMaxDuration is like Take, except that
 | 
			
		||||
// it will only take tokens from the bucket if the wait
 | 
			
		||||
// time for the tokens is no greater than maxWait.
 | 
			
		||||
//
 | 
			
		||||
// If it would take longer than maxWait for the tokens
 | 
			
		||||
// to become available, it does nothing and reports false,
 | 
			
		||||
// otherwise it returns the time that the caller should
 | 
			
		||||
// wait until the tokens are actually available, and reports
 | 
			
		||||
// true.
 | 
			
		||||
func (tb *Bucket) TakeMaxDuration(count int64, maxWait time.Duration) (time.Duration, bool) {
 | 
			
		||||
	return tb.take(tb.clock.Now(), count, maxWait)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TakeAvailable takes up to count immediately available tokens from the
 | 
			
		||||
// bucket. It returns the number of tokens removed, or zero if there are
 | 
			
		||||
// no available tokens. It does not block.
 | 
			
		||||
func (tb *Bucket) TakeAvailable(count int64) int64 {
 | 
			
		||||
	return tb.takeAvailable(tb.clock.Now(), count)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// takeAvailable is the internal version of TakeAvailable - it takes the
 | 
			
		||||
// current time as an argument to enable easy testing.
 | 
			
		||||
func (tb *Bucket) takeAvailable(now time.Time, count int64) int64 {
 | 
			
		||||
	if count <= 0 {
 | 
			
		||||
		return 0
 | 
			
		||||
	}
 | 
			
		||||
	tb.mu.Lock()
 | 
			
		||||
	defer tb.mu.Unlock()
 | 
			
		||||
 | 
			
		||||
	tb.adjust(now)
 | 
			
		||||
	if tb.avail <= 0 {
 | 
			
		||||
		return 0
 | 
			
		||||
	}
 | 
			
		||||
	if count > tb.avail {
 | 
			
		||||
		count = tb.avail
 | 
			
		||||
	}
 | 
			
		||||
	tb.avail -= count
 | 
			
		||||
	return count
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Available returns the number of available tokens. It will be negative
 | 
			
		||||
// when there are consumers waiting for tokens. Note that if this
 | 
			
		||||
// returns greater than zero, it does not guarantee that calls that take
 | 
			
		||||
// tokens from the buffer will succeed, as the number of available
 | 
			
		||||
// tokens could have changed in the meantime. This method is intended
 | 
			
		||||
// primarily for metrics reporting and debugging.
 | 
			
		||||
func (tb *Bucket) Available() int64 {
 | 
			
		||||
	return tb.available(tb.clock.Now())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// available is the internal version of available - it takes the current time as
 | 
			
		||||
// an argument to enable easy testing.
 | 
			
		||||
func (tb *Bucket) available(now time.Time) int64 {
 | 
			
		||||
	tb.mu.Lock()
 | 
			
		||||
	defer tb.mu.Unlock()
 | 
			
		||||
	tb.adjust(now)
 | 
			
		||||
	return tb.avail
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Capacity returns the capacity that the bucket was created with.
 | 
			
		||||
func (tb *Bucket) Capacity() int64 {
 | 
			
		||||
	return tb.capacity
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Rate returns the fill rate of the bucket, in tokens per second.
 | 
			
		||||
func (tb *Bucket) Rate() float64 {
 | 
			
		||||
	return 1e9 * float64(tb.quantum) / float64(tb.fillInterval)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// take is the internal version of Take - it takes the current time as
 | 
			
		||||
// an argument to enable easy testing.
 | 
			
		||||
func (tb *Bucket) take(now time.Time, count int64, maxWait time.Duration) (time.Duration, bool) {
 | 
			
		||||
	if count <= 0 {
 | 
			
		||||
		return 0, true
 | 
			
		||||
	}
 | 
			
		||||
	tb.mu.Lock()
 | 
			
		||||
	defer tb.mu.Unlock()
 | 
			
		||||
 | 
			
		||||
	currentTick := tb.adjust(now)
 | 
			
		||||
	avail := tb.avail - count
 | 
			
		||||
	if avail >= 0 {
 | 
			
		||||
		tb.avail = avail
 | 
			
		||||
		return 0, true
 | 
			
		||||
	}
 | 
			
		||||
	// Round up the missing tokens to the nearest multiple
 | 
			
		||||
	// of quantum - the tokens won't be available until
 | 
			
		||||
	// that tick.
 | 
			
		||||
	endTick := currentTick + (-avail+tb.quantum-1)/tb.quantum
 | 
			
		||||
	endTime := tb.startTime.Add(time.Duration(endTick) * tb.fillInterval)
 | 
			
		||||
	waitTime := endTime.Sub(now)
 | 
			
		||||
	if waitTime > maxWait {
 | 
			
		||||
		return 0, false
 | 
			
		||||
	}
 | 
			
		||||
	tb.avail = avail
 | 
			
		||||
	return waitTime, true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// adjust adjusts the current bucket capacity based on the current time.
 | 
			
		||||
// It returns the current tick.
 | 
			
		||||
func (tb *Bucket) adjust(now time.Time) (currentTick int64) {
 | 
			
		||||
	currentTick = int64(now.Sub(tb.startTime) / tb.fillInterval)
 | 
			
		||||
 | 
			
		||||
	if tb.avail >= tb.capacity {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	tb.avail += (currentTick - tb.availTick) * tb.quantum
 | 
			
		||||
	if tb.avail > tb.capacity {
 | 
			
		||||
		tb.avail = tb.capacity
 | 
			
		||||
	}
 | 
			
		||||
	tb.availTick = currentTick
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										51
									
								
								vendor/github.com/juju/ratelimit/reader.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										51
									
								
								vendor/github.com/juju/ratelimit/reader.go
									
									
									
										generated
									
									
										vendored
									
									
								
							@@ -1,51 +0,0 @@
 | 
			
		||||
// Copyright 2014 Canonical Ltd.
 | 
			
		||||
// Licensed under the LGPLv3 with static-linking exception.
 | 
			
		||||
// See LICENCE file for details.
 | 
			
		||||
 | 
			
		||||
package ratelimit
 | 
			
		||||
 | 
			
		||||
import "io"
 | 
			
		||||
 | 
			
		||||
type reader struct {
 | 
			
		||||
	r      io.Reader
 | 
			
		||||
	bucket *Bucket
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Reader returns a reader that is rate limited by
 | 
			
		||||
// the given token bucket. Each token in the bucket
 | 
			
		||||
// represents one byte.
 | 
			
		||||
func Reader(r io.Reader, bucket *Bucket) io.Reader {
 | 
			
		||||
	return &reader{
 | 
			
		||||
		r:      r,
 | 
			
		||||
		bucket: bucket,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *reader) Read(buf []byte) (int, error) {
 | 
			
		||||
	n, err := r.r.Read(buf)
 | 
			
		||||
	if n <= 0 {
 | 
			
		||||
		return n, err
 | 
			
		||||
	}
 | 
			
		||||
	r.bucket.Wait(int64(n))
 | 
			
		||||
	return n, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type writer struct {
 | 
			
		||||
	w      io.Writer
 | 
			
		||||
	bucket *Bucket
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Writer returns a reader that is rate limited by
 | 
			
		||||
// the given token bucket. Each token in the bucket
 | 
			
		||||
// represents one byte.
 | 
			
		||||
func Writer(w io.Writer, bucket *Bucket) io.Writer {
 | 
			
		||||
	return &writer{
 | 
			
		||||
		w:      w,
 | 
			
		||||
		bucket: bucket,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (w *writer) Write(buf []byte) (int, error) {
 | 
			
		||||
	w.bucket.Wait(int64(len(buf)))
 | 
			
		||||
	return w.w.Write(buf)
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user