ceremonyclient/pebble/internal_test.go

// 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
}