fix: include pagination.key at reverse mode (#20939)
This commit is contained in:
@@ -330,11 +330,11 @@ func encodeCollKey[K, V any, C Collection[K, V]](coll C, key K) ([]byte, error)
|
||||
func getCollIter[K, V any, C Collection[K, V]](ctx context.Context, coll C, prefix, start []byte, reverse bool) (collections.Iterator[K, V], error) {
|
||||
// TODO: maybe can be simplified
|
||||
if reverse {
|
||||
var end []byte
|
||||
if prefix != nil {
|
||||
start = storetypes.PrefixEndBytes(append(prefix, start...))
|
||||
end = prefix
|
||||
}
|
||||
// if we are in reverse mode, we need to increase the start key
|
||||
// to include the start key in the iteration.
|
||||
start = storetypes.PrefixEndBytes(append(prefix, start...))
|
||||
end := prefix
|
||||
|
||||
return coll.IterateRaw(ctx, end, start, collections.OrderDescending)
|
||||
}
|
||||
var end []byte
|
||||
|
||||
@@ -83,6 +83,16 @@ func TestCollectionPagination(t *testing.T) {
|
||||
},
|
||||
expResults: createResults(299, 200),
|
||||
},
|
||||
"with key and reverse": {
|
||||
req: &PageRequest{
|
||||
Key: encodeKey(199),
|
||||
Reverse: true,
|
||||
},
|
||||
expResp: &PageResponse{
|
||||
NextKey: encodeKey(99),
|
||||
},
|
||||
expResults: createResults(199, 100),
|
||||
},
|
||||
"with offset and count total": {
|
||||
req: &PageRequest{
|
||||
Offset: 50,
|
||||
|
||||
Reference in New Issue
Block a user