--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.controller.cluster.datastore;
+
+import akka.actor.Status;
+import akka.actor.Status.Failure;
+import akka.dispatch.ExecutionContexts;
+import akka.dispatch.Futures;
+import akka.dispatch.Recover;
+import akka.japi.Function;
+import akka.pattern.Patterns;
+import akka.util.Timeout;
+import com.google.common.base.Preconditions;
+import com.google.common.base.Throwables;
+import com.google.common.collect.Iterables;
+import java.util.Collection;
+import java.util.Iterator;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeoutException;
+import org.opendaylight.controller.cluster.datastore.DataTreeCohortActor.CanCommit;
+import org.opendaylight.controller.cluster.datastore.DataTreeCohortActor.Success;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateTip;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import scala.concurrent.Await;
+import scala.concurrent.Future;
+
+/**
+ *
+ * Composite cohort, which coordinates multiple user-provided cohorts as if it was only one cohort.
+ *
+ * It tracks current operation and list of cohorts which successfuly finished previous phase in
+ * case, if abort is necessary to invoke it only on cohort steps which are still active.
+ *
+ */
+class CompositeDataTreeCohort {
+
+ private enum State {
+ /**
+ * Cohorts are idle, no messages were sent.
+ */
+ IDLE,
+ /**
+ * CanCommit message was sent to all participating cohorts.
+ */
+ CAN_COMMIT_SENT,
+ /**
+ * Successful canCommit responses were received from every participating cohort.
+ */
+ CAN_COMMIT_SUCCESSFUL,
+ /**
+ * PreCommit message was sent to all participating cohorts.
+ */
+ PRE_COMMIT_SENT,
+ /**
+ * Successful preCommit responses were received from every participating cohort.
+ */
+ PRE_COMMIT_SUCCESSFUL,
+ /**
+ * Commit message was send to all participating cohorts.
+ */
+ COMMIT_SENT,
+ /**
+ * Successful commit responses were received from all participating cohorts.
+ */
+ COMMITED,
+ /**
+ * Some of cohorts responsed back with unsuccessful message.
+ *
+ */
+ FAILED,
+ /**
+ *
+ * Abort message was send to all cohorts which responded with success previously.
+ *
+ */
+ ABORTED
+ }
+
+ protected static final Recover<Object> EXCEPTION_TO_MESSAGE = new Recover<Object>() {
+ @Override
+ public Failure recover(Throwable error) throws Throwable {
+ return new Failure(error);
+ }
+ };
+
+
+ private final DataTreeCohortActorRegistry registry;
+ private final String txId;
+ private final SchemaContext schema;
+ private final Timeout timeout;
+ private Iterable<Success> successfulFromPrevious;
+ private State state = State.IDLE;
+
+ CompositeDataTreeCohort(DataTreeCohortActorRegistry registry, String txId, SchemaContext schema, Timeout timeout) {
+ this.registry = Preconditions.checkNotNull(registry);
+ this.txId = Preconditions.checkNotNull(txId);
+ this.schema = Preconditions.checkNotNull(schema);
+ this.timeout = Preconditions.checkNotNull(timeout);
+ }
+
+ void canCommit(DataTreeCandidateTip tip) throws ExecutionException, TimeoutException {
+ Collection<CanCommit> messages = registry.createCanCommitMessages(txId, tip, schema);
+ // FIXME: Optimize empty collection list with pre-created futures, containing success.
+ Future<Iterable<Object>> canCommitsFuture =
+ Futures.traverse(messages, new Function<CanCommit, Future<Object>>() {
+ @Override
+ public Future<Object> apply(CanCommit input) {
+ return Patterns.ask(input.getCohort(), input, timeout).recover(EXCEPTION_TO_MESSAGE,
+ ExecutionContexts.global());
+ }
+ }, ExecutionContexts.global());
+ changeStateFrom(State.IDLE, State.CAN_COMMIT_SENT);
+ processResponses(canCommitsFuture, State.CAN_COMMIT_SENT, State.CAN_COMMIT_SUCCESSFUL);
+ }
+
+ void preCommit() throws ExecutionException, TimeoutException {
+ Preconditions.checkState(successfulFromPrevious != null);
+ Future<Iterable<Object>> preCommitFutures = sendMesageToSuccessful(new DataTreeCohortActor.PreCommit(txId));
+ changeStateFrom(State.CAN_COMMIT_SUCCESSFUL, State.PRE_COMMIT_SENT);
+ processResponses(preCommitFutures, State.PRE_COMMIT_SENT, State.PRE_COMMIT_SUCCESSFUL);
+ }
+
+ void commit() throws ExecutionException, TimeoutException {
+ Preconditions.checkState(successfulFromPrevious != null);
+ Future<Iterable<Object>> commitsFuture = sendMesageToSuccessful(new DataTreeCohortActor.Commit(txId));
+ changeStateFrom(State.PRE_COMMIT_SUCCESSFUL, State.COMMIT_SENT);
+ processResponses(commitsFuture, State.COMMIT_SENT, State.COMMITED);
+ }
+
+ void abort() throws TimeoutException {
+ if (successfulFromPrevious != null) {
+ sendMesageToSuccessful(new DataTreeCohortActor.Abort(txId));
+ }
+ }
+
+ private Future<Iterable<Object>> sendMesageToSuccessful(final Object message) {
+ return Futures.traverse(successfulFromPrevious, new Function<DataTreeCohortActor.Success, Future<Object>>() {
+
+ @Override
+ public Future<Object> apply(DataTreeCohortActor.Success cohortResponse) throws Exception {
+ return Patterns.ask(cohortResponse.getCohort(), message, timeout);
+ }
+
+ }, ExecutionContexts.global());
+ }
+
+ private void processResponses(Future<Iterable<Object>> resultsFuture, State currentState, State afterState)
+ throws TimeoutException, ExecutionException {
+ final Iterable<Object> results;
+ try {
+ results = Await.result(resultsFuture, timeout.duration());
+ } catch (Exception e) {
+ successfulFromPrevious = null;
+ Throwables.propagateIfInstanceOf(e, TimeoutException.class);
+ throw Throwables.propagate(e);
+ }
+ Iterable<Failure> failed = Iterables.filter(results, Status.Failure.class);
+ Iterable<Success> successful = Iterables.filter(results, DataTreeCohortActor.Success.class);
+ successfulFromPrevious = successful;
+ if (!Iterables.isEmpty(failed)) {
+ changeStateFrom(currentState, State.FAILED);
+ Iterator<Failure> it = failed.iterator();
+ Throwable firstEx = it.next().cause();
+ while (it.hasNext()) {
+ firstEx.addSuppressed(it.next().cause());
+ }
+ Throwables.propagateIfPossible(firstEx, ExecutionException.class);
+ Throwables.propagateIfPossible(firstEx, TimeoutException.class);
+ throw Throwables.propagate(firstEx);
+ }
+ changeStateFrom(currentState, afterState);
+ }
+
+ void changeStateFrom(State expected, State followup) {
+ Preconditions.checkState(state == expected);
+ state = followup;
+ }
+}