fix iterator across empty pages

This commit is contained in:
vkd 2016-09-08 16:07:49 +03:00
parent 49f197e6d1
commit 76c33da3ee
2 changed files with 45 additions and 17 deletions

View File

@ -46,6 +46,7 @@ func (it *ScanIterator) Next() bool {
return true return true
} }
for {
// Return if there is more data to fetch. // Return if there is more data to fetch.
if it.ScanCmd.cursor == 0 { if it.ScanCmd.cursor == 0 {
return false return false
@ -64,7 +65,11 @@ func (it *ScanIterator) Next() bool {
} }
it.pos = 1 it.pos = 1
return len(it.ScanCmd.page) > 0 if len(it.ScanCmd.page) > 0 {
return true
}
}
return false
} }
// Val returns the key/field at the current cursor position. // Val returns the key/field at the current cursor position.

View File

@ -21,6 +21,18 @@ var _ = Describe("ScanIterator", func() {
return err return err
} }
var extraSeed = func(n int, m int) error {
pipe := client.Pipeline()
for i := 1; i <= m; i++ {
pipe.Set(fmt.Sprintf("A%02d", i), "x", 0).Err()
}
for i := 1; i <= n; i++ {
pipe.Set(fmt.Sprintf("K%02d", i), "x", 0).Err()
}
_, err := pipe.Exec()
return err
}
var hashKey = "K_HASHTEST" var hashKey = "K_HASHTEST"
var hashSeed = func(n int) error { var hashSeed = func(n int) error {
pipe := client.Pipeline() pipe := client.Pipeline()
@ -110,4 +122,15 @@ var _ = Describe("ScanIterator", func() {
Expect(vals).To(HaveLen(13)) Expect(vals).To(HaveLen(13))
}) })
It("should scan with match across empty pages", func() {
Expect(extraSeed(2, 10)).NotTo(HaveOccurred())
var vals []string
iter := client.Scan(0, "K*", 1).Iterator()
for iter.Next() {
vals = append(vals, iter.Val())
}
Expect(iter.Err()).NotTo(HaveOccurred())
Expect(vals).To(HaveLen(2))
})
}) })