80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"strings"
|
|
)
|
|
|
|
type transport struct {
|
|
name string
|
|
transport http.RoundTripper
|
|
}
|
|
|
|
func (t *transport) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
if IsDebugOrHigher() {
|
|
reqData, err := httputil.DumpRequestOut(req, true)
|
|
if err == nil {
|
|
log.Printf("[DEBUG] "+logReqMsg, t.name, prettyPrintJsonLines(reqData))
|
|
} else {
|
|
log.Printf("[ERROR] %s API Request error: %#v", t.name, err)
|
|
}
|
|
}
|
|
|
|
resp, err := t.transport.RoundTrip(req)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
|
|
if IsDebugOrHigher() {
|
|
respData, err := httputil.DumpResponse(resp, true)
|
|
if err == nil {
|
|
log.Printf("[DEBUG] "+logRespMsg, t.name, prettyPrintJsonLines(respData))
|
|
} else {
|
|
log.Printf("[ERROR] %s API Response error: %#v", t.name, err)
|
|
}
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// NewTransport creates a wrapper around a *http.RoundTripper,
|
|
// designed to be used for the `Transport` field of http.Client.
|
|
//
|
|
// This logs each pair of HTTP request/response that it handles.
|
|
// The logging is done via Go standard library `log` package.
|
|
//
|
|
// Deprecated: This will log the content of every http request/response
|
|
// at `[DEBUG]` level, without any filtering. Any sensitive information
|
|
// will appear as-is in your logs. Please use NewSubsystemLoggingHTTPTransport instead.
|
|
func NewTransport(name string, t http.RoundTripper) *transport {
|
|
return &transport{name, t}
|
|
}
|
|
|
|
// prettyPrintJsonLines iterates through a []byte line-by-line,
|
|
// transforming any lines that are complete json into pretty-printed json.
|
|
func prettyPrintJsonLines(b []byte) string {
|
|
parts := strings.Split(string(b), "\n")
|
|
for i, p := range parts {
|
|
if b := []byte(p); json.Valid(b) {
|
|
var out bytes.Buffer
|
|
_ = json.Indent(&out, b, "", " ") // already checked for validity
|
|
parts[i] = out.String()
|
|
}
|
|
}
|
|
return strings.Join(parts, "\n")
|
|
}
|
|
|
|
const logReqMsg = `%s API Request Details:
|
|
---[ REQUEST ]---------------------------------------
|
|
%s
|
|
-----------------------------------------------------`
|
|
|
|
const logRespMsg = `%s API Response Details:
|
|
---[ RESPONSE ]--------------------------------------
|
|
%s
|
|
-----------------------------------------------------`
|