Use node address instead of relying on loopback reported by redis

Author: dim

cluster.go

@@ -3,6 +3,7 @@ package redis
 import (
 	"fmt"
 	"math/rand"
+	"net"
 	"sync"
 	"sync/atomic"
 	"time"
@@ -244,7 +245,7 @@ type clusterState struct {
 	slots [][]*clusterNode
 }
 
-func newClusterState(nodes *clusterNodes, slots []ClusterSlot) (*clusterState, error) {
+func newClusterState(nodes *clusterNodes, slots []ClusterSlot, origin string) (*clusterState, error) {
 	c := clusterState{
 		nodes: nodes,
 		slots: make([][]*clusterNode, hashtag.SlotNumber),
@@ -253,7 +254,12 @@ func newClusterState(nodes *clusterNodes, slots []ClusterSlot) (*clusterState, e
 	for _, slot := range slots {
 		var nodes []*clusterNode
 		for _, slotNode := range slot.Nodes {
-			node, err := c.nodes.Get(slotNode.Addr)
+			addr := slotNode.Addr
+			if isLoopbackAddr(addr) && !isLoopbackAddr(origin) {
+				addr = origin
+			}
+
+			node, err := c.nodes.Get(addr)
 			if err != nil {
 				return nil, err
 			}
@@ -661,7 +667,7 @@ func (c *ClusterClient) reloadSlots() (*clusterState, error) {
 		return nil, err
 	}
 
-	return newClusterState(c.nodes, slots)
+	return newClusterState(c.nodes, slots, node.Client.opt.Addr)
 }
 
 // reaper closes idle connections to the cluster.
@@ -960,3 +966,17 @@ func (c *ClusterClient) txPipelineReadQueued(
 
 	return firstErr
 }
+
+func isLoopbackAddr(addr string) bool {
+	host, _, err := net.SplitHostPort(addr)
+	if err != nil {
+		return false
+	}
+
+	ip := net.ParseIP(host)
+	if ip == nil {
+		return false
+	}
+
+	return ip.IsLoopback()
+}