X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FTransactionChainProxy.java;h=cf00d5fa39e3d317fe99af05d2f30f112fcb6f22;hb=7161b121b21aeea325fe33485c841af31f9f0cfd;hp=4b75fbbd5cdbf61dde4c759896b62a521a3be048;hpb=9681454e5a6acb2ce5e87c72c06e6a03e901c765;p=controller.git diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/TransactionChainProxy.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/TransactionChainProxy.java index 4b75fbbd5c..cf00d5fa39 100644 --- a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/TransactionChainProxy.java +++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/TransactionChainProxy.java @@ -8,11 +8,19 @@ package org.opendaylight.controller.cluster.datastore; import akka.actor.ActorSelection; +import akka.dispatch.Futures; import akka.dispatch.OnComplete; import com.google.common.base.Preconditions; +import java.util.ArrayList; import java.util.Collection; +import java.util.List; +import java.util.Map.Entry; +import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.ConcurrentMap; import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicReferenceFieldUpdater; +import java.util.function.Function; +import javax.annotation.Nonnull; import org.opendaylight.controller.cluster.datastore.identifiers.TransactionChainIdentifier; import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier; import org.opendaylight.controller.cluster.datastore.messages.CloseTransactionChain; @@ -120,6 +128,27 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory + * A Promise is added via newReadOnlyTransaction. When the parent class completes the primary shard + * lookup and creates the TransactionContext (either success or failure), onTransactionContextCreated is + * called which completes the Promise. A write tx that is created prior to completion will wait on the + * Promise's Future via findPrimaryShard. + */ + private final ConcurrentMap> priorReadOnlyTxPromises = new ConcurrentHashMap<>(); + TransactionChainProxy(final TransactionContextFactory parent) { super(parent.getActorContext()); @@ -134,7 +163,9 @@ final class TransactionChainProxy extends AbstractTransactionContextFactorypromise()); + return transactionProxy; } @Override @@ -154,7 +185,13 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory() { + @Override + public Object apply(Short version) { + return new CloseTransactionChain(transactionChainId.toString(), version).toSerializable(); + } + }); } private TransactionProxy allocateWriteTransaction(final TransactionType type) { @@ -178,15 +215,16 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory findPrimaryShard(final String shardName, final String txId) { + protected Future findPrimaryShard(final String shardName, final TransactionIdentifier txId) { // Read current state atomically final State localState = currentState; // There are no outstanding futures, shortcut - final Future previous = localState.previousFuture(); + Future previous = localState.previousFuture(); if (previous == null) { - return parent.findPrimaryShard(shardName, txId); + return combineFutureWithPossiblePriorReadOnlyTxFutures(parent.findPrimaryShard(shardName, txId), txId); } final String previousTransactionId; @@ -199,8 +237,10 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory returnPromise = akka.dispatch.Futures.promise(); + final Promise returnPromise = Futures.promise(); final OnComplete onComplete = new OnComplete() { @Override @@ -224,6 +264,42 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory Future combineFutureWithPossiblePriorReadOnlyTxFutures(final Future future, + final TransactionIdentifier txId) { + if(!priorReadOnlyTxPromises.containsKey(txId) && !priorReadOnlyTxPromises.isEmpty()) { + Collection>> priorReadOnlyTxPromiseEntries = + new ArrayList<>(priorReadOnlyTxPromises.entrySet()); + if(priorReadOnlyTxPromiseEntries.isEmpty()) { + return future; + } + + List> priorReadOnlyTxFutures = new ArrayList<>(priorReadOnlyTxPromiseEntries.size()); + for(Entry> entry: priorReadOnlyTxPromiseEntries) { + LOG.debug("Tx: {} - waiting on future for prior read-only Tx {}", txId, entry.getKey()); + priorReadOnlyTxFutures.add(entry.getValue().future()); + } + + Future> combinedFutures = Futures.sequence(priorReadOnlyTxFutures, + getActorContext().getClientDispatcher()); + + final Promise returnPromise = Futures.promise(); + final OnComplete> onComplete = new OnComplete>() { + @Override + public void onComplete(final Throwable failure, final Iterable notUsed) { + LOG.debug("Tx: {} - prior read-only Tx futures complete", txId); + + // Complete the returned Promise with the original Future. + returnPromise.completeWith(future); + } + }; + + combinedFutures.onComplete(onComplete, getActorContext().getClientDispatcher()); + return returnPromise.future(); + } else { + return future; + } + } + @Override protected void onTransactionReady(final TransactionIdentifier transaction, final Collection> cohortFutures) { final State localState = currentState; @@ -238,8 +314,7 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory> combined = akka.dispatch.Futures.sequence( - cohortFutures, getActorContext().getClientDispatcher()); + final Future> combined = Futures.sequence(cohortFutures, getActorContext().getClientDispatcher()); // Record the we have outstanding futures final State newState = new Submitted(transaction, combined); @@ -255,6 +330,14 @@ final class TransactionChainProxy extends AbstractTransactionContextFactory promise = priorReadOnlyTxPromises.remove(transactionId); + if(promise != null) { + promise.success(null); + } + } + @Override protected TransactionIdentifier nextIdentifier() { return transactionChainId.newTransactionIdentifier();