plugeth/memory_database.go

package main

import (
  "fmt"
)

/*
 * This is a test memory database. Do not use for any production it does not get persisted
 */
type MemDatabase struct {
  db      map[string][]byte
}

func NewMemDatabase() (*MemDatabase, error) {
  db := &MemDatabase{db: make(map[string][]byte)}

  return db, nil
}

func (db *MemDatabase) Put(key []byte, value []byte) {
  db.db[string(key)] = value
}

func (db *MemDatabase) Get(key []byte) ([]byte, error) {
  return db.db[string(key)], nil
}

func (db *MemDatabase) Print() {
  for key, val := range db.db {
    fmt.Printf("%x(%d):", key, len(key))
    decoded := DecodeNode(val)
    PrintSlice(decoded)
  }
}