mirror of
https://github.com/go-gitea/gitea
synced 2024-08-18 13:20:51 +02:00
* update code.gitea.io/sdk/gitea v0.13.1 -> v0.13.2 * update github.com/go-swagger/go-swagger v0.25.0 -> v0.26.0 * update github.com/google/uuid v1.1.2 -> v1.2.0 * update github.com/klauspost/compress v1.11.3 -> v1.11.7 * update github.com/lib/pq 083382b7e6fc -> v1.9.0 * update github.com/markbates/goth v1.65.0 -> v1.66.1 * update github.com/mattn/go-sqlite3 v1.14.4 -> v1.14.6 * update github.com/mgechev/revive 246eac737dc7 -> v1.0.3 * update github.com/minio/minio-go/v7 v7.0.6 -> v7.0.7 * update github.com/niklasfasching/go-org v1.3.2 -> v1.4.0 * update github.com/olivere/elastic/v7 v7.0.21 -> v7.0.22 * update github.com/pquerna/otp v1.2.0 -> v1.3.0 * update github.com/xanzy/go-gitlab v0.39.0 -> v0.42.0 * update github.com/yuin/goldmark v1.2.1 -> v1.3.1
97 lines
2.2 KiB
Go
Vendored
97 lines
2.2 KiB
Go
Vendored
package handlers
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"runtime/debug"
|
|
)
|
|
|
|
// RecoveryHandlerLogger is an interface used by the recovering handler to print logs.
|
|
type RecoveryHandlerLogger interface {
|
|
Println(...interface{})
|
|
}
|
|
|
|
type recoveryHandler struct {
|
|
handler http.Handler
|
|
logger RecoveryHandlerLogger
|
|
printStack bool
|
|
}
|
|
|
|
// RecoveryOption provides a functional approach to define
|
|
// configuration for a handler; such as setting the logging
|
|
// whether or not to print stack traces on panic.
|
|
type RecoveryOption func(http.Handler)
|
|
|
|
func parseRecoveryOptions(h http.Handler, opts ...RecoveryOption) http.Handler {
|
|
for _, option := range opts {
|
|
option(h)
|
|
}
|
|
|
|
return h
|
|
}
|
|
|
|
// RecoveryHandler is HTTP middleware that recovers from a panic,
|
|
// logs the panic, writes http.StatusInternalServerError, and
|
|
// continues to the next handler.
|
|
//
|
|
// Example:
|
|
//
|
|
// r := mux.NewRouter()
|
|
// r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
// panic("Unexpected error!")
|
|
// })
|
|
//
|
|
// http.ListenAndServe(":1123", handlers.RecoveryHandler()(r))
|
|
func RecoveryHandler(opts ...RecoveryOption) func(h http.Handler) http.Handler {
|
|
return func(h http.Handler) http.Handler {
|
|
r := &recoveryHandler{handler: h}
|
|
return parseRecoveryOptions(r, opts...)
|
|
}
|
|
}
|
|
|
|
// RecoveryLogger is a functional option to override
|
|
// the default logger
|
|
func RecoveryLogger(logger RecoveryHandlerLogger) RecoveryOption {
|
|
return func(h http.Handler) {
|
|
r := h.(*recoveryHandler)
|
|
r.logger = logger
|
|
}
|
|
}
|
|
|
|
// PrintRecoveryStack is a functional option to enable
|
|
// or disable printing stack traces on panic.
|
|
func PrintRecoveryStack(print bool) RecoveryOption {
|
|
return func(h http.Handler) {
|
|
r := h.(*recoveryHandler)
|
|
r.printStack = print
|
|
}
|
|
}
|
|
|
|
func (h recoveryHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
h.log(err)
|
|
}
|
|
}()
|
|
|
|
h.handler.ServeHTTP(w, req)
|
|
}
|
|
|
|
func (h recoveryHandler) log(v ...interface{}) {
|
|
if h.logger != nil {
|
|
h.logger.Println(v...)
|
|
} else {
|
|
log.Println(v...)
|
|
}
|
|
|
|
if h.printStack {
|
|
stack := string(debug.Stack())
|
|
if h.logger != nil {
|
|
h.logger.Println(stack)
|
|
} else {
|
|
log.Println(stack)
|
|
}
|
|
}
|
|
}
|