register the panic-recover handler after the activity logger

This commit is contained in:
Gani Georgiev
2024-10-18 13:47:10 +03:00
parent dbc074ee9a
commit 6f2fe91da5
6 changed files with 83 additions and 43 deletions
-35
View File
@@ -4,12 +4,10 @@ import (
"bufio"
"encoding/json"
"errors"
"fmt"
"io"
"log"
"net"
"net/http"
"runtime"
"github.com/pocketbase/pocketbase/tools/hook"
)
@@ -129,12 +127,6 @@ func (r *Router[T]) loadMux(mux *http.ServeMux, group *RouterGroup[T], parents [
}
}
// add global panic-recover middleware
routeHook.Bind(&hook.Handler[T]{
Func: r.panicHandler,
Priority: -9999999, // before everything else
})
mux.HandleFunc(pattern, func(resp http.ResponseWriter, req *http.Request) {
// wrap the response to add write and status tracking
resp = &ResponseWriter{ResponseWriter: resp}
@@ -162,33 +154,6 @@ func (r *Router[T]) loadMux(mux *http.ServeMux, group *RouterGroup[T], parents [
return nil
}
// panicHandler registers a default panic-recover handling.
func (r *Router[T]) panicHandler(event T) (err error) {
// panic-recover
defer func() {
recoverResult := recover()
if recoverResult == nil {
return
}
recoverErr, ok := recoverResult.(error)
if !ok {
recoverErr = fmt.Errorf("%v", recoverResult)
} else if errors.Is(recoverErr, http.ErrAbortHandler) {
// don't recover ErrAbortHandler so the response to the client can be aborted
panic(recoverResult)
}
stack := make([]byte, 2<<10) // 2 KB
length := runtime.Stack(stack, true)
err = NewInternalServerError("", fmt.Errorf("[PANIC RECOVER] %w %s", recoverErr, stack[:length]))
}()
err = event.Next()
return err
}
func ErrorHandler(resp http.ResponseWriter, req *http.Request, err error) {
if err == nil {
return
-5
View File
@@ -64,10 +64,6 @@ func TestRouter(t *testing.T) {
calls += "/" + e.Request.PathValue("param")
return errors.New("test") // should be normalized to an ApiError
})
g1.GET("/panic", func(e *router.Event) error {
calls += "/panic"
panic("test")
})
mux, err := r.BuildMux()
if err != nil {
@@ -98,7 +94,6 @@ func TestRouter(t *testing.T) {
{http.MethodHead, "/a/b/1", "root_m:a_b_group_m:1_get_m:/1_get:cleanup"},
{http.MethodPost, "/a/b/1", "root_m:a_b_group_m:/1_post:cleanup"},
{http.MethodGet, "/a/b/456", "root_m:a_b_group_m:/456/error:cleanup"},
{http.MethodGet, "/a/b/panic", "root_m:a_b_group_m:/panic:cleanup"},
}
for _, s := range scenarios {