Summary: Fix a bug in RocksDBSample.java and minor code improvement in rocksjni.cc. Test Plan: make jni make jtest Reviewers: haobo, sdong Reviewed By: haobo CC: leveldb Differential Revision: https://reviews.facebook.net/D17325
80 lines
2.4 KiB
Java
80 lines
2.4 KiB
Java
// Copyright (c) 2014, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
import java.util.*;
|
|
import java.lang.*;
|
|
import org.rocksdb.*;
|
|
import java.io.IOException;
|
|
|
|
public class RocksDBSample {
|
|
static {
|
|
System.loadLibrary("rocksdbjni");
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
if (args.length < 1) {
|
|
System.out.println("usage: RocksDBSample db_path");
|
|
return;
|
|
}
|
|
String db_path = args[0];
|
|
|
|
System.out.println("RocksDBSample");
|
|
|
|
try {
|
|
RocksDB db = RocksDB.open(db_path);
|
|
db.put("hello".getBytes(), "world".getBytes());
|
|
byte[] value = db.get("hello".getBytes());
|
|
System.out.format("Get('hello') = %s\n",
|
|
new String(value));
|
|
|
|
for (int i = 1; i <= 9; ++i) {
|
|
for (int j = 1; j <= 9; ++j) {
|
|
db.put(String.format("%dx%d", i, j).getBytes(),
|
|
String.format("%d", i * j).getBytes());
|
|
}
|
|
}
|
|
|
|
for (int i = 1; i <= 9; ++i) {
|
|
for (int j = 1; j <= 9; ++j) {
|
|
System.out.format("%s ", new String(db.get(
|
|
String.format("%dx%d", i, j).getBytes())));
|
|
}
|
|
System.out.println("");
|
|
}
|
|
|
|
value = db.get("1x1".getBytes());
|
|
assert(value != null);
|
|
value = db.get("world".getBytes());
|
|
assert(value == null);
|
|
|
|
byte[] testKey = "asdf".getBytes();
|
|
byte[] testValue =
|
|
"asdfghjkl;'?><MNBVCXZQWERTYUIOP{+_)(*&^%$#@".getBytes();
|
|
db.put(testKey, testValue);
|
|
byte[] testResult = db.get(testKey);
|
|
assert(testResult != null);
|
|
assert(Arrays.equals(testValue, testResult));
|
|
assert(new String(testValue).equals(new String(testResult)));
|
|
|
|
byte[] insufficientArray = new byte[10];
|
|
byte[] enoughArray = new byte[50];
|
|
int len;
|
|
len = db.get(testKey, insufficientArray);
|
|
assert(len > insufficientArray.length);
|
|
len = db.get("asdfjkl;".getBytes(), enoughArray);
|
|
assert(len == RocksDB.NOT_FOUND);
|
|
len = db.get(testKey, enoughArray);
|
|
assert(len == testValue.length);
|
|
try {
|
|
db.close();
|
|
} catch (IOException e) {
|
|
System.err.println(e);
|
|
}
|
|
} catch (RocksDBException e) {
|
|
System.err.println(e);
|
|
}
|
|
}
|
|
}
|