loopdb/test: add database dump and restore tools

pull/205/head
Joost Jager 4 years ago
parent bd6e3f405a
commit 044c1c12dd
No known key found for this signature in database
GPG Key ID: A61B9D4C393C59C7

@ -0,0 +1,147 @@
package loopdb
import (
"encoding/hex"
"errors"
"fmt"
"strings"
"github.com/coreos/bbolt"
)
// DumpDB dumps go code describing the contents of the database to stdout. This
// function is only intended for use during development. Therefore also the
// linter unused warnings are suppressed.
//
// Example output:
//
// map[string]interface{}{
// Hex("1234"): map[string]interface{}{
// "human-readable": Hex("102030"),
// Hex("1111"): Hex("5783492373"),
// },
// }
func DumpDB(tx *bbolt.Tx) error { // nolint: unused
return tx.ForEach(func(k []byte, bucket *bbolt.Bucket) error {
key := toString(k)
fmt.Printf("%v: ", key)
err := dumpBucket(bucket)
if err != nil {
return err
}
fmt.Printf(",\n")
return nil
})
}
func dumpBucket(bucket *bbolt.Bucket) error { // nolint: unused
fmt.Printf("map[string]interface{} {\n")
err := bucket.ForEach(func(k, v []byte) error {
key := toString(k)
fmt.Printf("%v: ", key)
subBucket := bucket.Bucket(k)
if subBucket != nil {
err := dumpBucket(subBucket)
if err != nil {
return err
}
} else {
fmt.Print(toHex(v))
}
fmt.Printf(",\n")
return nil
})
if err != nil {
return err
}
fmt.Printf("}")
return nil
}
// RestoreDB primes the database with the given data set.
func RestoreDB(tx *bbolt.Tx, data map[string]interface{}) error {
for k, v := range data {
key := []byte(k)
value := v.(map[string]interface{})
subBucket, err := tx.CreateBucket(key)
if err != nil {
return fmt.Errorf("create bucket %v: %v",
string(key), err)
}
if err := restoreDB(subBucket, value); err != nil {
return err
}
}
return nil
}
func restoreDB(bucket *bbolt.Bucket, data map[string]interface{}) error {
for k, v := range data {
key := []byte(k)
switch value := v.(type) {
// Key contains value.
case string:
err := bucket.Put(key, []byte(value))
if err != nil {
return err
}
// Key contains a sub-bucket.
case map[string]interface{}:
subBucket, err := bucket.CreateBucket(key)
if err != nil {
return err
}
if err := restoreDB(subBucket, value); err != nil {
return err
}
default:
return errors.New("invalid type")
}
}
return nil
}
func toHex(v []byte) string { // nolint: unused
if len(v) == 0 {
return "nil"
}
return "Hex(\"" + hex.EncodeToString(v) + "\")"
}
func toString(v []byte) string { // nolint: unused
readableChars := "abcdefghijklmnopqrstuvwxyz0123456789-"
for _, c := range v {
if !strings.Contains(readableChars, string(c)) {
return toHex(v)
}
}
return "\"" + string(v) + "\""
}
// Hex is a test helper function to convert readable hex arrays to raw byte
// strings.
func Hex(value string) string {
b, err := hex.DecodeString(value)
if err != nil {
panic(err)
}
return string(b)
}
Loading…
Cancel
Save