Stefan/evcc-io
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve http request logging

Browse source
This commit is contained in:
andig 2021-05-15 11:26:52 +02:00
parent ac688cdab7
commit cf6d435b6e
1 changed files with 21 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

32
util/request/roundtrip.go
View file

@ -1,6 +1,7 @@
package request
import (
"bytes"
"net/http"
"net/http/httputil"
"strings"
@ -25,26 +26,35 @@ func NewTripper(log *util.Logger, transport http.RoundTripper) http.RoundTripper
return tripper
}
var bld = strings.Builder{}
func min(a, b int) int {
if a < b {
return a
}
return b
}
func (r *roundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
if body, err := httputil.DumpRequest(req, true); err == nil {
s := strings.TrimSpace(string(body))
if len(s) > max {
s = s[:max]
}
r.log.TRACE.Println(s)
bld.Reset()
if body, err := httputil.DumpRequestOut(req, true); err == nil {
bld.WriteString("\n")
bld.Write(bytes.TrimSpace(body[:min(max, len(body))]))
}
resp, err := r.transport.RoundTrip(req)
if resp != nil {
if body, err := httputil.DumpResponse(resp, true); err == nil {
s := strings.TrimSpace(string(body))
if len(s) > max {
s = s[:max]
}
r.log.TRACE.Println(s)
bld.WriteString("\n\n")
bld.Write(bytes.TrimSpace(body[:min(max, len(body))]))
}
}
if bld.Len() > 0 {
r.log.TRACE.Println(bld.String())
}
return resp, err
}