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.
19 lines
321 B
Go
19 lines
321 B
Go
4 years ago
|
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
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|