import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.SettableFuture;
-import org.opendaylight.controller.cluster.datastore.identifiers.TransactionIdentifier;
+import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
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;
* @author Thomas Pantelis
*/
abstract class LocalTransactionContext extends AbstractTransactionContext {
-
private final DOMStoreTransaction txDelegate;
- private final OperationLimiter limiter;
- LocalTransactionContext(TransactionIdentifier identifier, DOMStoreTransaction txDelegate, OperationLimiter limiter) {
- super(identifier);
+ LocalTransactionContext(DOMStoreTransaction txDelegate, OperationLimiter limiter) {
+ super(limiter);
this.txDelegate = Preconditions.checkNotNull(txDelegate);
- this.limiter = Preconditions.checkNotNull(limiter);
}
protected abstract DOMStoreWriteTransaction getWriteDelegate();
public void writeData(YangInstanceIdentifier path, NormalizedNode<?, ?> data) {
incrementModificationCount();
getWriteDelegate().write(path, data);
- limiter.release();
+ releaseOperation();
}
@Override
public void mergeData(YangInstanceIdentifier path, NormalizedNode<?, ?> data) {
incrementModificationCount();
getWriteDelegate().merge(path, data);
- limiter.release();
+ releaseOperation();
}
@Override
public void deleteData(YangInstanceIdentifier path) {
incrementModificationCount();
getWriteDelegate().delete(path);
- limiter.release();
+ releaseOperation();
}
@Override
public void readData(YangInstanceIdentifier path, final SettableFuture<Optional<NormalizedNode<?, ?>>> proxyFuture) {
Futures.addCallback(getReadDelegate().read(path), new FutureCallback<Optional<NormalizedNode<?, ?>>>() {
@Override
- public void onSuccess(Optional<NormalizedNode<?, ?>> result) {
+ public void onSuccess(final Optional<NormalizedNode<?, ?>> result) {
proxyFuture.set(result);
- limiter.release();
+ releaseOperation();
}
@Override
- public void onFailure(Throwable t) {
+ public void onFailure(final Throwable t) {
proxyFuture.setException(t);
- limiter.release();
+ releaseOperation();
}
});
}
public void dataExists(YangInstanceIdentifier path, final SettableFuture<Boolean> proxyFuture) {
Futures.addCallback(getReadDelegate().exists(path), new FutureCallback<Boolean>() {
@Override
- public void onSuccess(Boolean result) {
+ public void onSuccess(final Boolean result) {
proxyFuture.set(result);
- limiter.release();
+ releaseOperation();
}
@Override
- public void onFailure(Throwable t) {
+ public void onFailure(final Throwable t) {
proxyFuture.setException(t);
- limiter.release();
+ releaseOperation();
}
});
}
private LocalThreePhaseCommitCohort ready() {
logModificationCount();
- LocalThreePhaseCommitCohort ready = (LocalThreePhaseCommitCohort) getWriteDelegate().ready();
- limiter.release();
- return ready;
+ acquireOperation();
+ return (LocalThreePhaseCommitCohort) getWriteDelegate().ready();
+ }
+
+ @SuppressWarnings({ "unchecked", "rawtypes" })
+ private <T extends Future> T completeOperation(final ActorContext actorContext, final T operationFuture) {
+ operationFuture.onComplete(getLimiter(), actorContext.getClientDispatcher());
+ return operationFuture;
}
@Override
public Future<ActorSelection> readyTransaction() {
- return ready().initiateCoordinatedCommit();
+ final LocalThreePhaseCommitCohort cohort = ready();
+ return completeOperation(cohort.getActorContext(), cohort.initiateCoordinatedCommit());
}
@Override
public Future<Object> directCommit() {
- return ready().initiateDirectCommit();
+ final LocalThreePhaseCommitCohort cohort = ready();
+ return completeOperation(cohort.getActorContext(), cohort.initiateDirectCommit());
}
@Override
@Override
public void closeTransaction() {
txDelegate.close();
+ releaseOperation();
}
}