debug.go

package base

import (
	"fmt"
	"net/http"
	"net/http/httputil"
	"os"
)

func init() {
	defaultTransport := http.DefaultClient.Transport
	if defaultTransport == nil {
		defaultTransport = http.DefaultTransport
	}

	http.DefaultClient.Transport = loggingRoundTripper{defaultTransport}
}

type loggingRoundTripper struct {
	http.RoundTripper
}

func (tr loggingRoundTripper) RoundTrip(req *http.Request) (res *http.Response, err error) {
	trace, err := httputil.DumpRequestOut(req, false)
	if err != nil {
		fmt.Fprintf(os.Stderr, "DumpRequestOut() error: %v\n", err)
	} else {
		fmt.Fprintf(os.Stderr, "------------  HTTP REQUEST -----------\n%s", trace)
	}

	res, err = tr.RoundTripper.RoundTrip(req)
	if err != nil {
		fmt.Fprintf(os.Stderr, "RoundTrip() returned error: %v", err)
	}

	if res != nil {
		trace, err := httputil.DumpResponse(res, false)
		if err != nil {
			fmt.Fprintf(os.Stderr, "DumpResponse() error: %v\n", err)
		} else {
			fmt.Fprintf(os.Stderr, "------------  HTTP RESPONSE ----------\n%s", trace)
		}
	}

	return res, err
}