Add rand archive url

stellar · Jan 16, 2024 · d7f4881 · d7f4881
1 parent 3aa64b9
commit d7f4881
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/internal/utils/main.go b/internal/utils/main.go
@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"math/big"
+	"math/rand"
 	"time"
 
 	"github.com/spf13/pflag"
@@ -538,7 +539,15 @@ var futureArchiveURLs = []string{
 }
 
 func CreateHistoryArchiveClient(archiveURLS []string) (historyarchive.ArchiveInterface, error) {
-	return historyarchive.NewArchivePool(archiveURLS, historyarchive.ConnectOptions{UserAgent: "stellar-etl/1.0.0"})
+	var archiveInterface historyarchive.ArchiveInterface
+
+	archivePool, err := historyarchive.NewArchivePool(archiveURLS, historyarchive.ConnectOptions{UserAgent: "stellar-etl/1.0.0"})
+	if err != nil {
+		return archiveInterface, err
+	}
+
+	// Use random archive pool
+	return archivePool[int32(rand.Intn(len(archivePool)))], nil
 }
 
 // GetLatestLedgerSequence returns the latest ledger sequence