ceremonyclient/pebble/internal_test.go

89 lines
2.1 KiB
Go
Raw Normal View History

2024-01-03 07:31:42 +00:00
// Copyright 2019 The LevelDB-Go and Pebble Authors. All rights reserved. Use
// of this source code is governed by a BSD-style license that can be found in
// the LICENSE file.
package pebble
import "github.com/cockroachdb/pebble/internal/base"
// internalIterAdapter adapts the new internalIterator interface which returns
// the key and value from positioning methods (Seek*, First, Last, Next, Prev)
// to the old interface which returned a boolean corresponding to Valid. Only
// used by test code.
type internalIterAdapter struct {
internalIterator
key *InternalKey
val []byte
err error
}
func newInternalIterAdapter(iter internalIterator) *internalIterAdapter {
return &internalIterAdapter{
internalIterator: iter,
}
}
func (i *internalIterAdapter) update(key *InternalKey, val LazyValue) bool {
i.key = key
if v, _, err := val.Value(nil); err != nil {
i.key = nil
i.val = nil
i.err = err
} else {
i.val = v
}
return i.key != nil
}
func (i *internalIterAdapter) String() string {
return "internal-iter-adapter"
}
func (i *internalIterAdapter) SeekGE(key []byte, flags base.SeekGEFlags) bool {
return i.update(i.internalIterator.SeekGE(key, flags))
}
func (i *internalIterAdapter) SeekPrefixGE(prefix, key []byte, flags base.SeekGEFlags) bool {
return i.update(i.internalIterator.SeekPrefixGE(prefix, key, flags))
}
func (i *internalIterAdapter) SeekLT(key []byte, flags base.SeekLTFlags) bool {
return i.update(i.internalIterator.SeekLT(key, flags))
}
func (i *internalIterAdapter) First() bool {
return i.update(i.internalIterator.First())
}
func (i *internalIterAdapter) Last() bool {
return i.update(i.internalIterator.Last())
}
func (i *internalIterAdapter) Next() bool {
return i.update(i.internalIterator.Next())
}
func (i *internalIterAdapter) Prev() bool {
return i.update(i.internalIterator.Prev())
}
func (i *internalIterAdapter) Key() *InternalKey {
return i.key
}
func (i *internalIterAdapter) Value() []byte {
return i.val
}
func (i *internalIterAdapter) Valid() bool {
return i.key != nil
}
func (i *internalIterAdapter) Error() error {
err := i.internalIterator.Error()
if err != nil {
return err
}
return i.err
}