package org.ethereum.datasource.inmem;
import org.ethereum.datasource.DbSource;
import org.ethereum.util.ALock;
import org.ethereum.util.ByteArrayMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/**
* Created by Anton Nashatyrev on 12.10.2016.
*/
public class HashMapDBSimple<V> implements DbSource<V> {
protected final Map<byte[], V> storage;
public HashMapDBSimple() {
this(new ByteArrayMap<V>());
}
public HashMapDBSimple(ByteArrayMap<V> storage) {
this.storage = storage;
}
@Override
public void put(byte[] key, V val) {
if (val == null) {
delete(key);
} else {
storage.put(key, val);
}
}
@Override
public V get(byte[] key) {
return storage.get(key);
}
@Override
public void delete(byte[] key) {
storage.remove(key);
}
@Override
public boolean flush() {
return true;
}
@Override
public void setName(String name) {}
@Override
public String getName() {
return "in-memory";
}
@Override
public void init() {}
@Override
public boolean isAlive() {
return true;
}
@Override
public void close() {}
@Override
public Set<byte[]> keys() {
return getStorage().keySet();
}
@Override
public void updateBatch(Map<byte[], V> rows) {
for (Map.Entry<byte[], V> entry : rows.entrySet()) {
put(entry.getKey(), entry.getValue());
}
}
public Map<byte[], V> getStorage() {
return storage;
}
}