forked from mirror/client_golang
Fix a race condition in the http instrumentation.
computeApproximateRequestSize is run in a goroutine, but the handlerFunc that runs in parallel may modify the URL, which is also needed by computeApproximateRequestSize. So get the URL length beforehand. Change-Id: Idb84735845afe7be4ef79b3d642d5764f6d26a7c
This commit is contained in:
parent
6fbc8ef5c2
commit
f4be228ba0
|
@ -140,7 +140,11 @@ func InstrumentHandlerFuncWithOpts(opts SummaryOpts, handlerFunc func(http.Respo
|
||||||
|
|
||||||
delegate := &responseWriterDelegator{ResponseWriter: w}
|
delegate := &responseWriterDelegator{ResponseWriter: w}
|
||||||
out := make(chan int)
|
out := make(chan int)
|
||||||
go computeApproximateRequestSize(r, out)
|
urlLen := 0
|
||||||
|
if r.URL != nil {
|
||||||
|
urlLen = len(r.URL.String())
|
||||||
|
}
|
||||||
|
go computeApproximateRequestSize(r, out, urlLen)
|
||||||
handlerFunc(delegate, r)
|
handlerFunc(delegate, r)
|
||||||
|
|
||||||
elapsed := float64(time.Since(now)) / float64(time.Microsecond)
|
elapsed := float64(time.Since(now)) / float64(time.Microsecond)
|
||||||
|
@ -154,11 +158,8 @@ func InstrumentHandlerFuncWithOpts(opts SummaryOpts, handlerFunc func(http.Respo
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func computeApproximateRequestSize(r *http.Request, out chan int) {
|
func computeApproximateRequestSize(r *http.Request, out chan int, s int) {
|
||||||
s := len(r.Method)
|
s += len(r.Method)
|
||||||
if r.URL != nil {
|
|
||||||
s += len(r.URL.String())
|
|
||||||
}
|
|
||||||
s += len(r.Proto)
|
s += len(r.Proto)
|
||||||
for name, values := range r.Header {
|
for name, values := range r.Header {
|
||||||
s += len(name)
|
s += len(name)
|
||||||
|
|
Loading…
Reference in New Issue