TrueCloudLab/certificates
17
0
Fork 2
Code Issues 1 Pull requests 1 Releases Activity Actions

wip

Browse source
This commit is contained in:
max furman 2019-09-19 13:17:23 -07:00
parent 81093c3080
commit fe7973c060
14 changed files with 19 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

9
logging/responselogger.go
View file

@ -2,6 +2,7 @@ package logging
import (
"bufio"
"context"
"net"
"net/http"
)
@ -30,7 +31,7 @@ func NewResponseLogger(w http.ResponseWriter) ResponseLogger {
func wrapLogger(w http.ResponseWriter) (rw ResponseLogger) {
rw = &rwDefault{w, 200, 0, nil}
if c, ok := w.(http.CloseNotifier); ok {
if c, ok := w.(context.Context); ok {
rw = &rwCloseNotifier{rw, c}
}
if f, ok := w.(http.Flusher); ok {
@ -90,11 +91,11 @@ func (r *rwDefault) WithFields(fields map[string]interface{}) {
type rwCloseNotifier struct {
ResponseLogger
c http.CloseNotifier
ctx context.Context
}
func (r *rwCloseNotifier) CloseNotify() <-chan bool {
return r.CloseNotify()
func (r *rwCloseNotifier) Done() <-chan struct{} {
return r.Done()
}
type rwFlusher struct {