aboutsummaryrefslogtreecommitdiffstats
path: root/ethdb
diff options
context:
space:
mode:
Diffstat (limited to 'ethdb')
-rw-r--r--ethdb/memory_database.go16
1 files changed, 8 insertions, 8 deletions
diff --git a/ethdb/memory_database.go b/ethdb/memory_database.go
index 699bd0c9f..0dd93a279 100644
--- a/ethdb/memory_database.go
+++ b/ethdb/memory_database.go
@@ -37,6 +37,12 @@ func NewMemDatabase() (*MemDatabase, error) {
}, nil
}
+func NewMemDatabaseWithCap(size int) (*MemDatabase, error) {
+ return &MemDatabase{
+ db: make(map[string][]byte, size),
+ }, nil
+}
+
func (db *MemDatabase) Put(key []byte, value []byte) error {
db.lock.Lock()
defer db.lock.Unlock()
@@ -74,14 +80,6 @@ func (db *MemDatabase) Keys() [][]byte {
return keys
}
-/*
-func (db *MemDatabase) GetKeys() []*common.Key {
- data, _ := db.Get([]byte("KeyRing"))
-
- return []*common.Key{common.NewKeyFromBytes(data)}
-}
-*/
-
func (db *MemDatabase) Delete(key []byte) error {
db.lock.Lock()
defer db.lock.Unlock()
@@ -96,6 +94,8 @@ func (db *MemDatabase) NewBatch() Batch {
return &memBatch{db: db}
}
+func (db *MemDatabase) Len() int { return len(db.db) }
+
type kv struct{ k, v []byte }
type memBatch struct {