mirror of https://github.com/k3s-io/k3s
Merge pull request #14418 from gmarek/dead-code
Remove dead RateLimit function from API server handlerspull/6/head
commit
72992ae8f6
|
@ -34,7 +34,6 @@ import (
|
|||
"k8s.io/kubernetes/pkg/api/meta"
|
||||
"k8s.io/kubernetes/pkg/auth/authorizer"
|
||||
"k8s.io/kubernetes/pkg/httplog"
|
||||
"k8s.io/kubernetes/pkg/util"
|
||||
"k8s.io/kubernetes/pkg/util/sets"
|
||||
)
|
||||
|
||||
|
@ -96,17 +95,6 @@ func MaxInFlightLimit(c chan bool, longRunningRequestRE *regexp.Regexp, handler
|
|||
})
|
||||
}
|
||||
|
||||
// RateLimit uses rl to rate limit accepting requests to 'handler'.
|
||||
func RateLimit(rl util.RateLimiter, handler http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
||||
if rl.CanAccept() {
|
||||
handler.ServeHTTP(w, req)
|
||||
return
|
||||
}
|
||||
tooManyRequests(w)
|
||||
})
|
||||
}
|
||||
|
||||
func tooManyRequests(w http.ResponseWriter) {
|
||||
// Return a 429 status indicating "Too Many Requests"
|
||||
w.Header().Set("Retry-After", RetryAfter)
|
||||
|
|
|
@ -110,21 +110,6 @@ func TestMaxInFlight(t *testing.T) {
|
|||
expectHTTP(server.URL, http.StatusOK, t)
|
||||
}
|
||||
|
||||
func TestRateLimit(t *testing.T) {
|
||||
for _, allow := range []bool{true, false} {
|
||||
rl := fakeRL(allow)
|
||||
server := httptest.NewServer(RateLimit(rl, http.HandlerFunc(
|
||||
func(w http.ResponseWriter, req *http.Request) {
|
||||
if !allow {
|
||||
t.Errorf("Unexpected call")
|
||||
}
|
||||
},
|
||||
)))
|
||||
defer server.Close()
|
||||
http.Get(server.URL)
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadOnly(t *testing.T) {
|
||||
server := httptest.NewServer(ReadOnly(http.HandlerFunc(
|
||||
func(w http.ResponseWriter, req *http.Request) {
|
||||
|
|
Loading…
Reference in New Issue