| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 package memory | 5 package memory |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "bytes" | 8 "bytes" |
| 9 "sync" | 9 "sync" |
| 10 | 10 |
| 11 "github.com/luci/gae/service/datastore" | |
| 12 "github.com/luci/gae/service/datastore/serialize" | 11 "github.com/luci/gae/service/datastore/serialize" |
| 13 "github.com/luci/gkvlite" | 12 "github.com/luci/gkvlite" |
| 14 ) | 13 ) |
| 15 | 14 |
| 16 type iterDefinition struct { | 15 type iterDefinition struct { |
| 17 // The collection to iterate over | 16 // The collection to iterate over |
| 18 c *memCollection | 17 c *memCollection |
| 19 | 18 |
| 20 // The prefix to always assert for every row. A nil prefix matches every
row. | 19 // The prefix to always assert for every row. A nil prefix matches every
row. |
| 21 prefix []byte | 20 prefix []byte |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 93 } | 92 } |
| 94 } | 93 } |
| 95 if stop { | 94 if stop { |
| 96 return nil | 95 return nil |
| 97 } | 96 } |
| 98 if restart { | 97 if restart { |
| 99 continue | 98 continue |
| 100 } | 99 } |
| 101 | 100 |
| 102 if err := cb(suffix); err != nil { | 101 if err := cb(suffix); err != nil { |
| 103 if err == datastore.Stop { | |
| 104 return nil | |
| 105 } | |
| 106 return err | 102 return err |
| 107 } | 103 } |
| 108 suffix = nil | 104 suffix = nil |
| 109 skip = -1 | 105 skip = -1 |
| 110 } | 106 } |
| 111 } | 107 } |
| 112 | 108 |
| 113 type cmd struct { | 109 type cmd struct { |
| 114 targ []byte | 110 targ []byte |
| 115 cb func(*gkvlite.Item) | 111 cb func(*gkvlite.Item) |
| (...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 211 t.ch <- &cmd{targ, func(i *gkvlite.Item) { | 207 t.ch <- &cmd{targ, func(i *gkvlite.Item) { |
| 212 defer close(waiter) | 208 defer close(waiter) |
| 213 | 209 |
| 214 if i == nil { | 210 if i == nil { |
| 215 t.stop() | 211 t.stop() |
| 216 } | 212 } |
| 217 cb(i) | 213 cb(i) |
| 218 }} | 214 }} |
| 219 <-waiter | 215 <-waiter |
| 220 } | 216 } |
| OLD | NEW |