keel/net/http/roundtripware/dump.go

package roundtripware

import (
	"fmt"
	"net/http"

	"go.uber.org/zap"
)

// DumpRequest returns a RoundTripper which prints out the request object
func DumpRequest() RoundTripware {
	return func(l *zap.Logger, next Handler) Handler {
		return func(req *http.Request) (*http.Response, error) {
			if req.Header != nil && req.Header.Get("Content-Type") != "" {
				var body string
				if req.Body, body = readBodyPretty(req.Header.Get("Content-Type"), req.Body); body != "" {
					fmt.Printf("Request %s:\n%s\n", req.URL, body)
				}
			}
			return next(req)
		}
	}
}

// DumpResponse returns a RoundTripper which prints out the response object
func DumpResponse() RoundTripware {
	return func(l *zap.Logger, next Handler) Handler {
		return func(req *http.Request) (*http.Response, error) {
			resp, err := next(req)
			if resp.Header != nil && resp.Header.Get("Content-Type") != "" {
				var body string
				if resp.Body, body = readBodyPretty(resp.Header.Get("Content-Type"), resp.Body); body != "" {
					fmt.Printf("Response %s:\n%s\n", req.URL, body)
				}
			}
			return resp, err
		}
	}
}