recovery.go

package nice

import (
    "io"
    "log"
    "net/http/httputil"

    "github.com/go-errors/errors"

    "github.com/gin-gonic/gin"
)

func Recovery(f func(c *gin.Context, err interface{})) gin.HandlerFunc {
    return RecoveryWithWriter(f, gin.DefaultErrorWriter)
}

func RecoveryWithWriter(f func(c *gin.Context, err interface{}), out io.Writer) gin.HandlerFunc {
    var logger *log.Logger
    if out != nil {
        logger = log.New(out, "\n\n\x1b[31m", log.LstdFlags)
    }

    return func(c *gin.Context) {
        defer func() {
            if err := recover(); err != nil {
                if logger != nil {
                    httprequest, _ := httputil.DumpRequest(c.Request, false)
                    goErr := errors.Wrap(err, 3)
                    reset := string([]byte{27, 91, 48, 109})
                    logger.Printf("[Nice Recovery] panic recovered:\n\n%s%s\n\n%s%s", httprequest, goErr.Error(), goErr.Stack(), reset)
                }

                f(c, err)
            }
        }()
        c.Next() // execute all the handlers
    }
}