Add an option to store a timestamp with RecordingResponseWriter - for req. metrics

Author: Peter Mogensen

http/handlers/accesslog/loghandler.go

@@ -116,6 +116,8 @@ func (h *logHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 		recorder := rrwriter.MakeRecorder(w)
 
 		t := time.Now()
+		recorder.SetTimeStamp(t)
+
 		pbuf := h.bufpool.Get().(*buffer)
 		logbuf := pbuf[:0]
 
@@ -145,6 +147,7 @@ func (h *logHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 	} else {
 		if h.af != nil {
 			recorder := rrwriter.MakeRecorder(w)
+			//recorder.SetTimeStamp(time.Now())
 			h.handler.ServeHTTP(recorder, req)
 			h.af(recorder)
 		} else {

http/rrwriter/responselogger.go

@@ -14,6 +14,7 @@ import (
 	"io"
 	"net"
 	"net/http"
+	"time"
 )
 
 // MakeRecorder returns a RecordingResponseWriter which wraps around the
@@ -41,6 +42,8 @@ type RecordingResponseWriter interface {
 	http.Flusher
 	Status() int
 	Size() int
+	GetTimeStamp() time.Time
+	SetTimeStamp(time.Time)
 }
 
 // responseRecorder is wrapper of http.ResponseWriter that keeps track of its HTTP
@@ -49,12 +52,15 @@ type responseRecorder struct {
 	w      http.ResponseWriter
 	status int
 	size   int
+	ts     time.Time
 }
 
+// Header implements http.ResponseWriter
 func (l *responseRecorder) Header() http.Header {
 	return l.w.Header()
 }
 
+// Write implements http.ResponseWriter
 func (l *responseRecorder) Write(b []byte) (int, error) {
 	if l.status == 0 {
 		// The status will be StatusOK if WriteHeader has not been called yet
@@ -65,11 +71,30 @@ func (l *responseRecorder) Write(b []byte) (int, error) {
 	return size, err
 }
 
+// WriteHeader implements http.ResponseWriter
 func (l *responseRecorder) WriteHeader(s int) {
 	l.w.WriteHeader(s)
 	l.status = s
 }
 
+// GetTimeStamp returns the timestamp
+// This can be used to store a starttime for a request and return it
+// after the request is done.
+// It is not go-routine safe.
+func (l *responseRecorder) GetTimeStamp() (tout time.Time) {
+	tout = l.ts
+	return
+}
+
+// SetTimeStamp sets a timestamp.
+// This can be used to store a starttime for a request and return it
+// after the request is done.
+// It is not go-routine safe.
+func (l *responseRecorder) SetTimeStamp(tin time.Time) {
+	l.ts = tin
+}
+
+
 // Status returns the http status of the written request.
 func (l *responseRecorder) Status() int {
 	return l.status