ciaranarcher · July 27, 2014 06:53 · Jul 27, 2014
diff --git a/example.go b/example.go
@@ -0,0 +1,33 @@
+// Create our own MyResponseWriter to wrap a standard http.ResponseWriter
+// so we can store the status code.
+type MyResponseWriter struct {
+	status int 
+	http.ResponseWriter
+}
+
+func NewMyResponseWriter(res http.ResponseWriter) *MyResponseWriter {
+	// Default the status code to 200
+	return &MyResponseWriter{200, res}
+}
+
+// Give a way to get the status
+func (w MyResponseWriter) Status() int {
+	return w.status
+}
+
+// Satisfy the http.ResponseWriter interface
+func (w MyResponseWriter) Header() http.Header {
+	return w.ResponseWriter.Header()
+}
+
+func (w MyResponseWriter) Write(data []byte) (int, error) {
+	return w.ResponseWriter.Write(data)
+}
+
+func (w MyResponseWriter) WriteHeader(statusCode int) {
+	// Store the status code
+	w.status = statusCode
+
+	// Write the status code onward.
+	w.ResponseWriter.WriteHeader(statusCode)
+}