mirror of
https://github.com/arnaucube/go-merkletree-iden3.git
synced 2026-02-07 03:26:46 +01:00
Small fixes
This commit is contained in:
@@ -24,7 +24,7 @@ type LevelDbStorageTx struct {
|
||||
cache db.KvMap
|
||||
}
|
||||
|
||||
// NewLevelStorage returns a new LevelDbStorage
|
||||
// NewLevelDbStorage returns a new LevelDbStorage
|
||||
func NewLevelDbStorage(path string, errorIfMissing bool) (*LevelDbStorage, error) {
|
||||
o := &opt.Options{
|
||||
ErrorIfMissing: errorIfMissing,
|
||||
@@ -113,16 +113,16 @@ func (l *LevelDbStorage) Iterate(f func([]byte, []byte) (bool, error)) error {
|
||||
}
|
||||
|
||||
// Get retreives a value from a key in the interface db.Tx
|
||||
func (l *LevelDbStorageTx) Get(key []byte) ([]byte, error) {
|
||||
func (tx *LevelDbStorageTx) Get(key []byte) ([]byte, error) {
|
||||
var err error
|
||||
|
||||
fullkey := db.Concat(l.prefix, key)
|
||||
fullkey := db.Concat(tx.prefix, key)
|
||||
|
||||
if value, ok := l.cache.Get(fullkey); ok {
|
||||
if value, ok := tx.cache.Get(fullkey); ok {
|
||||
return value, nil
|
||||
}
|
||||
|
||||
value, err := l.ldb.Get(fullkey, nil)
|
||||
value, err := tx.ldb.Get(fullkey, nil)
|
||||
if err == errors.ErrNotFound {
|
||||
return nil, db.ErrNotFound
|
||||
}
|
||||
@@ -130,7 +130,7 @@ func (l *LevelDbStorageTx) Get(key []byte) ([]byte, error) {
|
||||
return value, err
|
||||
}
|
||||
|
||||
// Insert saves a key:value into the db.Storage
|
||||
// Put saves a key:value into the db.Storage
|
||||
func (tx *LevelDbStorageTx) Put(k, v []byte) {
|
||||
tx.cache.Put(db.Concat(tx.prefix, k[:]), v)
|
||||
}
|
||||
@@ -144,19 +144,19 @@ func (tx *LevelDbStorageTx) Add(atx db.Tx) {
|
||||
}
|
||||
|
||||
// Commit implements the method Commit of the interface db.Tx
|
||||
func (l *LevelDbStorageTx) Commit() error {
|
||||
func (tx *LevelDbStorageTx) Commit() error {
|
||||
var batch leveldb.Batch
|
||||
for _, v := range l.cache {
|
||||
for _, v := range tx.cache {
|
||||
batch.Put(v.K, v.V)
|
||||
}
|
||||
|
||||
l.cache = nil
|
||||
return l.ldb.Write(&batch, nil)
|
||||
tx.cache = nil
|
||||
return tx.ldb.Write(&batch, nil)
|
||||
}
|
||||
|
||||
// Close implements the method Close of the interface db.Tx
|
||||
func (l *LevelDbStorageTx) Close() {
|
||||
l.cache = nil
|
||||
func (tx *LevelDbStorageTx) Close() {
|
||||
tx.cache = nil
|
||||
}
|
||||
|
||||
// Close implements the method Close of the interface db.Storage
|
||||
|
||||
@@ -26,7 +26,7 @@ func NewMemoryStorage() *MemoryStorage {
|
||||
}
|
||||
|
||||
// Info implements the method Info of the interface db.Storage
|
||||
func (l *MemoryStorage) Info() string {
|
||||
func (m *MemoryStorage) Info() string {
|
||||
return "in-memory"
|
||||
}
|
||||
|
||||
@@ -41,21 +41,21 @@ func (m *MemoryStorage) NewTx() (db.Tx, error) {
|
||||
}
|
||||
|
||||
// Get retreives a value from a key in the db.Storage
|
||||
func (l *MemoryStorage) Get(key []byte) ([]byte, error) {
|
||||
if v, ok := l.kv.Get(db.Concat(l.prefix, key[:])); ok {
|
||||
func (m *MemoryStorage) Get(key []byte) ([]byte, error) {
|
||||
if v, ok := m.kv.Get(db.Concat(m.prefix, key[:])); ok {
|
||||
return v, nil
|
||||
}
|
||||
return nil, db.ErrNotFound
|
||||
}
|
||||
|
||||
// Iterate implements the method Iterate of the interface db.Storage
|
||||
func (l *MemoryStorage) Iterate(f func([]byte, []byte) (bool, error)) error {
|
||||
func (m *MemoryStorage) Iterate(f func([]byte, []byte) (bool, error)) error {
|
||||
kvs := make([]db.KV, 0)
|
||||
for _, v := range l.kv {
|
||||
if len(v.K) < len(l.prefix) || !bytes.Equal(v.K[:len(l.prefix)], l.prefix) {
|
||||
for _, v := range m.kv {
|
||||
if len(v.K) < len(m.prefix) || !bytes.Equal(v.K[:len(m.prefix)], m.prefix) {
|
||||
continue
|
||||
}
|
||||
localkey := v.K[len(l.prefix):]
|
||||
localkey := v.K[len(m.prefix):]
|
||||
kvs = append(kvs, db.KV{K: localkey, V: v.V})
|
||||
|
||||
}
|
||||
@@ -116,9 +116,9 @@ func (m *MemoryStorage) Close() {
|
||||
}
|
||||
|
||||
// List implements the method List of the interface db.Storage
|
||||
func (l *MemoryStorage) List(limit int) ([]db.KV, error) {
|
||||
func (m *MemoryStorage) List(limit int) ([]db.KV, error) {
|
||||
ret := []db.KV{}
|
||||
err := l.Iterate(func(key []byte, value []byte) (bool, error) {
|
||||
err := m.Iterate(func(key []byte, value []byte) (bool, error) {
|
||||
ret = append(ret, db.KV{K: db.Clone(key), V: db.Clone(value)})
|
||||
if len(ret) == limit {
|
||||
return false, nil
|
||||
|
||||
Reference in New Issue
Block a user