931818060

Committed 12 Jun 2021 07:56PM UTC coverage: 87.747% (-0.3%) from 88.025%

Build Type

github

Committed by Umputun

Commit Message

lint: close body suppress error check

Pull Request Pull Request #17: Throttle

Run Details

30 of 30 new or added lines in 1 file covered. (100.0%)

444 of 506 relevant lines covered (87.75%)

6.99 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

83.33

/throttle.go

package rest

import (
        "net/http"
)

// Throttle middleware checks how many request in-fly and rejects with 503 if exceeded
func Throttle(limit int64) func(http.Handler) http.Handler {

        ch := make(chan struct{}, limit)
        return func(h http.Handler) http.Handler {

                fn := func(w http.ResponseWriter, r *http.Request) {

                        if limit <= 0 {
                                h.ServeHTTP(w, r)
                                return
                        }

                        var acquired bool
                        defer func() {
                                if !acquired {
                                        return
                                }
                                select {
                                case <-ch:
                                        return
                                default:
                                        return
                                }
                        }()

                        select {
                        case ch <- struct{}{}:
                                acquired = true
                                h.ServeHTTP(w, r)
                                return
                        default:
                                w.WriteHeader(http.StatusServiceUnavailable)
                                return
                        }
                }

                return http.HandlerFunc(fn)
        }
}

1	package rest
2
3	import (
4	"net/http"
5	)
6
7	// Throttle middleware checks how many request in-fly and rejects with 503 if exceeded
8	func Throttle(limit int64) func(http.Handler) http.Handler {	1✔
9		1✔
10	ch := make(chan struct{}, limit)	1✔
11	return func(h http.Handler) http.Handler {	2✔
12		1✔
13	fn := func(w http.ResponseWriter, r *http.Request) {	103✔
14		102✔
15	if limit <= 0 {	102✔
16	h.ServeHTTP(w, r)	×
17	return	×
18	}	×
19
20	var acquired bool	102✔
21	defer func() {	204✔
22	if !acquired {	192✔
23	return	90✔
24	}	90✔
25	select {	12✔
26	case <-ch:	12✔
27	return	12✔
28	default:	×
29	return	×
30	}
31	}()
32
33	select {	102✔
34	case ch <- struct{}{}:	12✔
35	acquired = true	12✔
36	h.ServeHTTP(w, r)	12✔
37	return	12✔
38	default:	90✔
39	w.WriteHeader(http.StatusServiceUnavailable)	90✔
40	return	90✔
41	}
42	}
43
44	return http.HandlerFunc(fn)	1✔
45	}
46	}

go-pkgz / rest / 931818060

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous