20 lines
436 B
Go
20 lines
436 B
Go
package hws
|
|
|
|
import "net/http"
|
|
|
|
// Wraps the http.ResponseWriter, adding a statusCode field
|
|
type wrappedWriter struct {
|
|
http.ResponseWriter
|
|
statusCode int
|
|
}
|
|
|
|
// Extends WriteHeader to the ResponseWriter to add the status code
|
|
func (w *wrappedWriter) WriteHeader(statusCode int) {
|
|
w.ResponseWriter.WriteHeader(statusCode)
|
|
w.statusCode = statusCode
|
|
}
|
|
|
|
func (w *wrappedWriter) Unwrap() http.ResponseWriter {
|
|
return w.ResponseWriter
|
|
}
|