test_httpcontrol.go

package main

import (
	"fmt"
	"github.com/mwhooker/go.httpcontrol"
	"net"
	"net/http"
	"time"
)

func awsRetry(req *http.Request, res *http.Response, err error) bool {
	retry := false

	if err == nil && res != nil {
		retry = false
	}
	if neterr, ok := err.(net.Error); ok {
		if neterr.Temporary() {
			retry = true
		}
	}
	if res != nil {
		if 500 > res.StatusCode && res.StatusCode >= 400 {
			retry = true
		}
	}
	fmt.Println("retry", req.URL, retry)
	return retry
}

func main() {

	var c = &http.Client{
		Transport: &httpcontrol.Transport{
			ShouldRetry:    awsRetry,
			RequestTimeout: time.Second * 2,
			DialTimeout:    time.Second * 2,
			MaxTries:       3,
		},
	}
	fmt.Println(c.Get("http://httpbin.org/delay/0"))
	resp, err := c.Get("http://httpbin.org/status/418")
	if err == nil {
		fmt.Println(resp, err)
	}
}