add randomization on list of dmsg servers for connect

skycoin · Feb 1, 2024 · b6a6764 · b6a6764
1 parent ee0160c
commit b6a6764
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/dmsg/client.go b/pkg/dmsg/client.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"math/rand"
 	"net"
 	"sync"
 	"time"
@@ -176,7 +177,10 @@ func (ce *Client) Serve(ctx context.Context) {
 			}
 		} else {
 			entries, err = ce.discoverServers(cancellabelCtx, false)
-
+			// randomize dmsg servers list
+			rand.Shuffle(len(entries), func(i, j int) {
+				entries[i], entries[j] = entries[j], entries[i]
+			})
 			if err != nil {
 				ce.log.WithError(err).Warn("Failed to discover dmsg servers.")
 				if err == context.Canceled || err == context.DeadlineExceeded {