forked from mirrors/homebox
6529549289
* implement custom http handler interface * implement trace_id * normalize http method spacing for consistent logs * fix failing test * fix linter errors * cleanup old dead code * more route cleanup * cleanup some inconsistent errors * update and generate code * make taskfile more consistent * update task calls * run tidy * drop `@` tag for version * use relative paths * tidy * fix auto-setting variables * update build paths * add contributing guide * tidy
33 lines
753 B
Go
33 lines
753 B
Go
package mid
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"runtime/debug"
|
|
|
|
"github.com/hay-kot/homebox/backend/pkgs/server"
|
|
)
|
|
|
|
// Panic is a middleware that recovers from panics anywhere in the chain and wraps the error.
|
|
// and returns it up the middleware chain.
|
|
func Panic(develop bool) server.Middleware {
|
|
return func(h server.Handler) server.Handler {
|
|
return server.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (err error) {
|
|
defer func() {
|
|
if rec := recover(); rec != nil {
|
|
trace := debug.Stack()
|
|
|
|
if develop {
|
|
err = fmt.Errorf("PANIC [%v]", rec)
|
|
fmt.Printf("%s", string(trace))
|
|
} else {
|
|
err = fmt.Errorf("PANIC [%v] TRACE[%s]", rec, string(trace))
|
|
}
|
|
|
|
}
|
|
}()
|
|
|
|
return h.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|