mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
8fc4bdcf97
* WIP: add k8s grouper - This shares a great deal of behaviour with allGrouper. Once it's tested, refactor that to share as much as possible - Glide hell. Checked in the yaml and lock files but a glide i -v will be required to bring vendor/ up-to-date. Will address once this is ready. * Update README. Make the watch tracking work. (To follow: add the junk that was pulled in via the glide update.) * Vendor updates. * go fmt * Use the allGrouper with a k8s-backed DBStore instead. This is much tidier :-) * Fix up go vet
52 lines
999 B
Go
52 lines
999 B
Go
// 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)
|
|
}
|