Merge pull request #540 from redis/fix-cluster-move-ipv6

fix: ipv6 handling in MOVED and ASK messages of cluster mode
Signed-off-by: Rueian <[email protected]>

message.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"io"
+	"net"
 	"strconv"
 	"strings"
 	"time"
@@ -55,19 +56,28 @@ func (r *ValkeyError) IsNil() bool {
 // IsMoved checks if it is a valkey MOVED message and returns moved address.
 func (r *ValkeyError) IsMoved() (addr string, ok bool) {
 	if ok = strings.HasPrefix(r.string, "MOVED"); ok {
-		addr = strings.Split(r.string, " ")[2]
+		addr = fixIPv6HostPort(strings.Split(r.string, " ")[2])
 	}
 	return
 }
 
 // IsAsk checks if it is a valkey ASK message and returns ask address.
 func (r *ValkeyError) IsAsk() (addr string, ok bool) {
 	if ok = strings.HasPrefix(r.string, "ASK"); ok {
-		addr = strings.Split(r.string, " ")[2]
+		addr = fixIPv6HostPort(strings.Split(r.string, " ")[2])
 	}
 	return
 }
 
+func fixIPv6HostPort(addr string) string {
+	if strings.IndexByte(addr, '.') < 0 && len(addr) > 0 && addr[0] != '[' { // skip ipv4 and enclosed ipv6
+		if i := strings.LastIndexByte(addr, ':'); i >= 0 {
+			return net.JoinHostPort(addr[:i], addr[i+1:])
+		}
+	}
+	return addr
+}
+
 // IsTryAgain checks if it is a valkey TRYAGAIN message and returns ask address.
 func (r *ValkeyError) IsTryAgain() bool {
 	return strings.HasPrefix(r.string, "TRYAGAIN")

message_test.go

@@ -58,6 +58,38 @@ func TestIsValkeyErr(t *testing.T) {
 	}
 }
 
+func TestValkeyErrorIsMoved(t *testing.T) {
+	for _, c := range []struct {
+		err  string
+		addr string
+	}{
+		{err: "MOVED 1 127.0.0.1:1", addr: "127.0.0.1:1"},
+		{err: "MOVED 1 [::1]:1", addr: "[::1]:1"},
+		{err: "MOVED 1 ::1:1", addr: "[::1]:1"},
+	} {
+		e := ValkeyError{typ: '-', string: c.err}
+		if addr, ok := e.IsMoved(); !ok || addr != c.addr {
+			t.Fail()
+		}
+	}
+}
+
+func TestValkeyErrorIsAsk(t *testing.T) {
+	for _, c := range []struct {
+		err  string
+		addr string
+	}{
+		{err: "ASK 1 127.0.0.1:1", addr: "127.0.0.1:1"},
+		{err: "ASK 1 [::1]:1", addr: "[::1]:1"},
+		{err: "ASK 1 ::1:1", addr: "[::1]:1"},
+	} {
+		e := ValkeyError{typ: '-', string: c.err}
+		if addr, ok := e.IsAsk(); !ok || addr != c.addr {
+			t.Fail()
+		}
+	}
+}
+
 func TestIsValkeyBusyGroup(t *testing.T) {
 	err := errors.New("other")
 	if IsValkeyBusyGroup(err) {