* @author Robert Varga
*/
@Beta
-public final class ClientLocalHistory extends AbstractClientHistory implements AutoCloseable {
+public class ClientLocalHistory extends AbstractClientHistory implements AutoCloseable {
ClientLocalHistory(final AbstractDataStoreClientBehavior client, final LocalHistoryIdentifier historyId) {
super(client, historyId);
}
@Override
public void close() {
- final State local = state();
- if (local != State.CLOSED) {
- Preconditions.checkState(local == State.IDLE, "Local history %s has an open transaction", this);
- updateState(local, State.CLOSED);
- }
+ doClose();
}
- @Override
- ClientTransaction doCreateTransaction() {
+ private State ensureIdleState() {
final State local = state();
Preconditions.checkState(local == State.IDLE, "Local history %s state is %s", this, local);
- updateState(local, State.TX_OPEN);
+ return local;
+ }
+ @Override
+ ClientSnapshot doCreateSnapshot() {
+ ensureIdleState();
+ return new ClientSnapshot(this, new TransactionIdentifier(getIdentifier(), nextTx()));
+ }
+
+ @Override
+ ClientTransaction doCreateTransaction() {
+ updateState(ensureIdleState(), State.TX_OPEN);
return new ClientTransaction(this, new TransactionIdentifier(getIdentifier(), nextTx()));
}
@Override
- AbstractTransactionCommitCohort onTransactionReady(final TransactionIdentifier txId,
+ void onTransactionAbort(final AbstractClientHandle<?> snap) {
+ if (snap instanceof ClientTransaction) {
+ final State local = state();
+ if (local == State.TX_OPEN) {
+ updateState(local, State.IDLE);
+ }
+ }
+
+ super.onTransactionAbort(snap);
+ }
+
+ @Override
+ AbstractTransactionCommitCohort onTransactionReady(final ClientTransaction tx,
final AbstractTransactionCommitCohort cohort) {
+
final State local = state();
switch (local) {
case CLOSED:
- return super.onTransactionReady(txId, cohort);
+ return super.onTransactionReady(tx, cohort);
case IDLE:
throw new IllegalStateException(String.format("Local history %s is idle when readying transaction %s",
- this, txId));
+ this, tx.getIdentifier()));
case TX_OPEN:
updateState(local, State.IDLE);
- return super.onTransactionReady(txId, cohort);
+ return super.onTransactionReady(tx, cohort);
default:
throw new IllegalStateException(String.format("Local history %s in unhandled state %s", this, local));
@Override
ProxyHistory createHistoryProxy(final LocalHistoryIdentifier historyId,
final AbstractClientConnection<ShardBackendInfo> connection) {
- return ProxyHistory.createClient(connection, historyId);
+ return ProxyHistory.createClient(this, connection, historyId);
}
}