Enabled trace

auspost/client.go

@@ -26,8 +26,15 @@ func NewClient(apiKey string, baseUrl string) *Client {
 type Client struct {
 	restClient *resty.Client
 	baseUrl    string
+	trace      bool
 }
 
 func (c *Client) getUrl(uri string) string {
 	return fmt.Sprintf("%s/%s", c.baseUrl, uri)
 }
+
+func (c *Client) EnableTrace() *Client {
+	c.trace = true
+
+	return c
+}

auspost/postcode.go

@@ -2,6 +2,7 @@ package auspost
 
 import (
 	"errors"
+	"fmt"
 	"github.com/rhomber/auspost-postcode/auspost/model"
 	"strconv"
 )
@@ -13,7 +14,13 @@ const (
 )
 
 func (c *Client) PostcodeSearch(q string, state string, excludePostboxFlag bool) (model.PostcodeSearchResult, error) {
-	resp, err := c.restClient.R().
+	req := c.restClient.R()
+
+	if c.trace {
+		req.EnableTrace()
+	}
+
+	resp, err := req.
 		SetQueryParams(map[string]string{
 			"q":                  q,
 			"state":              state,
@@ -22,6 +29,33 @@ func (c *Client) PostcodeSearch(q string, state string, excludePostboxFlag bool)
 		SetResult(model.PostcodeSearchResult{}).
 		Get(c.getUrl(ausPostUriPostcodeSearch))
 
+	if c.trace {
+		fmt.Println("AusPost API Request")
+		fmt.Println("===================")
+		fmt.Println()
+		fmt.Println("Response Info:")
+		fmt.Println("Error      :", err)
+		fmt.Println("Status Code:", resp.StatusCode())
+		fmt.Println("Status     :", resp.Status())
+		fmt.Println("Time       :", resp.Time())
+		fmt.Println("Received At:", resp.ReceivedAt())
+		fmt.Println("Body       :\n", resp)
+		fmt.Println()
+
+		fmt.Println("Request Trace Info:")
+		ti := resp.Request.TraceInfo()
+		fmt.Println("DNSLookup    :", ti.DNSLookup)
+		fmt.Println("ConnTime     :", ti.ConnTime)
+		fmt.Println("TLSHandshake :", ti.TLSHandshake)
+		fmt.Println("ServerTime   :", ti.ServerTime)
+		fmt.Println("ResponseTime :", ti.ResponseTime)
+		fmt.Println("TotalTime    :", ti.TotalTime)
+		fmt.Println("IsConnReused :", ti.IsConnReused)
+		fmt.Println("IsConnWasIdle:", ti.IsConnWasIdle)
+		fmt.Println("ConnIdleTime :", ti.ConnIdleTime)
+		fmt.Println()
+	}
+
 	if err != nil {
 		return model.PostcodeSearchResult{}, err
 	}