update

Author: admpub

context_x_response.go

@@ -25,7 +25,7 @@ func (c *XContext) Response() engine.Response {
 // code. Templates can be registered using `Echo.SetRenderer()`.
 func (c *XContext) Render(name string, data interface{}, codes ...int) error {
 	if c.auto {
-		if ok, err := c.echo.AutoDetectRenderFormat(c, data); ok {
+		if ok, err := c.echo.AutoDetectRenderFormat(c, data, codes...); ok {
 			return err
 		}
 	}

echo.go

@@ -287,7 +287,7 @@ func (e *Echo) SetFormatRenderers(formatRenderers map[string]FormatRender) *Echo
 	return e
 }
 
-func (e *Echo) AddFormatRenderer(format string, renderer func(c Context, data interface{}) error) *Echo {
+func (e *Echo) AddFormatRenderer(format string, renderer FormatRender) *Echo {
 	e.formatRenderers[format] = renderer
 	return e
 }
@@ -299,7 +299,7 @@ func (e *Echo) RemoveFormatRenderer(formats ...string) *Echo {
 	return e
 }
 
-func (e *Echo) AutoDetectRenderFormat(c Context, data interface{}) (bool, error) {
+func (e *Echo) AutoDetectRenderFormat(c Context, data interface{}, code ...int) (bool, error) {
 	format := c.Format()
 	render, ok := e.formatRenderers[format]
 	if !ok || render == nil {
@@ -316,7 +316,7 @@ func (e *Echo) AutoDetectRenderFormat(c Context, data interface{}) (bool, error)
 	default:
 		c.Data().SetData(data, c.Data().GetCode().Int())
 	}
-	return true, render(c, data)
+	return true, render(c, data, code...)
 }
 
 func (e *Echo) SetDefaultExtension(ext string) {

middleware/render/middleware.go

@@ -118,7 +118,7 @@ func HTTPErrorHandler(opt *Options) echo.HTTPErrorHandler {
 	}
 	tmplNum := len(opt.ErrorPages)
 	defaultRender := func(c echo.Context, msg string, code int) {
-		if ok, err := c.Echo().AutoDetectRenderFormat(c, nil); ok {
+		if ok, err := c.Echo().AutoDetectRenderFormat(c, nil, code); ok {
 			if err == nil {
 				return
 			}

render_data.go

@@ -14,7 +14,7 @@ import (
 	"github.com/webx-top/echo/param"
 )
 
-type FormatRender func(ctx Context, data interface{}) error
+type FormatRender func(ctx Context, data interface{}, code ...int) error
 type DataWrapper func(Context, interface{}) interface{}
 
 func DefaultRenderDataWrapper(ctx Context, data interface{}) interface{} {

variable_default.go

@@ -54,17 +54,17 @@ var (
 		`*`: ContentTypeHTML,
 	}
 	DefaultFormatRenderers = map[string]FormatRender{
-		ContentTypeJSON: func(c Context, data interface{}) error {
-			return c.JSON(c.Data())
+		ContentTypeJSON: func(c Context, data interface{}, code ...int) error {
+			return c.JSON(c.Data(), code...)
 		},
-		ContentTypeJSONP: func(c Context, data interface{}) error {
-			return c.JSONP(c.Query(c.Echo().JSONPVarName), c.Data())
+		ContentTypeJSONP: func(c Context, data interface{}, code ...int) error {
+			return c.JSONP(c.Query(c.Echo().JSONPVarName), c.Data(), code...)
 		},
-		ContentTypeXML: func(c Context, data interface{}) error {
-			return c.XML(c.Data())
+		ContentTypeXML: func(c Context, data interface{}, code ...int) error {
+			return c.XML(c.Data(), code...)
 		},
-		ContentTypeText: func(c Context, data interface{}) error {
-			return c.String(fmt.Sprint(data))
+		ContentTypeText: func(c Context, data interface{}, code ...int) error {
+			return c.String(fmt.Sprint(data), code...)
 		},
 	}
 	DefaultBinderDecoders = map[string]func(interface{}, Context, BinderValueCustomDecoders, ...FormDataFilter) error{