mirror of https://github.com/astaxie/beego.git
47 lines
843 B
Go
47 lines
843 B
Go
package response
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"net/http"
|
|
|
|
beecontext "github.com/astaxie/beego/context"
|
|
)
|
|
|
|
const (
|
|
NotFound StatusCode = http.StatusNotFound
|
|
BadRequest StatusCode = http.StatusBadRequest
|
|
)
|
|
|
|
// Renderer defines an http response renderer
|
|
type Renderer interface {
|
|
Render(ctx *beecontext.Context)
|
|
}
|
|
|
|
type rendererFunc func(ctx *beecontext.Context)
|
|
|
|
func (f rendererFunc) Render(ctx *beecontext.Context) {
|
|
f(ctx)
|
|
}
|
|
|
|
// StatusCode sets the http response status code
|
|
type StatusCode int
|
|
|
|
func (s StatusCode) Error() string {
|
|
return strconv.Itoa(int(s))
|
|
}
|
|
|
|
// Render sets the http status code
|
|
func (s StatusCode) Render(ctx *beecontext.Context) {
|
|
ctx.Output.SetStatus(int(s))
|
|
}
|
|
|
|
type statusCodeWithRender struct {
|
|
statusCode int
|
|
rendererFunc
|
|
}
|
|
|
|
func (s statusCodeWithRender) Error() string {
|
|
return strconv.Itoa(s.statusCode)
|
|
}
|