package com.thinkaurelius.titan.diskstorage.berkeleyje;
import com.google.common.base.Preconditions;
import com.sleepycat.je.Cursor;
import com.sleepycat.je.DatabaseException;
import com.sleepycat.je.Transaction;
import com.thinkaurelius.titan.diskstorage.PermanentStorageException;
import com.thinkaurelius.titan.diskstorage.StorageException;
import com.thinkaurelius.titan.diskstorage.common.AbstractStoreTransaction;
import com.thinkaurelius.titan.diskstorage.keycolumnvalue.ConsistencyLevel;
import com.thinkaurelius.titan.diskstorage.keycolumnvalue.StoreTxConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.List;
public class BerkeleyJETx extends AbstractStoreTransaction {
private static final Logger log = LoggerFactory.getLogger(BerkeleyJETx.class);
private Transaction tx;
private List<Cursor> openCursors = new ArrayList<Cursor>();
public BerkeleyJETx(Transaction t, StoreTxConfig config) {
super(config);
tx = t;
}
public Transaction getTransaction() {
return tx;
}
void registerCursor(Cursor cursor) {
Preconditions.checkArgument(cursor != null);
synchronized (openCursors) {
//TODO: attempt to remove closed cursors if there are too many
openCursors.add(cursor);
}
}
private void closeOpenIterators() throws StorageException {
for (Cursor cursor : openCursors) {
cursor.close();
}
}
@Override
public synchronized void rollback() throws StorageException {
super.rollback();
if (tx == null) return;
try {
closeOpenIterators();
tx.abort();
tx = null;
} catch (DatabaseException e) {
throw new PermanentStorageException(e);
}
}
@Override
public synchronized void commit() throws StorageException {
super.commit();
if (tx == null) return;
try {
closeOpenIterators();
tx.commit();
tx = null;
} catch (DatabaseException e) {
throw new PermanentStorageException(e);
}
}
}