Enable StatusString for all readers (#50)

internal/proto/reader.go

@@ -55,28 +55,13 @@ func ParseErrorReply(line []byte) error {
 
 //------------------------------------------------------------------------------
 
-type readerOpt struct {
-	// useStatusStringType uses `StatusString` type instead of string for RespStatus.
-	useStatusStringType bool
-}
-
-func ReaderOptUseStatusStringType(o *readerOpt) {
-	o.useStatusStringType = true
-}
-
 type Reader struct {
-	rd  *bufio.Reader
-	opt readerOpt
+	rd *bufio.Reader
 }
 
-func NewReader(rd io.Reader, opts ...func(o *readerOpt)) *Reader {
-	opt := readerOpt{}
-	for _, applyOpt := range opts {
-		applyOpt(&opt)
-	}
+func NewReader(rd io.Reader) *Reader {
 	return &Reader{
-		rd:  bufio.NewReader(rd),
-		opt: opt,
+		rd: bufio.NewReader(rd),
 	}
 }
 
@@ -177,10 +162,7 @@ func (r *Reader) ReadReply() (interface{}, error) {
 
 	switch line[0] {
 	case RespStatus:
-		if r.opt.useStatusStringType {
-			return StatusString(line[1:]), nil
-		}
-		return string(line[1:]), nil
+		return StatusString(line[1:]), nil
 	case RespInt:
 		return util.ParseInt(line[1:], 10, 64)
 	case RespFloat:

internal/proto/writer_test.go

@@ -105,7 +105,7 @@ func TestWriteStatus(t *testing.T) {
 	inputStatusBytes := []byte("+status\r\n")
 
 	// Read it.
-	reader := proto.NewReader(bytes.NewReader(inputStatusBytes), proto.ReaderOptUseStatusStringType)
+	reader := proto.NewReader(bytes.NewReader(inputStatusBytes))
 	readStatus, err := reader.ReadReply()
 	if err != nil {
 		t.Errorf("Failed to ReadReply: %v", err)

proto.go

@@ -15,7 +15,7 @@ const RespArray = proto.RespArray
 const RespInt = proto.RespInt
 
 func NewReader(rd io.Reader) *Reader {
-	return proto.NewReader(rd, proto.ReaderOptUseStatusStringType)
+	return proto.NewReader(rd)
 }
 
 func NewWriter(wr *bytes.Buffer) *Writer {

pubsub.go

@@ -366,6 +366,10 @@ func (p *Pong) String() string {
 
 func (c *PubSub) newMessage(reply interface{}) (interface{}, error) {
 	switch reply := reply.(type) {
+	case proto.StatusString:
+		return &Pong{
+			Payload: string(reply),
+		}, nil
 	case string:
 		return &Pong{
 			Payload: reply,

redis_test.go

@@ -12,6 +12,7 @@ import (
 	. "github.com/bsm/gomega"
 
 	"github.com/redis/go-redis/v9"
+	"github.com/redis/go-redis/v9/internal/proto"
 )
 
 type redisHookError struct{}
@@ -192,7 +193,7 @@ var _ = Describe("Client", func() {
 		Expect(client.Echo(ctx, "hello").Err()).NotTo(HaveOccurred())
 
 		Expect(cmd.Err()).NotTo(HaveOccurred())
-		Expect(cmd.Val()).To(Equal("PONG"))
+		Expect(cmd.Val()).To(Equal(proto.StatusString("PONG")))
 	})
 
 	It("should retry command on network error", func() {