*/
package org.opendaylight.controller.cluster.datastore;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
import akka.actor.ActorRef;
-import com.google.common.base.Preconditions;
import com.google.common.primitives.UnsignedLong;
import com.google.common.util.concurrent.FutureCallback;
-import java.util.concurrent.ExecutionException;
-import java.util.concurrent.TimeoutException;
+import java.util.List;
+import java.util.Optional;
+import java.util.SortedSet;
import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
import org.opendaylight.controller.cluster.datastore.ShardCommitCoordinator.CohortDecorator;
import org.opendaylight.controller.cluster.datastore.modification.Modification;
final class CohortEntry {
private final ReadWriteShardDataTreeTransaction transaction;
- private final TransactionIdentifier transactionID;
+ private final TransactionIdentifier transactionId;
private final short clientVersion;
private RuntimeException lastBatchedModificationsException;
private int totalBatchedModificationsReceived;
+ private int totalOperationsProcessed;
private ShardDataTreeCohort cohort;
private boolean doImmediateCommit;
private ActorRef replySender;
private Shard shard;
private CohortEntry(final ReadWriteShardDataTreeTransaction transaction, final short clientVersion) {
- this.transaction = Preconditions.checkNotNull(transaction);
- this.transactionID = transaction.getId();
+ this.cohort = null;
+ this.transaction = requireNonNull(transaction);
+ this.transactionId = transaction.getIdentifier();
this.clientVersion = clientVersion;
}
private CohortEntry(final ShardDataTreeCohort cohort, final short clientVersion) {
- this.cohort = Preconditions.checkNotNull(cohort);
- this.transactionID = cohort.getIdentifier();
+ this.cohort = requireNonNull(cohort);
+ this.transactionId = cohort.getIdentifier();
this.transaction = null;
this.clientVersion = clientVersion;
}
return new CohortEntry(cohort, clientVersion);
}
- TransactionIdentifier getTransactionID() {
- return transactionID;
+ TransactionIdentifier getTransactionId() {
+ return transactionId;
}
short getClientVersion() {
return totalBatchedModificationsReceived;
}
+ int getTotalOperationsProcessed() {
+ return totalOperationsProcessed;
+ }
+
RuntimeException getLastBatchedModificationsException() {
return lastBatchedModificationsException;
}
@SuppressWarnings("checkstyle:IllegalCatch")
- void applyModifications(final Iterable<Modification> modifications) {
+ void applyModifications(final List<Modification> modifications) {
totalBatchedModificationsReceived++;
if (lastBatchedModificationsException == null) {
+ totalOperationsProcessed += modifications.size();
for (Modification modification : modifications) {
try {
modification.apply(transaction.getSnapshot());
cohort.commit(callback);
}
- void abort() throws InterruptedException, ExecutionException, TimeoutException {
- cohort.abort().get();
+ void abort(final FutureCallback<Void> callback) {
+ cohort.abort(callback);
}
- void ready(final CohortDecorator cohortDecorator) {
- Preconditions.checkState(cohort == null, "cohort was already set");
+ void ready(final Optional<SortedSet<String>> participatingShardNames, final CohortDecorator cohortDecorator) {
+ checkState(cohort == null, "cohort was already set");
- cohort = transaction.ready();
+ cohort = transaction.ready(participatingShardNames);
if (cohortDecorator != null) {
// Call the hook for unit tests.
- cohort = cohortDecorator.decorate(transactionID, cohort);
+ cohort = cohortDecorator.decorate(transactionId, cohort);
}
}
+ boolean isSealed() {
+ return cohort != null;
+ }
+
+ Optional<SortedSet<String>> getParticipatingShardNames() {
+ return cohort != null ? cohort.getParticipatingShardNames() : Optional.empty();
+ }
+
boolean isDoImmediateCommit() {
return doImmediateCommit;
}
@Override
public String toString() {
final StringBuilder builder = new StringBuilder();
- builder.append("CohortEntry [transactionID=").append(transactionID).append(", doImmediateCommit=")
+ builder.append("CohortEntry [transactionId=").append(transactionId).append(", doImmediateCommit=")
.append(doImmediateCommit).append("]");
return builder.toString();
}