Initial commit

responsewriterintercept.go

@@ -0,0 +1,34 @@
+package potency
+
+import (
+	"bytes"
+	"net/http"
+)
+
+type responseWriterIntercept struct {
+	dest       http.ResponseWriter
+	buf        bytes.Buffer
+	statusCode int
+}
+
+func newResponseWriterIntercept(dest http.ResponseWriter) *responseWriterIntercept {
+	return &responseWriterIntercept{
+		dest:       dest,
+		buf:        bytes.Buffer{},
+		statusCode: http.StatusOK,
+	}
+}
+
+func (rwi *responseWriterIntercept) Header() http.Header {
+	return rwi.dest.Header()
+}
+
+func (rwi *responseWriterIntercept) Write(data []byte) (int, error) {
+	rwi.buf.Write(data)
+	return rwi.dest.Write(data)
+}
+
+func (rwi *responseWriterIntercept) WriteHeader(statusCode int) {
+	rwi.statusCode = statusCode
+	rwi.dest.WriteHeader(statusCode)
+}