package com.thinkaurelius.titan.diskstorage;
import com.thinkaurelius.titan.diskstorage.util.ByteBufferUtil;
import com.thinkaurelius.titan.diskstorage.util.RecordIterator;
import com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration;
import com.thinkaurelius.titan.graphdb.database.serialize.DataOutput;
import com.thinkaurelius.titan.graphdb.database.serialize.Serializer;
import com.thinkaurelius.titan.graphdb.database.serialize.kryo.KryoSerializer;
import com.thinkaurelius.titan.testutil.RandomGenerator;
import org.apache.commons.configuration.Configuration;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.junit.Assert;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Arrays;
public class KeyValueStoreUtil {
private static final Logger log = LoggerFactory.getLogger(KeyValueStoreUtil.class);
public static Serializer serial = new KryoSerializer();
public static final long idOffset = 1000;
public static final StaticBuffer MIN_KEY = ByteBufferUtil.getLongBuffer(0);
public static final StaticBuffer MAX_KEY = ByteBufferUtil.getLongBuffer(-1);
public static void setUTF8Serializer() {
Configuration conf = new PropertiesConfiguration();
conf.setProperty(GraphDatabaseConfiguration.STRING_COMPACT_SERIALIZE,true);
KeyValueStoreUtil.serial = new KryoSerializer(conf);
}
public static void setDefaultSerializer() {
KeyValueStoreUtil.serial = new KryoSerializer();
}
public static String[] generateData(int numKeys) {
String[] ret = new String[numKeys];
for (int i = 0; i < numKeys; i++) {
ret[i] = RandomGenerator.randomString();
}
return ret;
}
public static String[][] generateData(int numKeys, int numColumns) {
String[][] ret = new String[numKeys][numColumns];
for (int i = 0; i < numKeys; i++) {
for (int j = 0; j < numColumns; j++) {
ret[i][j] = RandomGenerator.randomString();
}
}
return ret;
}
public static void print(String[] data) {
log.debug(Arrays.toString(data));
}
public static void print(String[][] data) {
for (int i = 0; i < data.length; i++) print(data[i]);
}
public static StaticBuffer getBuffer(int no) {
return ByteBufferUtil.getLongBuffer(no + idOffset);
}
public static int getID(StaticBuffer b) {
long res = b.getLong(0) - idOffset;
Assert.assertTrue(res >= 0 && res < Integer.MAX_VALUE);
return (int) res;
}
public static StaticBuffer getBuffer(String s) {
DataOutput out = serial.getDataOutput(50, true);
out.writeObjectNotNull(s);
return out.getStaticBuffer();
}
public static String getString(ReadBuffer b) {
return serial.readObjectNotNull(b, String.class);
}
public static String getString(StaticBuffer b) {
return serial.readObjectNotNull(b.asReadBuffer(), String.class);
}
public static int count(RecordIterator<?> iterator) throws StorageException {
int count = 0;
while (iterator.hasNext()) {
iterator.next();
count++;
}
return count;
}
}