88 lines
2.3 KiB
Java
88 lines
2.3 KiB
Java
package com.minres.scviewer.database.leveldb;
|
|
|
|
import static java.nio.charset.StandardCharsets.UTF_8;
|
|
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import java.util.Map.Entry;
|
|
|
|
import org.iq80.leveldb.Options;
|
|
import org.iq80.leveldb.impl.SeekingIterator;
|
|
import org.json.JSONObject;
|
|
|
|
import com.minres.scviewer.database.IWaveform;
|
|
import com.minres.scviewer.database.IWaveformDb;
|
|
import com.minres.scviewer.database.IWaveformDbLoader;
|
|
import com.minres.scviewer.database.RelationType;
|
|
|
|
public class LevelDBLoader implements IWaveformDbLoader {
|
|
|
|
static byte[] toByteArray(String value) {
|
|
return value.getBytes(UTF_8);
|
|
}
|
|
|
|
private TxDBWrapper levelDb;
|
|
private IWaveformDb db;
|
|
private Long maxTime=null;
|
|
private List<RelationType> usedRelationsList = new ArrayList<>();
|
|
|
|
@Override
|
|
public boolean load(IWaveformDb db, File inp) throws Exception {
|
|
try {
|
|
this.db=db;
|
|
levelDb = new TxDBWrapper(new Options(), inp);
|
|
JSONObject configVal = new JSONObject(levelDb.get("__config"));
|
|
if(!configVal.isEmpty())
|
|
levelDb.setTimeResolution(configVal.getLong("resolution"));
|
|
} catch(Exception e) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public Long getMaxTime() {
|
|
if(maxTime==null) {
|
|
SeekingIterator<String, String> it = levelDb.iterator();
|
|
it.seek("st~");
|
|
Entry<String, String> val = null;
|
|
while(it.hasNext()) {
|
|
Entry<String, String> v = it.next();
|
|
if(!v.getKey().startsWith("st~")) continue;
|
|
val=v;
|
|
}
|
|
if(val==null)
|
|
maxTime = 0L;
|
|
else {
|
|
String[] token = val.getKey().split("~");
|
|
maxTime = Long.parseLong(token[2], 16)*levelDb.getTimeResolution();
|
|
}
|
|
}
|
|
return maxTime;
|
|
}
|
|
|
|
@Override
|
|
public Collection<IWaveform> getAllWaves() {
|
|
List<IWaveform> streams=new ArrayList<IWaveform>();
|
|
SeekingIterator<String, String> it = levelDb.iterator();
|
|
it.seek("s~");
|
|
while(it.hasNext()) {
|
|
Entry<String, String> val = it.next();
|
|
if(!val.getKey().startsWith("s~")) break;
|
|
TxStream stream = new TxStream(levelDb, db, new JSONObject(val.getValue()));
|
|
stream.setRelationTypeList(usedRelationsList);
|
|
streams.add(stream);
|
|
}
|
|
return streams;
|
|
}
|
|
|
|
@Override
|
|
public Collection<RelationType> getAllRelationTypes() {
|
|
// return Collections.emptyList();
|
|
return usedRelationsList;
|
|
}
|
|
|
|
}
|