ceremonyclient/pebble/iterator_example_test.go

125 lines
2.6 KiB
Go
Raw Permalink Normal View History

2024-01-03 07:31:42 +00:00
// Copyright 2021 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_test
import (
"fmt"
"log"
"github.com/cockroachdb/pebble"
"github.com/cockroachdb/pebble/vfs"
)
func ExampleIterator() {
db, err := pebble.Open("", &pebble.Options{FS: vfs.NewMem()})
if err != nil {
log.Fatal(err)
}
keys := []string{"hello", "world", "hello world"}
for _, key := range keys {
if err := db.Set([]byte(key), nil, pebble.Sync); err != nil {
log.Fatal(err)
}
}
iter, _ := db.NewIter(nil)
for iter.First(); iter.Valid(); iter.Next() {
fmt.Printf("%s\n", iter.Key())
}
if err := iter.Close(); err != nil {
log.Fatal(err)
}
if err := db.Close(); err != nil {
log.Fatal(err)
}
// Output:
// hello
// hello world
// world
}
func ExampleIterator_prefixIteration() {
db, err := pebble.Open("", &pebble.Options{FS: vfs.NewMem()})
if err != nil {
log.Fatal(err)
}
keyUpperBound := func(b []byte) []byte {
end := make([]byte, len(b))
copy(end, b)
for i := len(end) - 1; i >= 0; i-- {
end[i] = end[i] + 1
if end[i] != 0 {
return end[:i+1]
}
}
return nil // no upper-bound
}
prefixIterOptions := func(prefix []byte) *pebble.IterOptions {
return &pebble.IterOptions{
LowerBound: prefix,
UpperBound: keyUpperBound(prefix),
}
}
keys := []string{"hello", "world", "hello world"}
for _, key := range keys {
if err := db.Set([]byte(key), nil, pebble.Sync); err != nil {
log.Fatal(err)
}
}
iter, _ := db.NewIter(prefixIterOptions([]byte("hello")))
for iter.First(); iter.Valid(); iter.Next() {
fmt.Printf("%s\n", iter.Key())
}
if err := iter.Close(); err != nil {
log.Fatal(err)
}
if err := db.Close(); err != nil {
log.Fatal(err)
}
// Output:
// hello
// hello world
}
func ExampleIterator_SeekGE() {
db, err := pebble.Open("", &pebble.Options{FS: vfs.NewMem()})
if err != nil {
log.Fatal(err)
}
keys := []string{"hello", "world", "hello world"}
for _, key := range keys {
if err := db.Set([]byte(key), nil, pebble.Sync); err != nil {
log.Fatal(err)
}
}
iter, _ := db.NewIter(nil)
if iter.SeekGE([]byte("a")); iter.Valid() {
fmt.Printf("%s\n", iter.Key())
}
if iter.SeekGE([]byte("hello w")); iter.Valid() {
fmt.Printf("%s\n", iter.Key())
}
if iter.SeekGE([]byte("w")); iter.Valid() {
fmt.Printf("%s\n", iter.Key())
}
if err := iter.Close(); err != nil {
log.Fatal(err)
}
if err := db.Close(); err != nil {
log.Fatal(err)
}
// Output:
// hello
// hello world
// world
}