Switch to using the dep tool and update all the dependencies

vendor/github.com/tsenart/tb/http/client.go

@@ -0,0 +1,45 @@
+package http
+
+import (
+	"github.com/tsenart/tb"
+	"net/http"
+	"time"
+)
+
+type roundTripperFunc func(r *http.Request) (*http.Response, error)
+
+func (f roundTripperFunc) RoundTrip(r *http.Request) (*http.Response, error) {
+	return f(r)
+}
+
+// ByteThrottledRoundTripper wraps another RoundTripper rt,
+// throttling all requests to the specified byte rate.
+func ByteThrottledRoundTripper(rt http.RoundTripper, rate int64) http.RoundTripper {
+	freq := time.Duration(1 * time.Millisecond)
+	bucket := tb.NewBucket(rate, freq)
+
+	return roundTripperFunc(func(r *http.Request) (*http.Response, error) {
+		got := bucket.Take(r.ContentLength)
+		for got < r.ContentLength {
+			got += bucket.Take(r.ContentLength - got)
+			time.Sleep(freq)
+		}
+		return rt.RoundTrip(r)
+	})
+}
+
+// ReqThrottledRoundTripper wraps another RoundTripper rt,
+// throttling all requests to the specified request rate.
+func ReqThrottledRoundTripper(rt http.RoundTripper, rate int64) http.RoundTripper {
+	freq := time.Duration(1e9 / rate)
+	bucket := tb.NewBucket(rate, freq)
+
+	return roundTripperFunc(func(r *http.Request) (*http.Response, error) {
+		got := bucket.Take(1)
+		for got != 1 {
+			got = bucket.Take(1)
+			time.Sleep(freq)
+		}
+		return rt.RoundTrip(r)
+	})
+}

vendor/github.com/tsenart/tb/http/server.go

@@ -0,0 +1,38 @@
+package http
+
+import (
+	"github.com/tsenart/tb"
+	"net"
+	"net/http"
+	"time"
+)
+
+var byteThrottler = tb.NewThrottler(25 * time.Millisecond)
+
+// ByteThrottledHandler wraps an http.Handler with per host byte throttling to
+// the specified byte rate, responding with 429 when throttled.
+func ByteThrottledHandler(h http.Handler, rate int64) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		host, _, _ := net.SplitHostPort(r.RemoteAddr)
+		if byteThrottler.Halt(host, r.ContentLength, rate) {
+			http.Error(w, "Too many requests", 429)
+			return
+		}
+		h.ServeHTTP(w, r)
+	})
+}
+
+var reqThrottler = tb.NewThrottler(5 * time.Millisecond)
+
+// ReqThrottledHandler wraps an http.Handler with per host request throttling
+// to the specified request rate, responding with 429 when throttled.
+func ReqThrottledHandler(h http.Handler, rate int64) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		host, _, _ := net.SplitHostPort(r.RemoteAddr)
+		if reqThrottler.Halt(host, 1, rate) {
+			http.Error(w, "Too many requests", 429)
+			return
+		}
+		h.ServeHTTP(w, r)
+	})
+}