1
0
Fork 0
forked from forgejo/forgejo

Update chi/middleware to chi/v5/middleware (#17888)

Fix #17880


Co-authored-by: Lauris BH <lauris@nix.lv>
This commit is contained in:
zeripath 2021-12-02 20:58:08 +00:00 committed by GitHub
parent 4646c7c52d
commit 957c3fcb59
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
42 changed files with 128 additions and 2572 deletions

View file

@ -0,0 +1,26 @@
package middleware
import (
"net/http"
"strings"
)
// Heartbeat endpoint middleware useful to setting up a path like
// `/ping` that load balancers or uptime testing external services
// can make a request before hitting any routes. It's also convenient
// to place this above ACL middlewares as well.
func Heartbeat(endpoint string) func(http.Handler) http.Handler {
f := func(h http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) {
if r.Method == "GET" && strings.EqualFold(r.URL.Path, endpoint) {
w.Header().Set("Content-Type", "text/plain")
w.WriteHeader(http.StatusOK)
w.Write([]byte("."))
return
}
h.ServeHTTP(w, r)
}
return http.HandlerFunc(fn)
}
return f
}