*/
package org.opendaylight.controller.cluster.datastore;
+import static java.util.Objects.requireNonNull;
+
import akka.actor.ActorSelection;
-import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.MoreExecutors;
import com.google.common.util.concurrent.SettableFuture;
+import java.util.Optional;
+import java.util.SortedSet;
+import java.util.function.Consumer;
import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
import org.opendaylight.controller.cluster.datastore.messages.AbstractRead;
-import org.opendaylight.controller.cluster.datastore.modification.AbstractModification;
-import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
-import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransaction;
-import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
+import org.opendaylight.mdsal.common.api.ReadFailedException;
+import org.opendaylight.mdsal.dom.spi.store.DOMStoreReadTransaction;
+import org.opendaylight.mdsal.dom.spi.store.DOMStoreTransaction;
+import org.opendaylight.mdsal.dom.spi.store.DOMStoreWriteTransaction;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
import scala.concurrent.Future;
/**
*
* @author Thomas Pantelis
*/
-abstract class LocalTransactionContext extends AbstractTransactionContext {
+abstract class LocalTransactionContext extends TransactionContext {
private final DOMStoreTransaction txDelegate;
private final LocalTransactionReadySupport readySupport;
private Exception operationError;
LocalTransactionContext(final DOMStoreTransaction txDelegate, final TransactionIdentifier identifier,
final LocalTransactionReadySupport readySupport) {
super(identifier);
- this.txDelegate = Preconditions.checkNotNull(txDelegate);
+ this.txDelegate = requireNonNull(txDelegate);
this.readySupport = readySupport;
}
- protected abstract DOMStoreWriteTransaction getWriteDelegate();
+ abstract DOMStoreWriteTransaction getWriteDelegate();
- protected abstract DOMStoreReadTransaction getReadDelegate();
+ abstract DOMStoreReadTransaction getReadDelegate();
- @Override
@SuppressWarnings("checkstyle:IllegalCatch")
- public void executeModification(final AbstractModification modification) {
+ private void executeModification(final Consumer<DOMStoreWriteTransaction> consumer) {
incrementModificationCount();
if (operationError == null) {
try {
- modification.apply(getWriteDelegate());
+ consumer.accept(getWriteDelegate());
} catch (Exception e) {
operationError = e;
}
}
@Override
- public <T> void executeRead(final AbstractRead<T> readCmd, final SettableFuture<T> proxyFuture) {
+ void executeDelete(final YangInstanceIdentifier path, final Boolean havePermit) {
+ executeModification(transaction -> transaction.delete(path));
+ }
+
+ @Override
+ void executeMerge(final YangInstanceIdentifier path, final NormalizedNode data, final Boolean havePermit) {
+ executeModification(transaction -> transaction.merge(path, data));
+ }
+
+ @Override
+ void executeWrite(final YangInstanceIdentifier path, final NormalizedNode data, final Boolean havePermit) {
+ executeModification(transaction -> transaction.write(path, data));
+ }
+
+ @Override
+ <T> void executeRead(final AbstractRead<T> readCmd, final SettableFuture<T> proxyFuture,
+ final Boolean havePermit) {
Futures.addCallback(readCmd.apply(getReadDelegate()), new FutureCallback<T>() {
@Override
public void onSuccess(final T result) {
@Override
public void onFailure(final Throwable failure) {
- proxyFuture.setException(failure);
+ proxyFuture.setException(failure instanceof Exception
+ ? ReadFailedException.MAPPER.apply((Exception) failure) : failure);
}
}, MoreExecutors.directExecutor());
}
- private LocalThreePhaseCommitCohort ready() {
- logModificationCount();
- return readySupport.onTransactionReady(getWriteDelegate(), operationError);
- }
-
@Override
- public Future<ActorSelection> readyTransaction() {
+ Future<ActorSelection> readyTransaction(final Boolean havePermit,
+ final Optional<SortedSet<String>> participatingShardNames) {
final LocalThreePhaseCommitCohort cohort = ready();
- return cohort.initiateCoordinatedCommit();
+ return cohort.initiateCoordinatedCommit(participatingShardNames);
}
@Override
- public Future<Object> directCommit() {
+ Future<Object> directCommit(final Boolean havePermit) {
final LocalThreePhaseCommitCohort cohort = ready();
return cohort.initiateDirectCommit();
}
@Override
- public void closeTransaction() {
+ void closeTransaction() {
txDelegate.close();
}
+
+ private LocalThreePhaseCommitCohort ready() {
+ logModificationCount();
+ return readySupport.onTransactionReady(getWriteDelegate(), operationError);
+ }
}