diff --git a/.gitignore b/.gitignore index ebfe903..b975a7b 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ *.rdb testdata/*/ +.idea/ diff --git a/pipeline_test.go b/pipeline_test.go index fb82e4e..a00da7a 100644 --- a/pipeline_test.go +++ b/pipeline_test.go @@ -1,6 +1,8 @@ package redis_test import ( + "strconv" + "github.com/go-redis/redis/v7" . "github.com/onsi/ginkgo" @@ -67,6 +69,21 @@ var _ = Describe("pipelining", func() { Expect(err).NotTo(HaveOccurred()) Expect(cmds).To(HaveLen(1)) }) + + It("handles large pipelines", func() { + for callCount := 1; callCount < 16; callCount++ { + for i := 1; i <= callCount; i++ { + pipe.SetNX(strconv.Itoa(i)+"_key", strconv.Itoa(i)+"_value", 0) + } + + cmds, err := pipe.Exec() + Expect(err).NotTo(HaveOccurred()) + Expect(cmds).To(HaveLen(callCount)) + for _, cmd := range cmds { + Expect(cmd).To(BeAssignableToTypeOf(&redis.BoolCmd{})) + } + } + }) } Describe("Pipeline", func() { diff --git a/redis.go b/redis.go index 3d9dfed..ac0569f 100644 --- a/redis.go +++ b/redis.go @@ -473,11 +473,11 @@ func wrapMultiExec(cmds []Cmder) []Cmder { if len(cmds) == 0 { panic("not reached") } - cmds = append(cmds, make([]Cmder, 2)...) - copy(cmds[1:], cmds[:len(cmds)-2]) - cmds[0] = NewStatusCmd("multi") - cmds[len(cmds)-1] = NewSliceCmd("exec") - return cmds + cmdCopy := make([]Cmder, len(cmds)+2) + cmdCopy[0] = NewStatusCmd("multi") + copy(cmdCopy[1:], cmds) + cmdCopy[len(cmdCopy)-1] = NewSliceCmd("exec") + return cmdCopy } func txPipelineReadQueued(rd *proto.Reader, statusCmd *StatusCmd, cmds []Cmder) error {