68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package accesslog
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/containous/traefik/middlewares"
|
|
)
|
|
|
|
var (
|
|
_ middlewares.Stateful = &captureResponseWriter{}
|
|
)
|
|
|
|
// captureResponseWriter is a wrapper of type http.ResponseWriter
|
|
// that tracks request status and size
|
|
type captureResponseWriter struct {
|
|
rw http.ResponseWriter
|
|
status int
|
|
size int64
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Header() http.Header {
|
|
return crw.rw.Header()
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Write(b []byte) (int, error) {
|
|
if crw.status == 0 {
|
|
crw.status = http.StatusOK
|
|
}
|
|
size, err := crw.rw.Write(b)
|
|
crw.size += int64(size)
|
|
return size, err
|
|
}
|
|
|
|
func (crw *captureResponseWriter) WriteHeader(s int) {
|
|
crw.rw.WriteHeader(s)
|
|
crw.status = s
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Flush() {
|
|
if f, ok := crw.rw.(http.Flusher); ok {
|
|
f.Flush()
|
|
}
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
|
if h, ok := crw.rw.(http.Hijacker); ok {
|
|
return h.Hijack()
|
|
}
|
|
return nil, nil, fmt.Errorf("not a hijacker: %T", crw.rw)
|
|
}
|
|
|
|
func (crw *captureResponseWriter) CloseNotify() <-chan bool {
|
|
if c, ok := crw.rw.(http.CloseNotifier); ok {
|
|
return c.CloseNotify()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Status() int {
|
|
return crw.status
|
|
}
|
|
|
|
func (crw *captureResponseWriter) Size() int64 {
|
|
return crw.size
|
|
}
|