AccessExample.java [plain text]
package collections.access;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.util.Iterator;
import java.util.Map;
import java.util.SortedMap;
import com.sleepycat.bind.ByteArrayBinding;
import com.sleepycat.collections.StoredSortedMap;
import com.sleepycat.collections.TransactionRunner;
import com.sleepycat.collections.TransactionWorker;
import com.sleepycat.db.Database;
import com.sleepycat.db.DatabaseConfig;
import com.sleepycat.db.DatabaseException;
import com.sleepycat.db.DatabaseType;
import com.sleepycat.db.Environment;
import com.sleepycat.db.EnvironmentConfig;
public class AccessExample
implements Runnable {
private static boolean create = true;
private static final int EXIT_SUCCESS = 0;
private static final int EXIT_FAILURE = 1;
public static void usage() {
System.out.println("usage: java " + AccessExample.class.getName() +
" [-r] [database]\n");
System.exit(EXIT_FAILURE);
}
public static void main(String[] argv) {
boolean removeExistingDatabase = false;
String databaseName = "access.db";
for (int i = 0; i < argv.length; i++) {
if (argv[i].equals("-r")) {
removeExistingDatabase = true;
} else if (argv[i].equals("-?")) {
usage();
} else if (argv[i].startsWith("-")) {
usage();
} else {
if ((argv.length - i) != 1)
usage();
databaseName = argv[i];
break;
}
}
try {
EnvironmentConfig envConfig = new EnvironmentConfig();
envConfig.setTransactional(true);
envConfig.setInitializeCache(true);
envConfig.setInitializeLocking(true);
if (create) {
envConfig.setAllowCreate(true);
}
Environment env = new Environment(new File("."), envConfig);
if (removeExistingDatabase) {
env.removeDatabase(null, databaseName, null);
}
AccessExample app = new AccessExample(env, databaseName);
app.run();
} catch (DatabaseException e) {
e.printStackTrace();
System.exit(1);
} catch (FileNotFoundException e) {
e.printStackTrace();
System.exit(1);
} catch (Exception e) {
e.printStackTrace();
System.exit(1);
}
System.exit(0);
}
private Database db;
private SortedMap map;
private Environment env;
public AccessExample(Environment env, String databaseName)
throws Exception {
this.env = env;
ByteArrayBinding keyBinding = new ByteArrayBinding();
ByteArrayBinding dataBinding = new ByteArrayBinding();
DatabaseConfig dbConfig = new DatabaseConfig();
if (create) {
dbConfig.setAllowCreate(true);
dbConfig.setType(DatabaseType.BTREE);
}
this.db = env.openDatabase(null, databaseName, null, dbConfig);
this.map = new StoredSortedMap(db, keyBinding, dataBinding, true);
}
public void run() {
final InputStreamReader reader = new InputStreamReader(System.in);
for (; ; ) {
final String line = askForLine(reader, System.out, "input> ");
if (line == null) {
break;
}
final String reversed =
(new StringBuffer(line)).reverse().toString();
log("adding: \"" +
line + "\" : \"" +
reversed + "\"");
TransactionRunner tr = new TransactionRunner(env);
try {
tr.run(
new TransactionWorker() {
public void doWork() {
if (!map.containsKey(line.getBytes()))
map.put(line.getBytes(),
reversed.getBytes());
else
System.out.println("Key " + line +
" already exists.");
}
});
} catch (com.sleepycat.db.DatabaseException e) {
System.err.println("AccessExample: " + e.toString());
System.exit(1);
} catch (java.lang.Exception e) {
System.err.println("AccessExample: " + e.toString());
System.exit(1);
}
}
System.out.println("");
Iterator iter = map.entrySet().iterator();
System.out.println("Reading data");
while (iter.hasNext()) {
Map.Entry entry = (Map.Entry) iter.next();
log("found \"" +
new String((byte[]) entry.getKey()) +
"\" key with data \"" +
new String((byte[]) entry.getValue()) + "\"");
}
}
String askForLine(InputStreamReader reader, PrintStream out,
String prompt) {
String result = "";
while (result != null && result.length() == 0) {
out.print(prompt);
out.flush();
result = getLine(reader);
}
return result;
}
String getLine(InputStreamReader reader) {
StringBuffer b = new StringBuffer();
int c;
try {
while ((c = reader.read()) != -1 && c != '\n') {
if (c != '\r') {
b.append((char) c);
}
}
} catch (IOException ioe) {
c = -1;
}
if (c == -1 && b.length() == 0) {
return null;
} else {
return b.toString();
}
}
private void log(String s) {
System.out.println(s);
System.out.flush();
}
}