15370: Fix flaky test.

[arvados.git] / sdk / go / httpserver / request_limiter.go

diff --git a/sdk/go/httpserver/request_limiter.go b/sdk/go/httpserver/request_limiter.go
index 23e6e016d303bbc78abefcc39bfa0fb65b8ef0fe..8889453125c4753a62927f830c4e236ecbc272a6 100644 (file)
--- a/sdk/go/httpserver/request_limiter.go
+++ b/sdk/go/httpserver/request_limiter.go
@@ -76,8 +76,9 @@ func (h *limiterHandler) Max() int {
 func (h *limiterHandler) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
        if cap(h.requests) == 0 {
                atomic.AddInt64(&h.count, 1)
+               defer atomic.AddInt64(&h.count, -1)
                h.handler.ServeHTTP(resp, req)
-               atomic.AddInt64(&h.count, -1)
+               return
        }
        select {
        case h.requests <- struct{}{}: