/*
* Copyright (c) [2016] [ <ether.camp> ]
* This file is part of the ethereumJ library.
*
* The ethereumJ library is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* The ethereumJ library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with the ethereumJ library. If not, see <http://www.gnu.org/licenses/>.
*/
package org.ethereum.datasource;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* Clue class between Source and BatchSource
*
* Created by Anton Nashatyrev on 29.11.2016.
*/
public class BatchSourceWriter<Key, Value> extends AbstractChainedSource<Key, Value, Key, Value> {
Map<Key, Value> buf = new HashMap<>();
public BatchSourceWriter(BatchSource<Key, Value> src) {
super(src);
}
private BatchSource<Key, Value> getBatchSource() {
return (BatchSource<Key, Value>) getSource();
}
@Override
public synchronized void delete(Key key) {
buf.put(key, null);
}
@Override
public synchronized void put(Key key, Value val) {
buf.put(key, val);
}
@Override
public Value get(Key key) {
return getSource().get(key);
}
@Override
public synchronized boolean flushImpl() {
if (!buf.isEmpty()) {
getBatchSource().updateBatch(buf);
buf.clear();
return true;
} else {
return false;
}
}
}