feat: setup backoff retry strategy

Signed-off-by: rare-magma <[email protected]>

Author: rare-magma

main.go

@@ -8,7 +8,7 @@ import (
 	"fmt"
 	"io"
 	"log"
-	"net"
+	"math"
 	"net/http"
 	"os"
 	"sync"
@@ -25,6 +25,52 @@ type Config struct {
 	GithubApiToken   string `json:"GithubApiToken"`
 }
 
+type retryableTransport struct {
+	transport             http.RoundTripper
+	TLSHandshakeTimeout   time.Duration
+	ResponseHeaderTimeout time.Duration
+}
+
+const retryCount = 3
+
+func shouldRetry(err error, resp *http.Response) bool {
+	if err != nil {
+		return true
+	}
+	switch resp.StatusCode {
+	case http.StatusInternalServerError, http.StatusBadGateway, http.StatusServiceUnavailable, http.StatusGatewayTimeout:
+		return true
+	default:
+		return false
+	}
+}
+
+func (t *retryableTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	var bodyBytes []byte
+	if req.Body != nil {
+		bodyBytes, _ = io.ReadAll(req.Body)
+		req.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
+	}
+	resp, err := t.transport.RoundTrip(req)
+	retries := 0
+	for shouldRetry(err, resp) && retries < retryCount {
+		backoff := time.Duration(math.Pow(2, float64(retries))) * time.Second
+		time.Sleep(backoff)
+		if resp.Body != nil {
+			io.Copy(io.Discard, resp.Body)
+			resp.Body.Close()
+		}
+		if req.Body != nil {
+			req.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
+		}
+		log.Printf("Previous request failed with %s", resp.Status)
+		log.Printf("Retry %d of request to: %s", retries+1, req.URL)
+		resp, err = t.transport.RoundTrip(req)
+		retries++
+	}
+	return resp, err
+}
+
 func main() {
 	confFilePath := "github_exporter.json"
 	confData, err := os.Open(confFilePath)
@@ -53,15 +99,14 @@ func main() {
 		log.Fatalln("GithubApiToken is required")
 	}
 
+	transport := &retryableTransport{
+		transport:             &http.Transport{},
+		TLSHandshakeTimeout:   30 * time.Second,
+		ResponseHeaderTimeout: 30 * time.Second,
+	}
 	client := &http.Client{
-		Timeout: 30 * time.Second,
-		Transport: &http.Transport{
-			DialContext: (&net.Dialer{
-				Timeout: 30 * time.Second,
-			}).DialContext,
-			TLSHandshakeTimeout:   30 * time.Second,
-			ResponseHeaderTimeout: 30 * time.Second,
-		},
+		Timeout:   30 * time.Second,
+		Transport: transport,
 	}
 	ghc := github.NewClient(client).WithAuthToken(config.GithubApiToken)
 	opts := &github.RepositoryListByAuthenticatedUserOptions{Type: "owner"}