Update dependencies

This commit is contained in:
Zlatko Čalušić
2018-01-21 19:43:52 +01:00
parent 0a5606e954
commit 9ef84dcdea
27 changed files with 507 additions and 51 deletions

View File

@@ -2,22 +2,26 @@ package datacounter
import (
"bufio"
"fmt"
"net"
"net/http"
"sync/atomic"
"time"
)
// ResponseWriterCounter is counter for http.ResponseWriter
type ResponseWriterCounter struct {
http.ResponseWriter
count uint64
writer http.ResponseWriter
count uint64
started time.Time
writer http.ResponseWriter
}
// NewResponseWriterCounter function create new ResponseWriterCounter
func NewResponseWriterCounter(rw http.ResponseWriter) *ResponseWriterCounter {
return &ResponseWriterCounter{
writer: rw,
started: time.Now(),
}
}
@@ -32,6 +36,7 @@ func (counter *ResponseWriterCounter) Header() http.Header {
}
func (counter *ResponseWriterCounter) WriteHeader(statusCode int) {
counter.Header().Set("X-Runtime", fmt.Sprintf("%.6f", time.Since(counter.started).Seconds()))
counter.writer.WriteHeader(statusCode)
}
@@ -43,3 +48,7 @@ func (counter *ResponseWriterCounter) Hijack() (net.Conn, *bufio.ReadWriter, err
func (counter *ResponseWriterCounter) Count() uint64 {
return atomic.LoadUint64(&counter.count)
}
func (counter *ResponseWriterCounter) Started() time.Time {
return counter.started
}