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/api/meta"
|
||||||
"k8s.io/kubernetes/pkg/auth/authorizer"
|
"k8s.io/kubernetes/pkg/auth/authorizer"
|
||||||
"k8s.io/kubernetes/pkg/httplog"
|
"k8s.io/kubernetes/pkg/httplog"
|
||||||
"k8s.io/kubernetes/pkg/util"
|
|
||||||
"k8s.io/kubernetes/pkg/util/sets"
|
"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) {
|
func tooManyRequests(w http.ResponseWriter) {
|
||||||
// Return a 429 status indicating "Too Many Requests"
|
// Return a 429 status indicating "Too Many Requests"
|
||||||
w.Header().Set("Retry-After", RetryAfter)
|
w.Header().Set("Retry-After", RetryAfter)
|
||||||
|
|
|
@ -110,21 +110,6 @@ func TestMaxInFlight(t *testing.T) {
|
||||||
expectHTTP(server.URL, http.StatusOK, 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) {
|
func TestReadOnly(t *testing.T) {
|
||||||
server := httptest.NewServer(ReadOnly(http.HandlerFunc(
|
server := httptest.NewServer(ReadOnly(http.HandlerFunc(
|
||||||
func(w http.ResponseWriter, req *http.Request) {
|
func(w http.ResponseWriter, req *http.Request) {
|
||||||
|
|
Loading…
Reference in New Issue