Add first version of retry connection in Go

Author: entropidelic

core/connection.go

@@ -0,0 +1,41 @@
+package connection
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/cenkalti/backoff/v4"
+)
+
+type PermanentError struct {
+	Inner error
+}
+
+func (e PermanentError) Error() string { return e.Inner.Error() }
+func (e PermanentError) Unwrap() error {
+	return e.Inner
+}
+func (e PermanentError) Is(err error) bool {
+	_, ok := err.(PermanentError)
+	return ok
+}
+
+func Retry(functionToRetry func() (interface{}, error), minDelay uint64, factor float64, maxTries uint64) (interface{}, error) {
+	f := func() (interface{}, error) {
+		val, err := functionToRetry()
+		if perm, ok := err.(PermanentError); err != nil && ok {
+			fmt.Println("ENTRE")
+			return nil, backoff.Permanent(perm.Inner)
+		}
+		return val, err
+	}
+
+	randomOption := backoff.WithRandomizationFactor(0)
+
+	initialRetryOption := backoff.WithInitialInterval(time.Millisecond * time.Duration(minDelay))
+	multiplierOption := backoff.WithMultiplier(factor)
+	expBackoff := backoff.NewExponentialBackOff(randomOption, multiplierOption, initialRetryOption)
+	maxRetriesBackoff := backoff.WithMaxRetries(expBackoff, maxTries)
+
+	return backoff.RetryWithData(f, maxRetriesBackoff)
+}

core/connection_test.go

@@ -0,0 +1,28 @@
+package connection_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/yetanotherco/aligned_layer/core"
+)
+
+func DummyFunction(x uint64) (uint64, error) {
+	fmt.Println("Doing some work...")
+	if x == 42 {
+		return 0, connection.PermanentError{Inner: fmt.Errorf("Permanent error!")}
+	} else if x < 42 {
+		return 0, fmt.Errorf("Transient error!")
+	}
+	return x, nil
+}
+
+func TestRetry(t *testing.T) {
+	function := func() (interface{}, error) { return DummyFunction(42) }
+	data, err := connection.Retry(function, 1000, 2, 3)
+	if err != nil {
+		t.Errorf("Retry error!: %s", err)
+	} else {
+		fmt.Printf("DATA: %d\n", data)
+	}
+}

go.mod

@@ -40,6 +40,7 @@ require (
 	github.com/bits-and-blooms/bitset v1.10.0 // indirect
 	github.com/blang/semver/v4 v4.0.0 // indirect
 	github.com/btcsuite/btcd/btcec/v2 v2.3.2 // indirect
+	github.com/cenkalti/backoff/v4 v4.3.0 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/consensys/bavard v0.1.13 // indirect
 	github.com/cpuguy83/go-md2man/v2 v2.0.3 // indirect

go.sum

@@ -56,6 +56,8 @@ github.com/btcsuite/btcd/chaincfg/chainhash v1.0.1 h1:q0rUy8C/TYNBQS1+CGKw68tLOF
 github.com/btcsuite/btcd/chaincfg/chainhash v1.0.1/go.mod h1:7SFka0XMvUgj3hfZtydOrQY2mwhPclbT2snogU7SQQc=
 github.com/cenkalti/backoff/v4 v4.2.1 h1:y4OZtCnogmCPw98Zjyt5a6+QwPLGkiQsYW5oUqylYbM=
 github.com/cenkalti/backoff/v4 v4.2.1/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
+github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
+github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
 github.com/cespare/cp v0.1.0 h1:SE+dxFebS7Iik5LK0tsi1k9ZCxEaFX4AjQmoyA+1dJk=
 github.com/cespare/cp v0.1.0/go.mod h1:SOGHArjBr4JWaSDEVpWpo/hNg6RoKrls6Oh40hiwW+s=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=

operator/sp1/sp1_test.go

@@ -8,7 +8,6 @@ import (
 )
 
 const ProofFilePath = "../../scripts/test_files/sp1/sp1_fibonacci.proof"
-
 const ElfFilePath = "../../scripts/test_files/sp1/sp1_fibonacci.elf"
 
 func TestFibonacciSp1ProofVerifies(t *testing.T) {