feat: retry context

runtime/retry.go

@@ -1,12 +1,14 @@
 package runtime
 
 import (
+	"context"
 	"errors"
 	"sync/atomic"
 	"time"
 
 	retry "github.com/avast/retry-go/v4"
 	"github.com/hjwalt/runway/logger"
+	"github.com/hjwalt/runway/reflect"
 )
 
 // constructor
@@ -94,6 +96,8 @@ func (c *Retry) Do(fnToDo func(int64) error) error {
 	return err
 }
 
+// utils
+
 var ErrRetryStopped = errors.New("retry stopped")
 
 func AlwaysTry(err error) bool {
@@ -102,3 +106,11 @@ func AlwaysTry(err error) bool {
 	}
 	return true
 }
+
+func SetRetryCount(ctx context.Context, trycount int64) context.Context {
+	return context.WithValue(ctx, Context("RetryCount"), trycount)
+}
+
+func GetRetryCount(ctx context.Context) int64 {
+	return reflect.GetInt64(ctx.Value(Context("RetryCount")))
+}

runtime/retry_test.go

@@ -1,6 +1,7 @@
 package runtime
 
 import (
+	"context"
 	"errors"
 	"testing"
 	"time"
@@ -99,3 +100,15 @@ func TestRetryAbsorb(t *testing.T) {
 	assert.Equal(lastiteration, int64(100))
 	assert.NoError(err)
 }
+
+func TestRetryContext(t *testing.T) {
+	assert := assert.New(t)
+
+	ctx := context.Background()
+	zeroCount := GetRetryCount(ctx)
+	assert.Equal(int64(0), zeroCount)
+
+	withTryCount := SetRetryCount(ctx, 100)
+	retrieved := GetRetryCount(withTryCount)
+	assert.Equal(int64(100), retrieved)
+}