-/*
- * FIXME: why do we need this interface? It should be possible to integrate it with
- * AbstractTransactionContext, which is the only implementation anyway.
- */
-interface TransactionContext {
- void closeTransaction();
+abstract class TransactionContext extends AbstractSimpleIdentifiable<TransactionIdentifier> {
+ private static final Logger LOG = LoggerFactory.getLogger(TransactionContext.class);
+
+ private final short transactionVersion;
+
+ private long modificationCount = 0;
+ private boolean handOffComplete;
+
+ TransactionContext(final TransactionIdentifier transactionIdentifier) {
+ this(transactionIdentifier, DataStoreVersions.CURRENT_VERSION);
+ }
+
+ TransactionContext(final TransactionIdentifier transactionIdentifier, final short transactionVersion) {
+ super(transactionIdentifier);
+ this.transactionVersion = transactionVersion;
+ }
+
+ final short getTransactionVersion() {
+ return transactionVersion;
+ }
+
+ final void incrementModificationCount() {
+ modificationCount++;
+ }
+
+ final void logModificationCount() {
+ LOG.debug("Total modifications on Tx {} = [ {} ]", getIdentifier(), modificationCount);
+ }
+
+ /**
+ * Invoked by {@link AbstractTransactionContextWrapper} when it has finished handing
+ * off operations to this context. From this point on, the context is responsible
+ * for throttling operations.
+ *
+ * <p>
+ * Implementations can rely on the wrapper calling this operation in a synchronized
+ * block, so they do not need to ensure visibility of this state transition themselves.
+ */
+ final void operationHandOffComplete() {
+ handOffComplete = true;
+ }
+
+ final boolean isOperationHandOffComplete() {
+ return handOffComplete;
+ }