Recover middleware: Added, fiber.Ctx as a first paremeter to StackTra…

…ceHandler. (#1623)
gofiber · Nov 29, 2021 · c9e959c · c9e959c
1 parent 5ff3bc8
commit c9e959c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/middleware/recover/README.md b/middleware/recover/README.md
@@ -50,7 +50,7 @@ type Config struct {
 	// StackTraceHandler defines a function to handle stack trace
 	//
 	// Optional. Default: defaultStackTraceHandler
-	StackTraceHandler func(e interface{})
+	StackTraceHandler func(c *fiber.Ctx, e interface{})
 }
 ```
 

diff --git a/middleware/recover/config.go b/middleware/recover/config.go
@@ -19,7 +19,7 @@ type Config struct {
 	// StackTraceHandler defines a function to handle stack trace
 	//
 	// Optional. Default: defaultStackTraceHandler
-	StackTraceHandler func(e interface{})
+	StackTraceHandler func(c *fiber.Ctx, e interface{})
 }
 
 var defaultStackTraceBufLen = 1024

diff --git a/middleware/recover/recover.go b/middleware/recover/recover.go
@@ -8,7 +8,7 @@ import (
 	"github.com/gofiber/fiber/v2"
 )
 
-func defaultStackTraceHandler(e interface{}) {
+func defaultStackTraceHandler(c *fiber.Ctx, e interface{}) {
 	buf := make([]byte, defaultStackTraceBufLen)
 	buf = buf[:runtime.Stack(buf, false)]
 	_, _ = os.Stderr.WriteString(fmt.Sprintf("panic: %v\n%s\n", e, buf))
@@ -30,7 +30,7 @@ func New(config ...Config) fiber.Handler {
 		defer func() {
 			if r := recover(); r != nil {
 				if cfg.EnableStackTrace {
-					cfg.StackTraceHandler(r)
+					cfg.StackTraceHandler(c, r)
 				}
 
 				var ok bool