Files
go-zero/core/httphandler/timeouthandler.go
2020-07-26 17:09:05 +08:00

19 lines
321 B
Go

package httphandler
import (
"net/http"
"time"
)
const reason = "Request Timeout"
func TimeoutHandler(duration time.Duration) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
if duration > 0 {
return http.TimeoutHandler(next, duration, reason)
} else {
return next
}
}
}