public void testKmin() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
Map<String, Integer> result = strKeyIndex.kmin(1);
assertNotNull(result);
assertEquals(new Integer(789), result.get("18 year"));
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
Map<Integer, String> result_2 = intKeyIndex.kmin(1);
assertNotNull(result_2);
assertEquals("abc", result_2.get(-12));
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
Map<FastString, byte[]> result_3 = fastStringIndex.kmin(1);
assertNotNull(result_3);
assertEquals("11789", new String(result_3.get(new FastString("中科院计算所"))));
fastStringIndex.close();
}
public void testBulkGet() throws Exception {
File strKeyIndexFile = prepareDataStringKey();
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(strKeyIndexFile);
String result =
strKeyIndex
.bulkGet(Arrays.asList(new String[] {"IBM", "18 year", "haha", "你好世界"}))
.toString();
assertEquals(result, "{18 year=789, IBM=456, 你好世界=7777}");
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
result = intKeyIndex.bulkGet(Arrays.asList(new Integer[] {-12, 345, 133, 99, 0})).toString();
assertEquals(result, "{-12=abc, 0=zzz, 133=def, 345=中国}");
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
int size =
fastStringIndex
.bulkGet(
Arrays.asList(new FastString[] {new FastString("18 year"), new FastString("IBM")}))
.size();
assertEquals(size, 2);
fastStringIndex.close();
}
public void testRange() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
Map<String, Integer> result = strKeyIndex.range("18 year", "aab", 1);
assertNotNull(result);
assertEquals(789, (int) result.get("18 year"));
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
Map<Integer, String> result_2 = intKeyIndex.range(0, 345, 1);
assertNotNull(result_2);
assertEquals("zzz", result_2.get(0));
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
Map<FastString, byte[]> result_3 =
fastStringIndex.range(new FastString("18 year"), new FastString("aab"), 1);
assertNotNull(result_3);
assertEquals("你好", new String(result_3.get(new FastString("18 year"))));
fastStringIndex.close();
}
public void testAfter() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
Map<String, Integer> result = strKeyIndex.after("18 year", 2);
assertNotNull(result);
assertEquals(456, (int) result.get("IBM"));
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
Map<Integer, String> result_2 = intKeyIndex.after(0, 2);
assertNotNull(result_2);
assertEquals("def", result_2.get(133));
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
Map<FastString, byte[]> result_3 = fastStringIndex.after(new FastString("18 year"), 2);
assertNotNull(result_3);
assertEquals("456", new String(result_3.get(new FastString("IBM"))));
fastStringIndex.close();
}
public void testKmax() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
Map<String, Integer> result = strKeyIndex.kmax(1);
assertNotNull(result);
assertEquals(7777, (int) result.get("你好世界"));
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
Map<Integer, String> result_2 = intKeyIndex.kmax(1);
assertNotNull(result_2);
assertEquals("中国", result_2.get(345));
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
Map<FastString, byte[]> result_3 = fastStringIndex.kmax(1);
assertNotNull(result_3);
assertEquals("123", new String(result_3.get(new FastString("aab"))));
fastStringIndex.close();
}
private File prepareDataFastStringKey() throws Exception {
File indexFile = new File("tmp/fast_str_key_index");
indexFile.delete();
DiskTreap<FastString, byte[]> treap = new DiskTreap<FastString, byte[]>(indexFile);
treap.put(new FastString("aab"), "123".getBytes());
treap.put(new FastString("IBM"), "456".getBytes());
treap.put(new FastString("中科院计算所"), "11789".getBytes());
treap.put(new FastString("18 year"), "你好".getBytes());
treap.close();
return indexFile;
}
private File prepareDataStringKey() throws Exception {
File indexFile = new File("tmp/str_key_index");
indexFile.delete();
DiskTreap<String, Integer> treap = new DiskTreap<String, Integer>(indexFile);
treap.put("aab", 123);
treap.put("IBM", 456);
treap.put("18 year", 789);
treap.put("你好世界", 7777);
treap.close();
return indexFile;
}
public void testBlukPut() throws Exception {
File indexFile = new File("tmp/bulkput");
indexFile.delete();
DiskTreap<String, Integer> index = new DiskTreap<String, Integer>(indexFile);
Map<String, Integer> map = new HashMap<String, Integer>();
map.put("foo", 1);
map.put("footbar", 2);
map.put("azb", 3);
map.put("dddddd", 4);
map.put("aojiao", 5);
Map<String, Integer> map2 = new HashMap<String, Integer>();
map2.put("azb", 13);
map2.put("footbar", 14);
map2.put("sjy", 140);
index.bulkPut(map);
index.bulkPut(map2);
index.close();
index = new DiskTreap<String, Integer>(indexFile);
assertEquals(index.prefix("a", 10).toString(), "{aojiao=5, azb=13}");
assertEquals(index.prefix("f", 10).toString(), "{foo=1, footbar=14}");
assertEquals(index.get("dddddd").toString(), "4");
assertEquals(index.length(), 6);
index.close();
}
@Override
public void execute(
DiskTreap<FastString, byte[]> diskTreap, String command, byte[] body, BufferedOutputStream os)
throws Exception {
Integer len = diskTreap.length();
os.write(("" + len + "\r\n").getBytes());
os.write(("END\r\n").getBytes());
}
public void testBulkPrefix() throws Exception {
File indexFile = new File("tmp/bulk_prefix");
indexFile.delete();
DiskTreap<FastString, String> index = new DiskTreap<FastString, String>(indexFile);
for (int i = 0; i < 1000; i++) {
for (int j = 0; j < 10; j++) {
index.put(new FastString("user" + i + ":" + j), "hello");
}
}
List<String> prefixList = new ArrayList<String>();
prefixList.add("user1:");
prefixList.add("user7:");
prefixList.add("user8:");
// long t1 = System.currentTimeMillis();
// System.out.println(index.bulkPrefix(prefixList, 2, null, false));
// index.prefix(new FastString("user1:"),30);
// System.out.println(System.currentTimeMillis()-t1);
assertEquals(
"{user8:9=hello, user8:8=hello, user7:9=hello, user7:8=hello, user1:9=hello, user1:8=hello}",
index.bulkPrefix(prefixList, 2, null, false).toString());
// System.out.println(index.bulkPrefix(prefixList, 3, null, true));
assertEquals(
"{user1:0=hello, user1:1=hello, user1:2=hello, user7:0=hello, user7:1=hello, user7:2=hello, user8:0=hello, user8:1=hello, user8:2=hello}",
index.bulkPrefix(prefixList, 3, null, true).toString());
// System.out.println(index.bulkPrefix(prefixList, 3,new FastString("user8:9"),false));
assertEquals(
"{user8:9=hello, user8:8=hello, user8:7=hello, user7:9=hello, user7:8=hello, user7:7=hello, user1:9=hello, user1:8=hello, user1:7=hello}",
index.bulkPrefix(prefixList, 3, new FastString("user8:9"), false).toString());
index.close();
}
private File prepareDataFastIntKey() throws Exception {
File indexFile = new File("tmp/int_key_index");
indexFile.delete();
DiskTreap<Integer, String> treap = new DiskTreap<Integer, String>(indexFile);
treap.put(-12, "abc");
treap.put(345, "中国");
treap.put(0, "zzz");
treap.put(133, "def");
treap.close();
return indexFile;
}
public void testRemove() throws Exception {
new File("tmp/remove_test").delete();
new File("tmp/remove_faststr_test").delete();
DiskTreap<String, Integer> treap = new DiskTreap<String, Integer>(new File("tmp/remove_test"));
for (int i = 0; i < 100; i++) {
treap.put("thing" + i, i);
}
assertEquals(100, treap.length());
for (int i = 0; i < 9; i++) {
treap.removePrefix("thing" + i);
}
assertEquals(11, treap.length());
assertEquals(
"{thing99=99, thing98=98, thing97=97, thing96=96, thing95=95, thing94=94, thing93=93, thing92=92, thing91=91, thing90=90, thing9=9}",
treap.kmax(100).toString());
treap.delete("thing9");
assertEquals(10, treap.length());
assertEquals(
"{thing99=99, thing98=98, thing97=97, thing96=96, thing95=95, thing94=94, thing93=93, thing92=92, thing91=91, thing90=90}",
treap.kmax(100).toString());
assertNull(treap.get("thing77"));
treap.close();
treap = new DiskTreap<String, Integer>(new File("tmp/remove_test"));
for (int i = 0; i < 100; i++) {
treap.put("thing" + i, i);
}
assertEquals(100, treap.length());
DiskTreap<FastString, Integer> treap2 =
new DiskTreap<FastString, Integer>(new File("tmp/remove_faststr_test"));
for (int i = 0; i < 100; i++) {
treap2.put(new FastString("thing" + i), i);
}
assertEquals(100, treap2.length());
for (int i = 0; i < 9; i++) {
treap2.removePrefix(new FastString("thing" + i));
}
assertEquals(11, treap2.length());
assertEquals(
"{thing99=99, thing98=98, thing97=97, thing96=96, thing95=95, thing94=94, thing93=93, thing92=92, thing91=91, thing90=90, thing9=9}",
treap2.kmax(100).toString());
treap2.delete(new FastString("thing9"));
assertEquals(10, treap2.length());
assertEquals(
"{thing99=99, thing98=98, thing97=97, thing96=96, thing95=95, thing94=94, thing93=93, thing92=92, thing91=91, thing90=90}",
treap2.kmax(100).toString());
assertNull(treap2.get(new FastString("thing77")));
treap2.close();
}
public void testLength() throws Exception {
File strKeyIndexFile = prepareDataStringKey();
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(strKeyIndexFile);
assertEquals(strKeyIndex.length(), 4);
strKeyIndex.close();
}
public void testRank() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
assertEquals(2, strKeyIndex.rank("IBM", true));
assertEquals(3, strKeyIndex.rank("IBM", false));
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
assertEquals(2, intKeyIndex.rank(0, true));
assertEquals(3, intKeyIndex.rank(0, false));
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
assertEquals(2, fastStringIndex.rank(new FastString("18 year"), true));
assertEquals(3, fastStringIndex.rank(new FastString("18 year"), false));
assertEquals(-1, fastStringIndex.rank(new FastString("xxxx"), true));
fastStringIndex.close();
}
public void testPrefix() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
Map<String, Integer> result = strKeyIndex.prefix("aa", 1);
assertNotNull(result);
assertEquals(1, result.size());
assertEquals(123, (int) result.get("aab"));
result = strKeyIndex.prefix("I", 1);
assertNotNull(result);
assertEquals(1, result.size());
assertEquals(456, (int) result.get("IBM"));
result = strKeyIndex.prefix("18", 1);
assertNotNull(result);
assertEquals(1, result.size());
assertEquals(789, (int) result.get("18 year"));
result = strKeyIndex.prefix("你好世", 1);
assertNotNull(result);
assertEquals(1, result.size());
assertEquals(7777, (int) result.get("你好世界"));
strKeyIndex.close();
// =======
/*DiskTreap<Integer, String> intKeyIndex = new DiskTreap<Integer, String>(prepareDataFastIntKey());
Map<Integer, String> result_2 = intKeyIndex.prefix(-12, 1) ;
assertNotNull(result_2) ;
assertEquals(1, result_2.size()) ;
System.out.println(result_2.size()) ;
assertEquals("abc", result_2.get(-12)) ;
result_2 = intKeyIndex.prefix(3, 1) ;
assertNotNull(result_2) ;
assertEquals(1, result_2.size()) ;
System.out.println(result_2.size()) ;
assertEquals("中国", result_2.get(345)) ;
result_2 = intKeyIndex.prefix(0, 1) ;
assertNotNull(result_2) ;
assertEquals(1, result_2.size()) ;
System.out.println(result_2.size()) ;
assertEquals("zzz", result_2.get(0)) ;
result_2 = intKeyIndex.prefix(13, 1) ;
assertNotNull(result_2) ;
assertEquals(1, result_2.size()) ;
System.out.println(result_2.size()) ;
assertEquals("def", result_2.get(133)) ;
intKeyIndex.close();*/
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
Map<FastString, byte[]> result_3 = fastStringIndex.prefix(new FastString("aa"), 1);
assertNotNull(result_3);
assertEquals(1, result_3.size());
assertEquals("123", new String(result_3.get(new FastString("aab"))));
result_3 = fastStringIndex.prefix(new FastString("IB"), 1);
assertNotNull(result_3);
assertEquals(1, result_3.size());
assertEquals("456", new String(result_3.get(new FastString("IBM"))));
result_3 = fastStringIndex.prefix(new FastString("中科"), 1);
assertNotNull(result_3);
assertEquals(1, result_3.size());
assertEquals("11789", new String(result_3.get(new FastString("中科院计算所"))));
result_3 = fastStringIndex.prefix(new FastString("18"), 1);
assertNotNull(result_3);
assertEquals(1, result_3.size());
assertEquals("你好", new String(result_3.get(new FastString("18 year"))));
for (int i = 10; i <= 100; i++) {
fastStringIndex.put(new FastString("" + i), (i + "").getBytes());
}
assertEquals(
"[100, 10]",
fastStringIndex
.prefix(new FastString("1"), 5, new FastString("100"), false)
.keySet()
.toString());
assertEquals(
"[10, 100, 11, 12, 13, 14, 15]",
fastStringIndex
.prefix(new FastString("1"), 7, new FastString("10"), true)
.keySet()
.toString());
fastStringIndex.close();
}
public void testKth() throws Exception {
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(prepareDataStringKey());
assertEquals("IBM", strKeyIndex.kth(2, true).getKey());
assertEquals("aab", strKeyIndex.kth(2, false).getKey());
strKeyIndex.close();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
assertEquals(new Integer(133), intKeyIndex.kth(3, true).getKey());
assertEquals(new Integer(0), intKeyIndex.kth(3, false).getKey());
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
assertEquals("IBM", fastStringIndex.kth(3, true).getKey().toString());
assertEquals("18 year", fastStringIndex.kth(3, false).getKey().toString());
assertNull(fastStringIndex.kth(-10, true));
assertNull(fastStringIndex.kth(100, false));
fastStringIndex.close();
}
public void testGet() throws Exception {
File strKeyIndexFile = prepareDataStringKey();
DiskTreap<String, Integer> strKeyIndex = new DiskTreap<String, Integer>(strKeyIndexFile);
assertEquals(strKeyIndex.get("aab"), new Integer(123));
assertEquals(strKeyIndex.get("IBM"), new Integer(456));
assertEquals(strKeyIndex.get("18 year"), new Integer(789));
assertEquals(strKeyIndex.get("你好世界"), new Integer(7777));
strKeyIndex.close();
strKeyIndexFile.delete();
// =======
DiskTreap<Integer, String> intKeyIndex =
new DiskTreap<Integer, String>(prepareDataFastIntKey());
assertEquals(intKeyIndex.get(-12), "abc");
assertEquals(intKeyIndex.get(0), "zzz");
assertEquals(intKeyIndex.get(133), "def");
assertEquals(intKeyIndex.get(345), "中国");
intKeyIndex.close();
// =======
DiskTreap<FastString, byte[]> fastStringIndex =
new DiskTreap<FastString, byte[]>(prepareDataFastStringKey());
assertEquals(new String(fastStringIndex.get(new FastString("aab"))), "123");
assertEquals(new String(fastStringIndex.get(new FastString("IBM"))), "456");
assertEquals(new String(fastStringIndex.get(new FastString("中科院计算所"))), "11789");
assertEquals(new String(fastStringIndex.get(new FastString("18 year"))), "你好");
fastStringIndex.close();
}