*/
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.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;
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.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.cohort = requireNonNull(cohort);
this.transactionId = cohort.getIdentifier();
this.transaction = null;
this.clientVersion = clientVersion;
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.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.
}
}
+ boolean isSealed() {
+ return cohort != null;
+ }
+
+ Optional<SortedSet<String>> getParticipatingShardNames() {
+ return cohort != null ? cohort.getParticipatingShardNames() : Optional.empty();
+ }
+
boolean isDoImmediateCommit() {
return doImmediateCommit;
}