go-zero/rest/handler/timeouthandler.go

package handler

import (
	"net/http"
	"time"
)

const reason = "Request Timeout"

// TimeoutHandler returns the handler with given 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)
		}

		return next
	}
}