vinchent/gin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adds Status() method to Context

Browse Source
This commit is contained in:
Manu Mtz.-Almeida 2016-01-26 14:55:45 +01:00
parent 704d690ac0
commit d0cd8b9fd9
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
context.go
View File

@ -112,7 +112,7 @@ func (c *Context) Abort() {
// AbortWithStatus calls `Abort()` and writes the headers with the specified status code. // AbortWithStatus calls `Abort()` and writes the headers with the specified status code.
// For example, a failed attempt to authentificate a request could use: context.AbortWithStatus(401). // For example, a failed attempt to authentificate a request could use: context.AbortWithStatus(401).
func (c *Context) AbortWithStatus(code int) { func (c *Context) AbortWithStatus(code int) {
c.Writer.WriteHeader(code) c.Status(code)
c.Abort() c.Abort()
} }
@ -310,6 +310,10 @@ func (c *Context) requestHeader(key string) string {
/******** RESPONSE RENDERING ********/ /******** RESPONSE RENDERING ********/
/************************************/ /************************************/
func (c *Context) Status(code int) {
c.writermem.WriteHeader(code)
}
// Header is a intelligent shortcut for c.Writer.Header().Set(key, value) // Header is a intelligent shortcut for c.Writer.Header().Set(key, value)
// It writes a header in the response. // It writes a header in the response.
// If value == "", this method removes the header `c.Writer.Header().Del(key)` // If value == "", this method removes the header `c.Writer.Header().Del(key)`
@ -322,7 +326,7 @@ func (c *Context) Header(key, value string) {
} }
func (c *Context) Render(code int, r render.Render) { func (c *Context) Render(code int, r render.Render) {
c.writermem.WriteHeader(code) c.Status(code)
if err := r.Render(c.Writer); err != nil { if err := r.Render(c.Writer); err != nil {
c.renderError(err) c.renderError(err)
} }
@ -352,7 +356,7 @@ func (c *Context) IndentedJSON(code int, obj interface{}) {
// JSON serializes the given struct as JSON into the response body. // JSON serializes the given struct as JSON into the response body.
// It also sets the Content-Type as "application/json". // It also sets the Content-Type as "application/json".
func (c *Context) JSON(code int, obj interface{}) { func (c *Context) JSON(code int, obj interface{}) {
c.writermem.WriteHeader(code) c.Status(code)
if err := render.WriteJSON(c.Writer, obj); err != nil { if err := render.WriteJSON(c.Writer, obj); err != nil {
c.renderError(err) c.renderError(err)
} }
@ -366,7 +370,7 @@ func (c *Context) XML(code int, obj interface{}) {
// String writes the given string into the response body. // String writes the given string into the response body.
func (c *Context) String(code int, format string, values ...interface{}) { func (c *Context) String(code int, format string, values ...interface{}) {
c.writermem.WriteHeader(code) c.Status(code)
render.WriteString(c.Writer, format, values) render.WriteString(c.Writer, format, values)
} }