try.go

package try

import (
	"errors"
	"fmt"
)

// MaxRetries is the maximum number of retries
var MaxRetries = 3

// Do keep trying the function until max retry limit or no return error
func Do(fn func(attempt int) error, maxRetries ...int) error {
	var (
		err        error
		attempt    = 1
		maxAttempt = MaxRetries
	)

	if len(maxRetries) > 0 {
		maxAttempt = maxRetries[0]
	}

	for {
		func() {
			defer func() {
				if r := recover(); r != nil {
					err = fmt.Errorf("panic: %v", r)
				}
			}()
			err = fn(attempt)
		}()
		if err == nil {
			break
		}

		attempt++
		if attempt > maxAttempt {
			if err != nil {
				return err
			}
			return ErrMaxRetriesReached
		}
	}
	return err
}

// IsMaxRetries is a function to check if the error has reached the maximum to try
func IsMaxRetries(err error) bool {
	return errors.Is(err, ErrMaxRetriesReached)
}