64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/uptrace/bunrouter"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type HTTPError struct {
|
|
StatusCode int `json:"code"`
|
|
Message string `json:"message"`
|
|
OriginError error `json:"-"`
|
|
}
|
|
|
|
func (e HTTPError) Error() string {
|
|
return e.Message
|
|
}
|
|
|
|
func NewHTTPError(err error) HTTPError {
|
|
return HTTPError{
|
|
StatusCode: http.StatusInternalServerError,
|
|
Message: "Internal server error with unknown reason",
|
|
OriginError: err,
|
|
}
|
|
}
|
|
|
|
func (self *Handlers) ErrorHandler(
|
|
next bunrouter.HandlerFunc,
|
|
) bunrouter.HandlerFunc {
|
|
return func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
err := next(w, req)
|
|
|
|
var httpErr HTTPError
|
|
switch err := err.(type) {
|
|
case nil:
|
|
return nil
|
|
|
|
case HTTPError:
|
|
httpErr = err
|
|
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "unhandled error, %v\n", err)
|
|
zap.L().Error("unhandled error",
|
|
zap.Error(err))
|
|
httpErr = NewHTTPError(err)
|
|
}
|
|
|
|
if httpErr.OriginError == nil {
|
|
zap.L().Warn(httpErr.Message)
|
|
} else {
|
|
zap.L().Warn(httpErr.Message,
|
|
zap.Error(httpErr.OriginError))
|
|
}
|
|
|
|
w.WriteHeader(httpErr.StatusCode)
|
|
_ = bunrouter.JSON(w, httpErr)
|
|
|
|
return err
|
|
}
|
|
}
|