/usr/share/gocode/src/github.com/jacobsa/ratelimit/throttled_reader.go is in golang-github-jacobsa-ratelimit-dev 0.0~git20150723.0.2ca5e0c-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | // Copyright 2015 Google Inc. All Rights Reserved.
//
// Licensed 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 ratelimit
import (
"io"
"golang.org/x/net/context"
)
// Create a reader that limits the bandwidth of reads made from r according to
// the supplied throttler. Reads are assumed to be made under the supplied
// context.
func ThrottledReader(
ctx context.Context,
r io.Reader,
throttle Throttle) io.Reader {
return &throttledReader{
ctx: ctx,
wrapped: r,
throttle: throttle,
}
}
type throttledReader struct {
ctx context.Context
wrapped io.Reader
throttle Throttle
}
func (tr *throttledReader) Read(p []byte) (n int, err error) {
// We can't serve a read larger than the throttle's capacity.
if uint64(len(p)) > tr.throttle.Capacity() {
p = p[:int(tr.throttle.Capacity())]
}
// Wait for permission to continue.
err = tr.throttle.Wait(tr.ctx, uint64(len(p)))
if err != nil {
return
}
// Serve the full amount we acquired from the throttle (unless we hit an
// early error, including EOF).
for len(p) > 0 && err == nil {
var tmp int
tmp, err = tr.wrapped.Read(p)
n += tmp
p = p[tmp:]
}
return
}
|