Skip to content

Commit 914a544

Browse files
committed
Updated recover middleware
Signed-off-by: Vishal Rana <[email protected]>
1 parent b5de47f commit 914a544

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

middleware/recover.go

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@ import (
55
"runtime"
66

77
"github.com/labstack/echo"
8-
"github.com/labstack/gommon/color"
98
)
109

1110
type (
@@ -57,23 +56,21 @@ func RecoverWithConfig(config RecoverConfig) echo.MiddlewareFunc {
5756
}
5857

5958
return func(next echo.HandlerFunc) echo.HandlerFunc {
60-
return func(c echo.Context) (err error) {
59+
return func(c echo.Context) error {
6160
if config.Skipper(c) {
6261
return next(c)
6362
}
6463

6564
defer func() {
6665
if r := recover(); r != nil {
67-
switch r := r.(type) {
68-
case error:
69-
err = r
70-
default:
66+
err, ok := r.(error)
67+
if !ok {
7168
err = fmt.Errorf("%v", r)
7269
}
7370
stack := make([]byte, config.StackSize)
7471
length := runtime.Stack(stack, !config.DisableStackAll)
7572
if !config.DisablePrintStack {
76-
c.Logger().Printf("[%s] %s %s\n", color.Red("PANIC RECOVER"), err, stack[:length])
73+
c.Logger().Printf("[PANIC RECOVER] %v %s\n", err, stack[:length])
7774
}
7875
c.Error(err)
7976
}

0 commit comments

Comments
 (0)