homelab/traefik
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add access log integration test

Browse source
This commit is contained in:
David Tootill 2016-04-20 11:54:57 -07:00
parent c8c0d208be
commit 2a76a717e6
5 changed files with 161 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
middlewares/logger.go
View file

@ -38,7 +38,7 @@ type frontendBackendLoggingHandler struct {
var (
reqidCounter uint64 // Request ID
infoRwMap = cmap.New() // Map of reqid to response writer
backend2FrontendMap map[string]string
backend2FrontendMap *map[string]string
)
// logInfoResponseWriter is a wrapper of type http.ResponseWriter
@ -65,7 +65,7 @@ func NewLogger(file string) *Logger {
// SetBackend2FrontendMap is called by server.go to set up frontend translation
func SetBackend2FrontendMap(newMap *map[string]string) {
backend2FrontendMap = *newMap
backend2FrontendMap = newMap
}
func (l *Logger) ServeHTTP(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
@ -93,7 +93,7 @@ func saveBackendNameForLogger(r *http.Request, backendName string) {
reqid := reqidHdr[0]
if infoRw, ok := infoRwMap.Get(reqid); ok {
infoRw.(*logInfoResponseWriter).SetBackend(backendName)
infoRw.(*logInfoResponseWriter).SetFrontend(backend2FrontendMap[backendName])
infoRw.(*logInfoResponseWriter).SetFrontend((*backend2FrontendMap)[backendName])
}
}
}