blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-15 04:33:40 +08:00
Code Issues Projects Releases Wiki Activity

Merge c4d81efecaff52c9db4bb91109cd040620a787b3 into c3abaf9afcd167475951249d5a406ba10b24a1d9

Browse Source
This commit is contained in:
jasonrhansen 2014-07-31 18:29:36 +00:00
commit ceb1212460
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
response_writer.go
View File

@ -9,6 +9,8 @@ type (
http.ResponseWriter
Status() int
Written() bool
// Before allows for a function to be called before the ResponseWriter has been written to.
Before(BeforeFunc)
// private
setStatus(int)
@ -16,15 +18,19 @@ type (
responseWriter struct {
http.ResponseWriter
status int
written bool
status int
written bool
beforeFuncs []BeforeFunc
}
BeforeFunc func(ResponseWriter)
)
func (w *responseWriter) reset(writer http.ResponseWriter) {
w.ResponseWriter = writer
w.status = 0
w.written = false
w.beforeFuncs = w.beforeFuncs[:0]
}
func (w *responseWriter) setStatus(code int) {
@ -32,6 +38,7 @@ func (w *responseWriter) setStatus(code int) {
}
func (w *responseWriter) WriteHeader(code int) {
w.callBefore()
w.status = code
w.written = true
w.ResponseWriter.WriteHeader(code)
@ -44,3 +51,13 @@ func (w *responseWriter) Status() int {
func (w *responseWriter) Written() bool {
return w.written
}
func (w *responseWriter) Before(before BeforeFunc) {
w.beforeFuncs = append(w.beforeFuncs, before)
}
func (w *responseWriter) callBefore() {
for i := len(w.beforeFuncs) - 1; i >= 0; i-- {
w.beforeFuncs[i](w)
}
}