Add status code to error handler to be able to send 4xx errors too for example

Author: Mario de Frutos

context.go

@@ -80,8 +80,8 @@ func (c *Context) NoContent(code int) error {
 }
 
 // Error invokes the registered HTTP error handler.
-func (c *Context) Error(err error) {
-	c.echo.httpErrorHandler(err, c)
+func (c *Context) Error(err error, code int) {
+	c.echo.httpErrorHandler(err, code, c)
 }
 
 // Get retrieves data from the context.

echo.go

@@ -34,7 +34,7 @@ type (
 	HandlerFunc    func(*Context) error
 
 	// HTTPErrorHandler is a centralized HTTP error handler.
-	HTTPErrorHandler func(error, *Context)
+	HTTPErrorHandler func(error, int, *Context)
 
 	BindFunc func(*http.Request, interface{}) error
 
@@ -122,11 +122,11 @@ func New() (e *Echo) {
 	e.NotFoundHandler(func(c *Context) {
 		http.Error(c.Response, http.StatusText(http.StatusNotFound), http.StatusNotFound)
 	})
-	e.HTTPErrorHandler(func(err error, c *Context) {
+	e.HTTPErrorHandler(func(err error, code int, c *Context) {
 		if err != nil {
 			// TODO: Warning
 			log.Printf("echo: %s", color.Yellow("http error handler not registered"))
-			http.Error(c.Response, err.Error(), http.StatusInternalServerError)
+			http.Error(c.Response, err.Error(), code)
 		}
 	})
 	e.Binder(func(r *http.Request, v interface{}) error {
@@ -303,7 +303,7 @@ func (e *Echo) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	// Execute chain
 	if err := h(c); err != nil {
-		e.httpErrorHandler(err, c)
+		e.httpErrorHandler(err, http.StatusInternalServerError, c)
 	}
 
 	e.pool.Put(c)

middleware/logger.go

@@ -2,6 +2,7 @@ package middleware
 
 import (
 	"log"
+	"net/http"
 	"time"
 
 	"github.com/labstack/echo"
@@ -12,7 +13,7 @@ func Logger(h echo.HandlerFunc) echo.HandlerFunc {
 	return func(c *echo.Context) error {
 		start := time.Now()
 		if err := h(c); err != nil {
-			c.Error(err)
+			c.Error(err, http.StatusInternalServerError)
 		}
 		end := time.Now()
 		m := c.Request.Method