You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/zeromicro/go-zero/core/breaker"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/core/stat"
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|
"github.com/zeromicro/go-zero/rest/internal/security"
|
|
)
|
|
|
|
const breakerSeparator = "://"
|
|
|
|
// BreakerHandler returns a break circuit middleware.
|
|
func BreakerHandler(method, path string, metrics *stat.Metrics) func(http.Handler) http.Handler {
|
|
brk := breaker.NewBreaker(breaker.WithName(strings.Join([]string{method, path}, breakerSeparator)))
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
promise, err := brk.Allow()
|
|
if err != nil {
|
|
metrics.AddDrop()
|
|
logx.Errorf("[http] dropped, %s - %s - %s",
|
|
r.RequestURI, httpx.GetRemoteAddr(r), r.UserAgent())
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
return
|
|
}
|
|
|
|
cw := &security.WithCodeResponseWriter{Writer: w}
|
|
defer func() {
|
|
if cw.Code < http.StatusInternalServerError {
|
|
promise.Accept()
|
|
} else {
|
|
promise.Reject(fmt.Sprintf("%d %s", cw.Code, http.StatusText(cw.Code)))
|
|
}
|
|
}()
|
|
next.ServeHTTP(cw, r)
|
|
})
|
|
}
|
|
}
|