go-zero/rest/handler/timeouthandler.go

package handler

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
		}
	}
}